Update GetCQLPerformance.java

Author: george0st

nifi/cql-processor/nifi-cql-processors/src/test/java/org/george0st/processors/cql/GetCQLPerformance.java

@@ -92,4 +92,28 @@ public void testRead100K() throws Exception {
         }
     }
 
+    @Test
+    public void testRead150K() throws Exception {
+        MockFlowFile result;
+
+        // Prepare data for performance test
+        PutCQLPerformance putCQL = new PutCQLPerformance();
+        putCQL.init();
+        putCQL.csvRandomWrite150K();
+
+        //  sleep before read (2 seconds)
+        Thread.sleep(2000);
+
+        // Read data
+        for (TestSetupRead setup : setups) {
+            setup.columnNames="colbigint, colint";
+
+            result = runTest(setup);
+
+            //  check amount of read items
+            assertNotNull(result, String.format("Issue with processing in '%s'", setup.name));
+            assertEquals(150_000, Long.parseLong(result.getAttribute(CQLAttributes.READ_COUNT)));
+        }
+    }
+
 }