def test_post_invoke(self, mock_user_pkgs): mock_user_pkgs.return_value = ['vim', 'gedit'] apt_run = AptRun() apt_run.set_folder(AptRunTests.TEST_FOLDER) apt_run.enable() with open(apt_run.installed_pkgs_file, 'w') as text: text.write('gedit') installed_pkgs = apt_run.post_invoke() self.assertEqual(['vim'], installed_pkgs)