def test_set_absent(self):
'''
Test to verify the set is absent.
'''
name = 'salt'
ret = {'name': name,
'result': True,
'comment': '',
'changes': {}}
mock = MagicMock(side_effect=[False, True, True, ''])
with patch.dict(ipset.__salt__, {'ipset.check_set': mock,
'ipset.flush': mock}):
comt = ('ipset set salt is already absent for family ipv4')
ret.update({'comment': comt})
self.assertDictEqual(ipset.set_absent(name), ret)
with patch.dict(ipset.__opts__, {'test': True}):
comt = ('ipset set salt needs to be removed family ipv4')
ret.update({'comment': comt, 'result': None})
self.assertDictEqual(ipset.set_absent(name), ret)
with patch.dict(ipset.__opts__, {'test': False}):
comt = ('Failed to flush salt set: for ipv4')
ret.update({'comment': comt, 'result': False, 'changes': {}})
self.assertDictEqual(ipset.set_absent(name), ret)
def _runner(
self,
expected_ret,
test=False,
check_set=True,
delete_set="",
flush_assertion=False,
delete_set_assertion=False,
):
mock_check_set = MagicMock(return_value=check_set)
mock_flush = MagicMock()
mock_delete_set = (MagicMock() if delete_set is None else MagicMock(
return_value=delete_set))
with patch.dict(ipset.__opts__, {"test": test}):
with patch.dict(
ipset.__salt__,
{
"ipset.check_set": mock_check_set,
"ipset.flush": mock_flush,
"ipset.delete_set": mock_delete_set,
},
):
actual_ret = ipset.set_absent(self.fake_name)
mock_check_set.assert_called_once_with(self.fake_name, "ipv4")
if flush_assertion:
mock_flush.assert_called_once_with(self.fake_name, "ipv4")
else:
self.assertTrue(mock_flush.call_count == 0)
if delete_set_assertion:
mock_delete_set.assert_called_once_with(self.fake_name, "ipv4")
else:
self.assertTrue(mock_delete_set.call_count == 0)
self.assertDictEqual(actual_ret, expected_ret)
def _runner(self, expected_ret, test=False, check_set=True, delete_set='',
flush_assertion=False, delete_set_assertion=False):
mock_check_set = MagicMock(return_value=check_set)
mock_flush = MagicMock()
mock_delete_set = MagicMock() if delete_set is None else MagicMock(return_value=delete_set)
with patch.dict(ipset.__opts__, {'test': test}):
with patch.dict(ipset.__salt__, {'ipset.check_set': mock_check_set,
'ipset.flush': mock_flush,
'ipset.delete_set': mock_delete_set}):
actual_ret = ipset.set_absent(self.fake_name)
mock_check_set.assert_called_once_with(self.fake_name, 'ipv4')
if flush_assertion:
mock_flush.assert_called_once_with(self.fake_name, 'ipv4')
else:
mock_flush.assert_not_called()
if delete_set_assertion:
mock_delete_set.assert_called_once_with(self.fake_name, 'ipv4')
else:
mock_delete_set.assert_not_called()
self.assertDictEqual(actual_ret, expected_ret)
