Skip to content

Commit 8a6a4af

Browse files
committed
create queue if necessary
1 parent 3f1d2c6 commit 8a6a4af

File tree

1 file changed

+11
-0
lines changed

1 file changed

+11
-0
lines changed

third_party/airflow/test/integration/test_airflow_operator_logic.py

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -57,6 +57,17 @@ def no_auth_client() -> ArmadaClient:
5757
return ArmadaClient(channel=grpc.insecure_channel(f"{server_name}:{server_port}"))
5858

5959

60+
# Ensure DEFAULT_QUEUE exists before any tests run
61+
@pytest.fixture(scope="session", autouse=True)
62+
def ensure_default_queue(client: ArmadaClient):
63+
try:
64+
client.get_queue(DEFAULT_QUEUE)
65+
except grpc.RpcError:
66+
# Create queue with default params if missing
67+
q = client.create_queue_request(name=DEFAULT_QUEUE, priority_factor=1.0)
68+
client.create_queue(q)
69+
70+
6071
def sleep_pod(image: str):
6172
pod = core_v1.PodSpec(
6273
containers=[

0 commit comments

Comments
 (0)