From 605c37504d7b6b8678e61c78fc3caa7f50d8ee50 Mon Sep 17 00:00:00 2001 From: Salomon Popp Date: Tue, 16 Jan 2024 11:29:12 +0100 Subject: [PATCH] Refactor test --- tests/components/test_kafka_sink_connector.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/tests/components/test_kafka_sink_connector.py b/tests/components/test_kafka_sink_connector.py index 07a104db5..ef4f7caa3 100644 --- a/tests/components/test_kafka_sink_connector.py +++ b/tests/components/test_kafka_sink_connector.py @@ -207,10 +207,7 @@ def test_reset_when_dry_run_is_false( mock_clean_connector = mocker.patch.object( connector.handlers.connector_handler, "clean_connector" ) - resetter = connector._resetter - assert isinstance(resetter, KafkaConnectorResetter) - - mock_resetter_reset = mocker.spy(resetter, "reset") + mock_resetter_reset = mocker.spy(connector._resetter, "reset") mock = mocker.MagicMock() mock.attach_mock(mock_clean_connector, "mock_clean_connector")