fix: integration test fixes

Author: dervoeti

tests/templates/kuttl/fault-tolerant-execution/04-copy-scripts.yaml

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

tests/templates/kuttl/fault-tolerant-execution/04_check-fte.py

@@ -27,39 +27,14 @@ def get_connection(coordinator):
         required=True,
         help="Trino Coordinator Host to connect to",
     )
-    all_args.add_argument(
-        "-w",
-        "--workers",
-        required=True,
-        help="Expected amount of workers to be present",
-    )
 
     args = vars(all_args.parse_args())
 
-    expected_workers = args["workers"]
     conn = get_connection(args["coordinator"])
 
     try:
         cursor = conn.cursor()
 
-        # Check that workers are active
-        cursor.execute(
-            "SELECT COUNT(*) as nodes FROM system.runtime.nodes WHERE coordinator=false AND state='active'"
-        )
-        (active_workers,) = cursor.fetchone()
-
-        if int(active_workers) != int(expected_workers):
-            print(
-                "Mismatch: [expected/active] workers ["
-                + str(expected_workers)
-                + "/"
-                + str(active_workers)
-                + "]"
-            )
-            exit(-1)
-
-        print(f"Active workers check passed: {active_workers}/{expected_workers}")
-
         # Test that TPCH connector is working
         cursor.execute("SELECT COUNT(*) FROM tpch.tiny.nation")
         result = cursor.fetchone()
@@ -90,6 +65,9 @@ def get_connection(coordinator):
 
         print("Complex query test passed")
 
+        cursor.close()
+        conn.close()
+
     except Exception as e:
         print(f"Test failed with error: {e}")
         import traceback

tests/templates/kuttl/fault-tolerant-execution/05-run-tests.yaml

@@ -2,11 +2,11 @@
 apiVersion: kuttl.dev/v1beta1
 kind: TestStep
 commands:
-  - script: kubectl exec -n $NAMESPACE trino-test-helper-0 -- python /tmp/04_check-fte.py -c trino-fte-coordinator -w 2
+  - script: kubectl exec -n $NAMESPACE trino-test-helper-0 -- python /tmp/check-active-workers.py -u admin -p "" -c trino-fte-coordinator -w 2
+  - script: kubectl exec -n $NAMESPACE trino-test-helper-0 -- python /tmp/04_check-fte.py -c trino-fte-coordinator
     timeout: 120
   # Verify that the exchange bucket contains data
   - script: |
-      sleep 10
       kubectl exec -n $NAMESPACE deployment/minio -- mc alias set local https://localhost:9000 minioAccessKey minioSecretKey --api S3v4
       count=$(kubectl exec -n $NAMESPACE deployment/minio -- mc stat --insecure local/exchange-bucket | awk '/Objects count:/ {print $3}')
       if [ "$count" -gt 0 ]; then