From 332521ce32ff411636260680da7bd625df3b1e2b Mon Sep 17 00:00:00 2001 From: Jim Baldwin <14864200+jimwbaldwin@users.noreply.github.com> Date: Tue, 5 Nov 2024 09:44:28 +1100 Subject: [PATCH] Add serialization test for glue trigger --- .../tests/amazon/aws/triggers/test_glue.py | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) diff --git a/providers/tests/amazon/aws/triggers/test_glue.py b/providers/tests/amazon/aws/triggers/test_glue.py index e39e38d8b76099..f42e08108b6c97 100644 --- a/providers/tests/amazon/aws/triggers/test_glue.py +++ b/providers/tests/amazon/aws/triggers/test_glue.py @@ -82,6 +82,25 @@ async def test_wait_job_failed(self, get_state_mock: mock.MagicMock): assert get_state_mock.call_count == 3 + def test_serialization(self): + trigger = GlueJobCompleteTrigger( + job_name="job_name", + run_id="JobRunId", + verbose=False, + aws_conn_id="aws_conn_id", + job_poll_interval=0.1, + ) + classpath, kwargs = trigger.serialize() + assert classpath == "airflow.providers.amazon.aws.triggers.glue.GlueJobCompleteTrigger" + assert bool(kwargs["verbose"]) == False + assert kwargs == { + "job_name": "job_name", + "run_id": "JobRunId", + "verbose": False, + "aws_conn_id": "aws_conn_id", + "job_poll_interval": 0.1, + } + class TestGlueCatalogPartitionSensorTrigger: @pytest.mark.asyncio