[nifi] branch main updated: NIFI-11064 Changed TestPutSFTP run iterations based on files queued

[pv...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

pvillard pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/nifi.git


The following commit(s) were added to refs/heads/main by this push:
     new 7db6f17350 NIFI-11064 Changed TestPutSFTP run iterations based on files queued
7db6f17350 is described below

commit 7db6f17350cb3911e435893465cedbc656cf956c
Author: exceptionfactory <ex...@apache.org>
AuthorDate: Thu Jan 19 09:47:22 2023 -0600

    NIFI-11064 Changed TestPutSFTP run iterations based on files queued
    
    Signed-off-by: Pierre Villard <pi...@gmail.com>
    
    This closes #6863.
---
 .../test/java/org/apache/nifi/processors/standard/TestPutSFTP.java   | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/nifi-nar-bundles/nifi-standard-bundle/nifi-standard-processors/src/test/java/org/apache/nifi/processors/standard/TestPutSFTP.java b/nifi-nar-bundles/nifi-standard-bundle/nifi-standard-processors/src/test/java/org/apache/nifi/processors/standard/TestPutSFTP.java
index e10f42b851..32ce4d7ca3 100644
--- a/nifi-nar-bundles/nifi-standard-bundle/nifi-standard-processors/src/test/java/org/apache/nifi/processors/standard/TestPutSFTP.java
+++ b/nifi-nar-bundles/nifi-standard-bundle/nifi-standard-processors/src/test/java/org/apache/nifi/processors/standard/TestPutSFTP.java
@@ -222,9 +222,10 @@ class TestPutSFTP {
         secondAttributes.put(TRANSFER_HOST_ATTRIBUTE, LOCALHOST_ADDRESS);
         runner.enqueue(FLOW_FILE_CONTENTS, secondAttributes);
 
-        runner.run();
+        final int flowFilesQueued = runner.getQueueSize().getObjectCount();
+        runner.run(flowFilesQueued);
 
-        runner.assertTransferCount(PutSFTP.REL_SUCCESS, 2);
+        runner.assertTransferCount(PutSFTP.REL_SUCCESS, flowFilesQueued);
 
         final List<ProvenanceEventRecord> records = runner.getProvenanceEvents();