From ac8e3dae58339c91177d5b8565e28fd3d5d74a83 Mon Sep 17 00:00:00 2001
From: hj28 <hj28@fa22-cs527-097.cs.illinois.edu>
Date: Sun, 4 Dec 2022 09:31:00 -0600
Subject: [PATCH] NIFI-10637 Deterministic TestGenerateTableFetch
 testMultipleColumnTypeMissing

This closes #6754

Signed-off-by: David Handermann <exceptionfactory@apache.org>
---
 .../nifi/processors/standard/TestGenerateTableFetch.java      | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/nifi-nar-bundles/nifi-standard-bundle/nifi-standard-processors/src/test/java/org/apache/nifi/processors/standard/TestGenerateTableFetch.java b/nifi-nar-bundles/nifi-standard-bundle/nifi-standard-processors/src/test/java/org/apache/nifi/processors/standard/TestGenerateTableFetch.java
index 11a209997f..53dc086bbb 100644
--- a/nifi-nar-bundles/nifi-standard-bundle/nifi-standard-processors/src/test/java/org/apache/nifi/processors/standard/TestGenerateTableFetch.java
+++ b/nifi-nar-bundles/nifi-standard-bundle/nifi-standard-processors/src/test/java/org/apache/nifi/processors/standard/TestGenerateTableFetch.java
@@ -1467,9 +1467,7 @@ public class TestGenerateTableFetch {
 
 
         // Remove one element from columnTypeMap to simulate it's re-cache partial state
-        Map.Entry<String, Integer> entry = processor.columnTypeMap.entrySet().iterator().next();
-        String key = entry.getKey();
-        processor.columnTypeMap.remove(key);
+        processor.columnTypeMap.remove("TEST_QUERY_DB_TABLE");
 
         // Insert new records
         stmt.execute("insert into TEST_QUERY_DB_TABLE (id, bucket) VALUES (2, 0)");