[GitHub] nifi pull request #2695: NIFI-5044 SelectHiveQL accept only one statement

[pvillard31]

Github user pvillard31 commented on a diff in the pull request:

    https://github.com/apache/nifi/pull/2695#discussion_r190637767
  
    --- Diff: 
nifi-nar-bundles/nifi-hive-bundle/nifi-hive-processors/src/test/java/org/apache/nifi/processors/hive/TestSelectHiveQL.java
 ---
    @@ -198,6 +200,51 @@ public void testWithSqlException() throws SQLException 
{
             runner.assertAllFlowFilesTransferred(SelectHiveQL.REL_FAILURE, 1);
         }
     
    +    @Test
    +    public void invokeOnTriggerExceptionInPreQieriesNoIncomingFlows()
    +            throws InitializationException, ClassNotFoundException, 
SQLException, IOException {
    +
    +        doOnTrigger(QUERY_WITHOUT_EL, false, CSV,
    +                "select 'no exception' from persons; select exception from 
persons",
    +                null);
    +
    +        runner.assertAllFlowFilesTransferred(SelectHiveQL.REL_FAILURE, 1);
    +    }
    +
    +    @Test
    +    public void invokeOnTriggerExceptionInPreQieriesWithIncomingFlows()
    +            throws InitializationException, ClassNotFoundException, 
SQLException, IOException {
    +
    +        doOnTrigger(QUERY_WITHOUT_EL, true, CSV,
    +                "select 'no exception' from persons; select exception from 
persons",
    +                null);
    +
    +        runner.assertAllFlowFilesTransferred(SelectHiveQL.REL_FAILURE, 1);
    +    }
    +
    +    @Test
    +    public void invokeOnTriggerExceptionInPostQieriesNoIncomingFlows()
    +            throws InitializationException, ClassNotFoundException, 
SQLException, IOException {
    +
    +        doOnTrigger(QUERY_WITHOUT_EL, false, CSV,
    +                null,
    +                "select 'no exception' from persons; select exception from 
persons");
    +
    +        runner.assertAllFlowFilesTransferred(SelectHiveQL.REL_FAILURE, 1);
    --- End diff --
    
    Same here


---