Github user gvramana commented on a diff in the pull request:
https://github.com/apache/carbondata/pull/2224#discussion_r183800623
--- Diff:
store/sdk/src/test/java/org/apache/carbondata/sdk/file/CSVCarbonWriterTest.java
---
@@ -205,4 +205,70 @@ public void testSchemaPersistence() throws IOException
{
FileUtils.deleteDirectory(new File(path));
}
+ @Test
+ public void testTaskNo() throws IOException {
+ // TODO: write all data type and read by CarbonRecordReader to verify
the content
+ String path = "./testWriteFiles";
+ FileUtils.deleteDirectory(new File(path));
+
+ Field[] fields = new Field[2];
+ fields[0] = new Field("stringField", DataTypes.STRING);
+ fields[1] = new Field("intField", DataTypes.INT);
+
+
+ try {
+ CarbonWriterBuilder builder = CarbonWriter.builder()
+ .withSchema(new Schema(fields))
+ .isTransactionalTable(true).taskNo("5")
+ .outputPath(path);
+
+ CarbonWriter writer = builder.buildWriterForCSVInput();
+
+ for (int i = 0; i < 2; i++) {
+ String[] row = new String[]{
+ "robot" + (i % 10),
+ String.valueOf(i)
+ };
+ writer.write(row);
+ }
+ writer.close();
+ } catch (Exception e) {
+ e.printStackTrace();
+ Assert.fail(e.getMessage());
+ }
+
+ File segmentFolder = new File(CarbonTablePath.getSegmentPath(path,
"null"));
+ Assert.assertTrue(segmentFolder.exists());
+
+ File[] dataFiles = segmentFolder.listFiles(new FileFilter() {
+ @Override public boolean accept(File pathname) {
+ return
pathname.getName().endsWith(CarbonCommonConstants.FACT_FILE_EXT);
+ }
+ });
+ Assert.assertNotNull(dataFiles);
+ Assert.assertTrue(dataFiles.length > 0);
+ String[] splitedcarbonFileName =
getSplitedcarbonFileName(dataFiles[0]);
--- End diff --
call this in finally block FileUtils.deleteDirectory(new File(path));
---
