From 595a2decc67d464534ce33f18440543620165bbb Mon Sep 17 00:00:00 2001 From: Koji Kawamura Date: Fri, 28 Dec 2018 13:09:50 +0900 Subject: [PATCH] NIFI-5917 Fix TestSelectHiveQL.testNoTimeLimit Signed-off-by: Pierre Villard This closes #3237. --- .../org/apache/nifi/processors/hive/TestSelectHiveQL.java | 5 ++++- .../org/apache/nifi/processors/hive/TestSelectHive3QL.java | 5 ++++- 2 files changed, 8 insertions(+), 2 deletions(-) diff --git a/nifi-nar-bundles/nifi-hive-bundle/nifi-hive-processors/src/test/java/org/apache/nifi/processors/hive/TestSelectHiveQL.java b/nifi-nar-bundles/nifi-hive-bundle/nifi-hive-processors/src/test/java/org/apache/nifi/processors/hive/TestSelectHiveQL.java index 484fa67c24..bb8448654f 100644 --- a/nifi-nar-bundles/nifi-hive-bundle/nifi-hive-processors/src/test/java/org/apache/nifi/processors/hive/TestSelectHiveQL.java +++ b/nifi-nar-bundles/nifi-hive-bundle/nifi-hive-processors/src/test/java/org/apache/nifi/processors/hive/TestSelectHiveQL.java @@ -134,7 +134,7 @@ public class TestSelectHiveQL { invokeOnTrigger(QUERY_WITH_EL, true, "Avro"); final List 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()); } diff --git a/nifi-nar-bundles/nifi-hive-bundle/nifi-hive3-processors/src/test/java/org/apache/nifi/processors/hive/TestSelectHive3QL.java b/nifi-nar-bundles/nifi-hive-bundle/nifi-hive3-processors/src/test/java/org/apache/nifi/processors/hive/TestSelectHive3QL.java index 67f0dd462b..77a20f8c83 100644 --- a/nifi-nar-bundles/nifi-hive-bundle/nifi-hive3-processors/src/test/java/org/apache/nifi/processors/hive/TestSelectHive3QL.java +++ b/nifi-nar-bundles/nifi-hive-bundle/nifi-hive3-processors/src/test/java/org/apache/nifi/processors/hive/TestSelectHive3QL.java @@ -134,7 +134,7 @@ public class TestSelectHive3QL { invokeOnTrigger(QUERY_WITH_EL, true, "Avro"); final List 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()); }