Refactor test

bakdata · Jan 16, 2024 · 605c375 · 605c375
1 parent 3e5d5b9
commit 605c375
Showing 1 changed file with 1 addition and 4 deletions.
diff --git 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")