def test_test(): """ Test for Test if an entry is in the specified set. """ assert ipset.test() == "Error: Set needs to be specified" assert ipset.test("s") == "Error: Entry needs to be specified" with patch.object(ipset, "_find_set_type", return_value=None): assert ipset.test("set", "entry") == "Error: Set set does not exist" with patch.object(ipset, "_find_set_type", return_value=True): mock = MagicMock(side_effect=[{"retcode": 1}, {"retcode": -1}]) with patch.dict(ipset.__salt__, {"cmd.run_all": mock}): assert not ipset.test("set", "entry") assert ipset.test("set", "entry")
def test_test(self): ''' Test for Test if an entry is in the specified set. ''' self.assertEqual(ipset.test(), 'Error: Set needs to be specified') self.assertEqual(ipset.test('s'), 'Error: Entry needs to be specified') with patch.object(ipset, '_find_set_type', return_value=None): self.assertEqual(ipset.test('set', 'entry'), 'Error: Set set does not exist') with patch.object(ipset, '_find_set_type', return_value=True): mock = MagicMock(side_effect=[{'retcode': 1}, {'retcode': -1}]) with patch.dict(ipset.__salt__, {'cmd.run_all': mock}): self.assertFalse(ipset.test('set', 'entry')) self.assertTrue(ipset.test('set', 'entry'))