def test_remove(self): ''' Test if it remove a package and all its dependencies which are not in use by other packages. ''' mock_pkg = MagicMock(side_effect=MinionError) with patch.dict(pkgutil.__salt__, {'pkg_resource.parse_targets': mock_pkg}): self.assertRaises(CommandExecutionError, pkgutil.remove) mock_ret = MagicMock(return_value=True) mock_run = MagicMock(return_value='A\t B\t SAME') mock_run_all = MagicMock(return_value={'stdout': ''}) mock_pkg = MagicMock(return_value=['']) with patch.dict(pkgutil.__salt__, {'pkg_resource.parse_targets': mock_pkg, 'pkg_resource.stringify': mock_pkg, 'pkg_resource.sort_pkglist': mock_pkg, 'cmd.run_all': mock_run_all, 'cmd.run': mock_run}): with patch.dict(pkgutil.__context__, {'pkg.list_pkgs': mock_ret}): self.assertDictEqual(pkgutil.remove(), {}) mock_pkg = MagicMock(return_value=[{"bar": "1.2.3"}]) with patch.dict(pkgutil.__salt__, {'pkg_resource.parse_targets': mock_pkg, 'pkg_resource.stringify': mock_pkg, 'pkg_resource.sort_pkglist': mock_pkg, 'cmd.run_all': mock_run_all, 'cmd.run': mock_run}): with patch.dict(pkgutil.__context__, {'pkg.list_pkgs': mock_ret}): with patch.object(pkgutil, 'list_pkgs', return_value={"bar": "1.2.3"}): self.assertDictEqual(pkgutil.remove(pkgs='["foo", "bar"]'), {})
def test_remove(self): """ Test if it remove a package and all its dependencies which are not in use by other packages. """ mock_pkg = MagicMock(side_effect=MinionError) with patch.dict(pkgutil.__salt__, {"pkg_resource.parse_targets": mock_pkg}): self.assertRaises(CommandExecutionError, pkgutil.remove) mock_ret = MagicMock(return_value=True) mock_run = MagicMock(return_value="A\t B\t SAME") mock_run_all = MagicMock(return_value={"stdout": ""}) mock_pkg = MagicMock(return_value=[""]) with patch.dict( pkgutil.__salt__, { "pkg_resource.parse_targets": mock_pkg, "pkg_resource.stringify": mock_pkg, "pkg_resource.sort_pkglist": mock_pkg, "cmd.run_all": mock_run_all, "cmd.run": mock_run, }, ): with patch.dict(pkgutil.__context__, {"pkg.list_pkgs": mock_ret}): self.assertDictEqual(pkgutil.remove(), {}) mock_pkg = MagicMock(return_value=[{"bar": "1.2.3"}]) with patch.dict( pkgutil.__salt__, { "pkg_resource.parse_targets": mock_pkg, "pkg_resource.stringify": mock_pkg, "pkg_resource.sort_pkglist": mock_pkg, "cmd.run_all": mock_run_all, "cmd.run": mock_run, }, ): with patch.dict(pkgutil.__context__, {"pkg.list_pkgs": mock_ret}): with patch.object(pkgutil, "list_pkgs", return_value={"bar": "1.2.3"}): self.assertDictEqual(pkgutil.remove(pkgs='["foo", "bar"]'), {})