diff --git a/connector/connector-sdk/src/test/java/org/apache/sqoop/connector/common/TestSqoopIDFUtils.java b/connector/connector-sdk/src/test/java/org/apache/sqoop/connector/common/TestSqoopIDFUtils.java index e9108b0e..68890ccb 100644 --- a/connector/connector-sdk/src/test/java/org/apache/sqoop/connector/common/TestSqoopIDFUtils.java +++ b/connector/connector-sdk/src/test/java/org/apache/sqoop/connector/common/TestSqoopIDFUtils.java @@ -175,5 +175,18 @@ public void testEncodeMapToCSVString() { String encodedText = encodeToCSVMap(map, mapCol); assertEquals(encodedText, "'{\"A\":[\"A\",\"B\"]}'"); } + + @Test + public void testParseCSVString() { + + String csv= "'hello, world','34',45"; + String[] arr = parseCSVString(csv); + assertEquals(arr.length, 3); + assertEquals(arr[0], "'hello, world'"); + assertEquals(arr[1], "'34'"); + assertEquals(arr[2], "45"); + + } + } \ No newline at end of file diff --git a/connector/connector-sdk/src/test/java/org/apache/sqoop/connector/idf/TestCSVIntermediateDataFormat.java b/connector/connector-sdk/src/test/java/org/apache/sqoop/connector/idf/TestCSVIntermediateDataFormat.java index 2602c613..64b18fd4 100644 --- a/connector/connector-sdk/src/test/java/org/apache/sqoop/connector/idf/TestCSVIntermediateDataFormat.java +++ b/connector/connector-sdk/src/test/java/org/apache/sqoop/connector/idf/TestCSVIntermediateDataFormat.java @@ -197,6 +197,19 @@ public void testInputAsCSVTextInAndDataOut() { assertEquals(testData, dataFormat.getData()); } + @Test + public void testInputAsCSVTextInObjectOutWithSingleColumn() { + + String testData = "'\"hello, world\"'"; + Schema schema = new Schema("test"); + schema.addColumn(new Text("text")); + + dataFormat.setSchema(schema); + dataFormat.setCSVTextData(testData); + + Object[] out = dataFormat.getObjectData(); + assertEquals("\"hello, world\"",out[0]); + } @Test public void testInputAsCSVTextInObjectOut() {