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()
