diff --git a/tests/gems/flatpak/test_flatpak.py b/tests/gems/flatpak/test_flatpak.py index 7e10df91..a65b16bd 100644 --- a/tests/gems/flatpak/test_flatpak.py +++ b/tests/gems/flatpak/test_flatpak.py @@ -7,6 +7,7 @@ from bauh.gems.flatpak import flatpak, VERSION_1_2 class FlatpakTest(TestCase): + @patch(f'{__app_name__}.gems.flatpak.flatpak.SimpleProcess') @patch(f'{__app_name__}.gems.flatpak.flatpak.ProcessHandler.handle_simple', return_value=(True, """ Looking for updates... @@ -14,8 +15,9 @@ class FlatpakTest(TestCase): 1.\t \torg.xpto.Xnote\tx86_64\tstable\tflathub\t< 4.3 MB """)) - def test_map_update_download_size__for_flatpak_1_2(self, handle_simple: Mock): + def test_map_update_download_size__for_flatpak_1_2(self, SimpleProcess: Mock, handle_simple: Mock): download_size = flatpak.map_update_download_size(app_ids={'org.xpto.Xnote'}, installation='user', version=VERSION_1_2) + SimpleProcess.assert_called_once() handle_simple.assert_called_once() self.assertEqual({'org.xpto.Xnote': 4300000}, download_size)