def test_list_sets(self): ''' Test for List all ipset sets. ''' with patch.object(ipset, '_ipset_cmd', return_value='A'): mock = MagicMock(return_value="A:a") with patch.dict(ipset.__salt__, {'cmd.run': mock}): self.assertEqual(ipset.list_sets(), [{'A': ''}])
def test_list_sets(): """ Test for List all ipset sets. """ with patch.object(ipset, "_ipset_cmd", return_value="A"): mock = MagicMock(return_value="A:a") with patch.dict(ipset.__salt__, {"cmd.run": mock}): assert ipset.list_sets() == [{"A": ""}]