DRILL-446 Fix Json Array parsing

1 files changed, 29 insertions, 0 deletions

diff --git a/exec/java-exec/src/test/java/org/apache/drill/exec/store/json/JSONRecordReaderTest.java b/exec/java-exec/src/test/java/org/apache/drill/exec/store/json/JSONRecordReaderTest.java
index bfeb8929b..e812997b6 100644
--- a/exec/java-exec/src/test/java/org/apache/drill/exec/store/json/JSONRecordReaderTest.java
+++ b/exec/java-exec/src/test/java/org/apache/drill/exec/store/json/JSONRecordReaderTest.java
@@ -387,4 +387,33 @@ public class JSONRecordReaderTest {
     assertEquals(0, jr.next());
     assertTrue(mutator.getRemovedFields().isEmpty());
   }
+  
+  @Test
+  public void testJsonArrayandNormalFields(@Injectable final FragmentContext context) throws ExecutionSetupException, IOException {
+    new Expectations() {
+      {
+        context.getAllocator();
+        returns(new TopLevelAllocator());
+      }
+    };
+
+    JSONRecordReader jr = new JSONRecordReader(context,
+        FileUtils.getResourceAsFile("/scan_json_test_7.json").toURI().toString(),
+        FileSystem.getLocal(new Configuration()), null, null);
+
+    MockOutputMutator mutator = new MockOutputMutator();
+    List<ValueVector> addFields = mutator.getAddFields();
+    jr.setup(mutator);
+    assertEquals(2, jr.next());
+    assertEquals(3, addFields.size());
+    
+    assertField(addFields.get(0), 0, MinorType.VARCHAR, "ABC", "test");
+    assertField(addFields.get(2), 0, MinorType.VARCHAR, "drill", "a");
+    assertField(addFields.get(0), 1, MinorType.VARCHAR, "abc", "test");
+    assertField(addFields.get(2), 1, MinorType.VARCHAR, "apache", "a");
+
+
+    assertEquals(0, jr.next());
+    assertTrue(mutator.getRemovedFields().isEmpty());
+  }
 }