def test_delete_set():
"""
Test for Delete ipset set.
"""
assert ipset.delete_set() == "Error: Set needs to be specified"
with patch.object(ipset, "_ipset_cmd", return_value="A"):
mock = MagicMock(return_value=True)
with patch.dict(ipset.__salt__, {"cmd.run": mock}):
assert ipset.delete_set("set", "family")
def test_delete_set(self):
'''
Test for Delete ipset set.
'''
self.assertEqual(ipset.delete_set(),
'Error: Set needs to be specified')
with patch.object(ipset, '_ipset_cmd', return_value='A'):
mock = MagicMock(return_value=True)
with patch.dict(ipset.__salt__, {'cmd.run': mock}):
self.assertTrue(ipset.delete_set('set', 'family'))
def test_delete_set(self):
'''
Test for Delete ipset set.
'''
self.assertEqual(ipset.delete_set(),
'Error: Set needs to be specified')
with patch.object(ipset, '_ipset_cmd', return_value='A'):
mock = MagicMock(return_value=True)
with patch.dict(ipset.__salt__, {'cmd.run': mock}):
self.assertTrue(ipset.delete_set('set', 'family'))