NIFI-11064 Changed TestPutSFTP run iterations based on files queued

Signed-off-by: Pierre Villard <pierre.villard.fr@gmail.com>

This closes #6863.
This commit is contained in:
exceptionfactory 2023-01-19 09:47:22 -06:00 committed by Pierre Villard
parent 3fafb8798f
commit 7db6f17350
No known key found for this signature in database
GPG Key ID: F92A93B30C07C6D5
1 changed files with 3 additions and 2 deletions

View File

@ -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();