def test_capability_removed_failure(self):
'''
Test removing a capability which fails with DISM
'''
expected = {
'comment': "Failed to remove Capa2: Failed",
'changes': {},
'name': 'Capa2',
'result': False
}
mock_removed = MagicMock(
side_effect=[['Capa1', 'Capa2'], ['Capa1', 'Capa2']])
mock_remove = MagicMock(return_value={
'retcode': 67,
'stdout': 'Failed'
})
with patch.dict(
dism.__salt__, {
'dism.installed_capabilities': mock_removed,
'dism.remove_capability': mock_remove
}):
with patch.dict(dism.__opts__, {'test': False}):
out = dism.capability_removed('Capa2')
mock_removed.assert_called_with()
mock_remove.assert_called_once_with('Capa2', None, False)
self.assertEqual(out, expected)
def test_capability_removed_removed(self):
'''
Test removing a capability already removed
'''
expected = {
'comment': "The capability Capa2 is already removed",
'changes': {},
'name': 'Capa2',
'result': True
}
mock_removed = MagicMock(return_value=["Capa1"])
mock_remove = MagicMock()
with patch.dict(
dism.__salt__, {
'dism.installed_capabilities': mock_removed,
'dism.add_capability': mock_remove
}):
out = dism.capability_removed('Capa2', 'somewhere', True)
mock_removed.assert_called_once_with()
assert not mock_remove.called
self.assertEqual(out, expected)
def test_capability_removed_removed():
"""
Test removing a capability already removed
"""
expected = {
"comment": "The capability Capa2 is already removed",
"changes": {},
"name": "Capa2",
"result": True,
}
mock_removed = MagicMock(return_value=["Capa1"])
mock_remove = MagicMock()
with patch.dict(
dism.__salt__,
{
"dism.installed_capabilities": mock_removed,
"dism.add_capability": mock_remove,
},
):
out = dism.capability_removed("Capa2", "somewhere", True)
mock_removed.assert_called_once_with()
assert not mock_remove.called
assert out == expected
def test_capability_removed(self):
'''
Test capability removed state
'''
expected = {
'comment': "Removed Capa2",
'changes': {
'capability': {
'old': 'Capa2'
},
'retcode': 0
},
'name': 'Capa2',
'result': True
}
mock_removed = MagicMock(side_effect=[['Capa1', 'Capa2'], ['Capa1']])
mock_remove = MagicMock(return_value={'retcode': 0})
with patch.dict(
dism.__salt__, {
'dism.installed_capabilities': mock_removed,
'dism.remove_capability': mock_remove
}):
with patch.dict(dism.__opts__, {'test': False}):
out = dism.capability_removed('Capa2')
mock_removed.assert_called_with()
mock_remove.assert_called_once_with('Capa2', None, False)
self.assertEqual(out, expected)
def test_capability_removed_failure():
"""
Test removing a capability which fails with DISM
"""
expected = {
"comment": "Failed to remove Capa2: Failed",
"changes": {},
"name": "Capa2",
"result": False,
}
mock_removed = MagicMock(
side_effect=[["Capa1", "Capa2"], ["Capa1", "Capa2"]])
mock_remove = MagicMock(return_value={"retcode": 67, "stdout": "Failed"})
with patch.dict(
dism.__salt__,
{
"dism.installed_capabilities": mock_removed,
"dism.remove_capability": mock_remove,
},
):
with patch.dict(dism.__opts__, {"test": False}):
out = dism.capability_removed("Capa2")
mock_removed.assert_called_with()
mock_remove.assert_called_once_with("Capa2", None, False)
assert out == expected
def test_capability_removed():
"""
Test capability removed state
"""
expected = {
"comment": "Removed Capa2",
"changes": {
"capability": {
"old": ["Capa2"]
},
"retcode": 0
},
"name": "Capa2",
"result": True,
}
mock_removed = MagicMock(side_effect=[["Capa1", "Capa2"], ["Capa1"]])
mock_remove = MagicMock(return_value={"retcode": 0})
with patch.dict(
dism.__salt__,
{
"dism.installed_capabilities": mock_removed,
"dism.remove_capability": mock_remove,
},
):
with patch.dict(dism.__opts__, {"test": False}):
out = dism.capability_removed("Capa2")
mock_removed.assert_called_with()
mock_remove.assert_called_once_with("Capa2", None, False)
assert out == expected
def test_capability_removed_failure(self):
'''
Test removing a capability which fails with DISM
'''
expected = {
'comment': "Failed to remove Capa2: Failed",
'changes': {},
'name': 'Capa2',
'result': False}
mock_removed = MagicMock(
side_effect=[['Capa1', 'Capa2'], ['Capa1', 'Capa2']])
mock_remove = MagicMock(
return_value={'retcode': 67, 'stdout': 'Failed'})
with patch.dict(
dism.__salt__, {'dism.installed_capabilities': mock_removed,
'dism.remove_capability': mock_remove}):
with patch.dict(dism.__opts__, {'test': False}):
out = dism.capability_removed('Capa2')
mock_removed.assert_called_with()
mock_remove.assert_called_once_with(
'Capa2', None, False)
self.assertEqual(out, expected)
def test_capability_removed(self):
'''
Test capability removed state
'''
expected = {
'comment': "Removed Capa2",
'changes': {'capability': {'old': 'Capa2'},
'retcode': 0},
'name': 'Capa2',
'result': True}
mock_removed = MagicMock(
side_effect=[['Capa1', 'Capa2'], ['Capa1']])
mock_remove = MagicMock(
return_value={'retcode': 0})
with patch.dict(
dism.__salt__, {'dism.installed_capabilities': mock_removed,
'dism.remove_capability': mock_remove}):
with patch.dict(dism.__opts__, {'test': False}):
out = dism.capability_removed('Capa2')
mock_removed.assert_called_with()
mock_remove.assert_called_once_with('Capa2', None, False)
self.assertEqual(out, expected)
def test_capability_removed_removed(self):
'''
Test removing a capability already removed
'''
expected = {
'comment': "The capability Capa2 is already removed",
'changes': {},
'name': 'Capa2',
'result': True}
mock_removed = MagicMock(
return_value=["Capa1"])
mock_remove = MagicMock()
with patch.dict(
dism.__salt__, {'dism.installed_capabilities': mock_removed,
'dism.add_capability': mock_remove}):
out = dism.capability_removed('Capa2', 'somewhere', True)
mock_removed.assert_called_once_with()
assert not mock_remove.called
self.assertEqual(out, expected)
