NIFI-5917 Fix TestSelectHiveQL.testNoTimeLimit
authorKoji Kawamura <ijokarumawak@apache.org>
Fri, 28 Dec 2018 04:09:50 +0000 (13:09 +0900)
committerPierre Villard <pierre.villard.fr@gmail.com>
Thu, 3 Jan 2019 10:31:38 +0000 (11:31 +0100)
Signed-off-by: Pierre Villard <pierre.villard.fr@gmail.com>
This closes #3237.

nifi-nar-bundles/nifi-hive-bundle/nifi-hive-processors/src/test/java/org/apache/nifi/processors/hive/TestSelectHiveQL.java
nifi-nar-bundles/nifi-hive-bundle/nifi-hive3-processors/src/test/java/org/apache/nifi/processors/hive/TestSelectHive3QL.java

index 484fa67..bb84486 100644 (file)
@@ -134,7 +134,7 @@ public class TestSelectHiveQL {
         invokeOnTrigger(QUERY_WITH_EL, true, "Avro");
 
         final List<ProvenanceEventRecord> provenanceEvents = runner.getProvenanceEvents();
-        assertEquals(2, provenanceEvents.size());
+        assertEquals(3, provenanceEvents.size());
 
         final ProvenanceEventRecord provenance0 = provenanceEvents.get(0);
         assertEquals(ProvenanceEventType.FORK, provenance0.getEventType());
@@ -142,6 +142,9 @@ public class TestSelectHiveQL {
         final ProvenanceEventRecord provenance1 = provenanceEvents.get(1);
         assertEquals(ProvenanceEventType.FETCH, provenance1.getEventType());
         assertEquals("jdbc:derby:target/db;create=true", provenance1.getTransitUri());
+
+        final ProvenanceEventRecord provenance2 = provenanceEvents.get(2);
+        assertEquals(ProvenanceEventType.FORK, provenance2.getEventType());
     }
 
 
index 67f0dd4..77a20f8 100644 (file)
@@ -134,7 +134,7 @@ public class TestSelectHive3QL {
         invokeOnTrigger(QUERY_WITH_EL, true, "Avro");
 
         final List<ProvenanceEventRecord> provenanceEvents = runner.getProvenanceEvents();
-        assertEquals(2, provenanceEvents.size());
+        assertEquals(3, provenanceEvents.size());
 
         final ProvenanceEventRecord provenance0 = provenanceEvents.get(0);
         assertEquals(ProvenanceEventType.FORK, provenance0.getEventType());
@@ -142,6 +142,9 @@ public class TestSelectHive3QL {
         final ProvenanceEventRecord provenance1 = provenanceEvents.get(1);
         assertEquals(ProvenanceEventType.FETCH, provenance1.getEventType());
         assertEquals("jdbc:derby:target/db;create=true", provenance1.getTransitUri());
+
+        final ProvenanceEventRecord provenance2 = provenanceEvents.get(2);
+        assertEquals(ProvenanceEventType.FORK, provenance2.getEventType());
     }