From 64be4287c8ba73baf10a18182743747e302a6997 Mon Sep 17 00:00:00 2001 From: Wei Lee Date: Wed, 28 Aug 2024 10:57:17 +0800 Subject: [PATCH] test(providers/yandex): fix test_yandex_lockbox_secret_backend_get_connection_from_json by removing non-json extra --- tests/providers/yandex/secrets/test_lockbox.py | 18 ++++-------------- 1 file changed, 4 insertions(+), 14 deletions(-) diff --git a/tests/providers/yandex/secrets/test_lockbox.py b/tests/providers/yandex/secrets/test_lockbox.py index aad00d4212f0e..7b9ab19d3e87a 100644 --- a/tests/providers/yandex/secrets/test_lockbox.py +++ b/tests/providers/yandex/secrets/test_lockbox.py @@ -27,7 +27,6 @@ import yandex.cloud.lockbox.v1.secret_pb2 as secret_pb import yandex.cloud.lockbox.v1.secret_service_pb2 as secret_service_pb -from airflow.exceptions import RemovedInAirflow3Warning from airflow.providers.yandex.secrets.lockbox import LockboxSecretBackend from airflow.providers.yandex.utils.defaults import default_conn_name @@ -60,7 +59,7 @@ def test_yandex_lockbox_secret_backend_get_connection(self, mock_get_value): def test_yandex_lockbox_secret_backend_get_connection_from_json(self, mock_get_value): conn_id = "airflow_to_yandexcloud" conn_type = "yandex_cloud" - extra = "some extra values" + extra = '{"some": "extra values"}' c = { "conn_type": conn_type, "extra": extra, @@ -68,18 +67,9 @@ def test_yandex_lockbox_secret_backend_get_connection_from_json(self, mock_get_v mock_get_value.return_value = json.dumps(c) - match = "Encountered non-JSON in `extra` field for connection 'airflow_to_yandexcloud'. Support for non-JSON `extra` will be removed in Airflow 3.0" - with pytest.warns( - RemovedInAirflow3Warning, - match=match, - ): - conn = LockboxSecretBackend().get_connection(conn_id) - - with pytest.warns( - RemovedInAirflow3Warning, - match=match, - ): - assert conn.extra == extra + conn = LockboxSecretBackend().get_connection(conn_id) + + assert conn.extra == extra assert conn.conn_id == conn_id assert conn.conn_type == conn_type