From cbbf0030cabc4bd47cdde998c5c47c4a3f2d7e31 Mon Sep 17 00:00:00 2001 From: Shahar Epstein Date: Sat, 8 Jun 2024 23:58:54 +0300 Subject: [PATCH] Revert deletion of `test_delete_topic` --- tests/providers/apache/kafka/hooks/test_client.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/tests/providers/apache/kafka/hooks/test_client.py b/tests/providers/apache/kafka/hooks/test_client.py index f131db10d6252..0d92287969996 100644 --- a/tests/providers/apache/kafka/hooks/test_client.py +++ b/tests/providers/apache/kafka/hooks/test_client.py @@ -98,3 +98,12 @@ def test_create_topic_warning(self, admin_client, caplog): self.hook.create_topic(topics=[("topic_name", 0, 1)]) assert "The topic topic_name already exists" in caplog.text + @patch( + "airflow.providers.apache.kafka.hooks.client.AdminClient", + ) + def test_delete_topic(self, admin_client): + mock_f = MagicMock() + admin_client.return_value.delete_topics.return_value = {"topic_name": mock_f} + self.hook.delete_topic(topics=["topic_name"]) + admin_client.return_value.delete_topics.assert_called_with(["topic_name"]) + mock_f.result.assert_called_once()