def test_add_capability(self):
'''
Test installing a capability with DISM
'''
mock = MagicMock()
with patch.dict(dism.__salt__, {'cmd.run_all': mock}):
with patch.dict(dism.__grains__, {'osversion': 10}):
dism.add_capability("test")
mock.assert_called_once_with(
['DISM', '/Quiet', '/Online', '/Add-Capability',
'/CapabilityName:test', '/NoRestart'])
def test_add_capability(self):
'''
Test installing a capability with DISM
'''
mock = MagicMock()
with patch.dict(dism.__salt__, {'cmd.run_all': mock}):
with patch.dict(dism.__grains__, {'osversion': 10}):
dism.add_capability("test")
mock.assert_called_once_with([
'DISM', '/Quiet', '/Online', '/Add-Capability',
'/CapabilityName:test', '/NoRestart'
])
def test_add_capability_with_extras(self):
'''
Test installing a capability with DISM
'''
mock = MagicMock()
with patch.dict(dism.__salt__, {'cmd.run_all': mock}):
with patch.dict(dism.__grains__, {'osversion': 10}):
dism.add_capability("test", "life", True)
mock.assert_called_once_with([
dism.bin_dism, '/Quiet', '/Online', '/Add-Capability',
'/CapabilityName:test', '/Source:life', '/LimitAccess',
'/NoRestart'
])
def test_add_capability():
"""
Test installing a capability with DISM
"""
mock = MagicMock()
with patch.dict(dism.__salt__, {"cmd.run_all": mock}):
with patch.dict(dism.__grains__, {"osversion": 10}):
dism.add_capability("test")
mock.assert_called_once_with([
dism.bin_dism,
"/Quiet",
"/Online",
"/Add-Capability",
"/CapabilityName:test",
"/NoRestart",
])