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