rename python script

Author: razvan

tests/templates/kuttl/client-spooling/04-copy-scripts.yaml

@@ -3,4 +3,4 @@ apiVersion: kuttl.dev/v1beta1
 kind: TestStep
 commands:
   - script: kubectl cp -n $NAMESPACE ../../../../templates/kuttl/commons/check-active-workers.py trino-test-helper:/tmp || true
-  - script: kubectl cp -n $NAMESPACE 04_check-fte.py trino-test-helper:/tmp/
+  - script: kubectl cp -n $NAMESPACE 04_query.py trino-test-helper:/tmp/

…es/kuttl/client-spooling/04_check-fte.py …plates/kuttl/client-spooling/04_query.pytests/templates/kuttl/client-spooling/04_check-fte.py renamed to tests/templates/kuttl/client-spooling/04_query.py tests/templates/kuttl/client-spooling/04_check-fte.py renamed to tests/templates/kuttl/client-spooling/04_query.py

@@ -38,6 +38,12 @@ def get_connection(coordinator):
     try:
         cursor = conn.cursor()
 
+        # The table tpch.sf100.customer has 15 million rows but Python consumes
+        # too much memory to retrieve all of them at once.
+        # Fetching them one by one is too slow, so we fetch only 10k rows which seems to be enough
+        # for Trino to use the spooling protocol.
+        # Fetching less rows is too risky IMO as Trino might decide to not use spooling.
+
         print("🚜 fetching many rows from Trino to trigger spooling...")
 
         cursor.execute("SELECT * FROM tpch.sf100.customer")
@@ -48,7 +54,10 @@ def get_connection(coordinator):
             cursor.fetchmany(1_000)
             customer_count += 1_000
             batch = batch - 1
-        # assert customer_count == 15_000_000, f"TPCH test failed: expected 15 mil customers, got {customer_count}"
+
+        assert customer_count == 10_000, (
+            f"💀 crap! expected 10_000 customers, got {customer_count}"
+        )
 
         print("🎉 major success!")
 

tests/templates/kuttl/client-spooling/05-run-tests.yaml

@@ -3,5 +3,5 @@ apiVersion: kuttl.dev/v1beta1
 kind: TestStep
 commands:
   - script: kubectl exec -n $NAMESPACE trino-test-helper -- pip install --no-cache-dir trino==0.336.0
-  - script: kubectl exec -n $NAMESPACE trino-test-helper -- python /tmp/04_check-fte.py -c trino-coordinator
+  - script: kubectl exec -n $NAMESPACE trino-test-helper -- python /tmp/04_query.py -c trino-coordinator
     timeout: 120