diff --git a/tests/pipeline/test_clean.py b/tests/pipeline/test_clean.py index 1fbc7cab8..b42bce2ab 100644 --- a/tests/pipeline/test_clean.py +++ b/tests/pipeline/test_clean.py @@ -16,11 +16,10 @@ class TestClean: @pytest.fixture(autouse=True) def helm_mock(self, mocker: MockerFixture) -> MagicMock: - async_mock = AsyncMock() return mocker.patch( - "kpops.components.base_components.helm_app.Helm", - return_value=async_mock, - ).return_value + "kpops.component_handlers.helm_wrapper.helm.Helm", + return_value=AsyncMock(), + ) # TODO: test using public Pipeline API # @pytest.fixture() diff --git a/tests/pipeline/test_reset.py b/tests/pipeline/test_reset.py index 07c7e1868..163ea3b1b 100644 --- a/tests/pipeline/test_reset.py +++ b/tests/pipeline/test_reset.py @@ -15,11 +15,11 @@ @pytest.mark.usefixtures("mock_env", "load_yaml_file_clear_cache") class TestReset: @pytest.fixture(autouse=True) - def mock_helm(self, mocker: MockerFixture) -> MagicMock: + def helm_mock(self, mocker: MockerFixture) -> MagicMock: return mocker.patch( - "kpops.components.base_components.helm_app.Helm", + "kpops.component_handlers.helm_wrapper.helm.Helm", return_value=AsyncMock(), - ).return_value + ) def test_order(self, mocker: MockerFixture): producer_app_mock_reset = mocker.patch(