final JavaRDD<String> dataToWrite; if (this.conf.isColumnHeader()) { final String header = this.converter.getHeader(data); dataToWrite = addColumnHeader(header, dataRepartitioned); } else {
@Test public void testGetHeaderWithCsv() { final String separator = ","; final JavaRDD<AvroPayload> payloadData = AvroPayloadUtil.generateTestData(this.jsc.get(), 10, StringTypes.EMPTY); final Configuration conf = initConf(separator, "csv"); log.info("Starting to get data header."); final FileSinkDataConverter converter = new FileSinkDataConverter(conf, new ErrorExtractor()); final String header = converter.getHeader(payloadData); final String resultHeader = "int_field,string_field,boolean_field"; Assert.assertEquals(resultHeader, header); log.info("Header: {}", header); }