def test_restore_clean_yml(self, status, fopen, load, remove): ''' Test to restore an old state ''' status.return_value = True salt_mock = { 'pkg.list_pkgs': MagicMock(return_value={}), 'pkg.list_repos': MagicMock(return_value={}), 'pkg.install': MagicMock(), } with patch.dict(freezer.__salt__, salt_mock): self.assertEqual( freezer.restore(clean=True), { 'pkgs': { 'add': [], 'remove': [] }, 'repos': { 'add': [], 'remove': [] }, 'comment': [], }) salt_mock['pkg.list_pkgs'].assert_called() salt_mock['pkg.list_repos'].assert_called() fopen.assert_called() load.assert_called() self.assertEqual(remove.call_count, 2)
def test_restore_remove_extra_repo(self, status, fopen, load): ''' Test to restore an old state ''' # Only an extra repository is removed status.return_value = True load.side_effect = ({}, {}) salt_mock = { 'pkg.list_pkgs': MagicMock(return_value={}), 'pkg.list_repos': MagicMock(return_value={'extra-repo': {}}), 'pkg.del_repo': MagicMock(), } with patch.dict(freezer.__salt__, salt_mock): self.assertEqual( freezer.restore(), { 'pkgs': { 'add': [], 'remove': [] }, 'repos': { 'add': [], 'remove': ['extra-repo'] }, 'comment': [], }) salt_mock['pkg.list_pkgs'].assert_called() salt_mock['pkg.list_repos'].assert_called() salt_mock['pkg.del_repo'].assert_called_once() fopen.assert_called() load.assert_called()
def test_restore_add_missing_package(self, status, fopen, load): ''' Test to restore an old state ''' # Only a missing package is installed status.return_value = True load.side_effect = ({'missing-package': {}}, {}) salt_mock = { 'pkg.list_pkgs': MagicMock(return_value={}), 'pkg.list_repos': MagicMock(return_value={}), 'pkg.install': MagicMock(), } with patch.dict(freezer.__salt__, salt_mock): self.assertEqual( freezer.restore(), { 'pkgs': { 'add': ['missing-package'], 'remove': [] }, 'repos': { 'add': [], 'remove': [] }, 'comment': [], }) salt_mock['pkg.list_pkgs'].assert_called() salt_mock['pkg.list_repos'].assert_called() salt_mock['pkg.install'].assert_called_once() fopen.assert_called() load.assert_called()
def test_restore_clean_yml(self, status, fopen, load, remove): """ Test to restore an old state """ status.return_value = True salt_mock = { "pkg.list_pkgs": MagicMock(return_value={}), "pkg.list_repos": MagicMock(return_value={}), "pkg.install": MagicMock(), } with patch.dict(freezer.__salt__, salt_mock): self.assertEqual( freezer.restore(clean=True), { "pkgs": { "add": [], "remove": [] }, "repos": { "add": [], "remove": [] }, "comment": [], }, ) salt_mock["pkg.list_pkgs"].assert_called() salt_mock["pkg.list_repos"].assert_called() fopen.assert_called() load.assert_called() self.assertEqual(remove.call_count, 2)
def test_restore_remove_extra_repo(self, status, fopen, load): """ Test to restore an old state """ # Only an extra repository is removed status.return_value = True load.side_effect = ({}, {}) salt_mock = { "pkg.list_pkgs": MagicMock(return_value={}), "pkg.list_repos": MagicMock(return_value={"extra-repo": {}}), "pkg.del_repo": MagicMock(), } with patch.dict(freezer.__salt__, salt_mock): self.assertEqual( freezer.restore(), { "pkgs": { "add": [], "remove": [] }, "repos": { "add": [], "remove": ["extra-repo"] }, "comment": [], }, ) salt_mock["pkg.list_pkgs"].assert_called() salt_mock["pkg.list_repos"].assert_called() salt_mock["pkg.del_repo"].assert_called_once() fopen.assert_called() load.assert_called()
def test_restore_add_missing_package(self, status, fopen, load): """ Test to restore an old state """ # Only a missing package is installed status.return_value = True load.side_effect = ({"missing-package": {}}, {}) salt_mock = { "pkg.list_pkgs": MagicMock(return_value={}), "pkg.list_repos": MagicMock(return_value={}), "pkg.install": MagicMock(), } with patch.dict(freezer.__salt__, salt_mock): self.assertEqual( freezer.restore(), { "pkgs": { "add": ["missing-package"], "remove": [] }, "repos": { "add": [], "remove": [] }, "comment": [], }, ) salt_mock["pkg.list_pkgs"].assert_called() salt_mock["pkg.list_repos"].assert_called() salt_mock["pkg.install"].assert_called_once() fopen.assert_called() load.assert_called()