fix: fix failing test for in_out_handler

Author: rustyconover

scripts/generate-types-tests.py

@@ -179,25 +179,23 @@ def custom_sorted(values: Sequence[Any]) -> list[Any]:
 # Require statement will ensure this test is run with this extension loaded
 require airport
 
-require-env AIRPORT_TEST_SERVER_AVAILABLE
-
 # Create the initial secret, the token value doesn't matter.
 statement ok
 CREATE SECRET airport_testing (
   type airport,
   auth_token uuid(),
-  scope 'grpc://localhost:50003/');
+  scope 'grpc+tls://airport-ci.query.farm/');
 
 # Reset the test server
 statement ok
-CALL airport_action('grpc://localhost:50003/', 'reset');
+CALL airport_action('grpc+tls://airport-ci.query.farm/', 'reset');
 
 # Create the initial database
 statement ok
-CALL airport_action('grpc://localhost:50003/', 'create_database', 'test1');
+CALL airport_action('grpc+tls://airport-ci.query.farm/', 'create_database', 'test1');
 
 statement ok
-ATTACH 'test1' (TYPE  AIRPORT, location 'grpc://localhost:50003/');
+ATTACH 'test1' (TYPE  AIRPORT, location 'grpc+tls://airport-ci.query.farm/');
 """,
             file=f,
         )
@@ -251,7 +249,7 @@ def custom_sorted(values: Sequence[Any]) -> list[Any]:
         print(
             """# Reset the test server
     statement ok
-    CALL airport_action('grpc://localhost:50003/', 'reset');
+    CALL airport_action('grpc+tls://airport-ci.query.farm/', 'reset');
     """,
             file=f,
         )

src/query_farm_airport_test_server/server.py

@@ -875,10 +875,13 @@ def in_out_handler(
                     if input_chunk is None:
                         break
 
-                    assert parameters.parameters
+                    assert parameters.parameters is not None
+                    parameter_value = parameters.parameters.column(0).to_pylist()[0]
+
+                    # Since input chunks could be different sizes, standardize it.
                     result = pa.RecordBatch.from_arrays(
                         [
-                            parameters.parameters.column(0),
+                            [parameter_value] * len(input_chunk),
                             input_chunk.column(0),
                         ],
                         schema=output_schema,