def test_flush(self):
'''
Test to flush current ipset set.
'''
name = 'salt'
ret = {'name': name,
'result': False,
'comment': '',
'changes': {}}
mock = MagicMock(side_effect=[False, True, True, True])
mock_f = MagicMock(side_effect=[True, False])
with patch.dict(ipset.__salt__, {'ipset.check_set': mock,
'ipset.flush': mock_f}):
comt = ('ipset set {0} does not exist for ipv4'.format(name))
ret.update({'comment': comt})
self.assertDictEqual(ipset.flush(name), ret)
with patch.dict(ipset.__opts__, {'test': True}):
comt = ('ipset entries in set {0} family ipv4 needs to '
'be flushed'.format(name))
ret.update({'comment': comt, 'result': None})
self.assertDictEqual(ipset.flush(name), ret)
with patch.dict(ipset.__opts__, {'test': False}):
comt = ('Flush ipset entries in set {0} family ipv4'
.format(name))
ret.update({'comment': comt, 'result': True,
'changes': {'locale': 'salt'}})
self.assertDictEqual(ipset.flush(name), ret)
comt = ('Failed to flush ipset entries')
ret.update({'comment': comt, 'result': False, 'changes': {}})
self.assertDictEqual(ipset.flush(name), ret)
def _runner(self, expected_ret, test=False, check_set=True, flush=True,
flush_assertion=True):
mock_check_set = MagicMock(return_value=check_set)
mock_flush = MagicMock(return_value=flush)
with patch.dict(ipset.__opts__, {'test': test}):
with patch.dict(ipset.__salt__, {'ipset.check_set': mock_check_set,
'ipset.flush': mock_flush}):
actual_ret = ipset.flush(self.fake_name)
mock_check_set.assert_called_once_with(self.fake_name)
if flush_assertion:
mock_flush.assert_called_once_with(self.fake_name, 'ipv4')
else:
self.assertTrue(mock_flush.call_count == 0)
self.assertDictEqual(actual_ret, expected_ret)
def _runner(self, expected_ret, test=False, check_set=True, flush=True,
flush_assertion=True):
mock_check_set = MagicMock(return_value=check_set)
mock_flush = MagicMock(return_value=flush)
with patch.dict(ipset.__opts__, {'test': test}):
with patch.dict(ipset.__salt__, {'ipset.check_set': mock_check_set,
'ipset.flush': mock_flush}):
actual_ret = ipset.flush(self.fake_name)
mock_check_set.assert_called_once_with(self.fake_name)
if flush_assertion:
mock_flush.assert_called_once_with(self.fake_name, 'ipv4')
else:
mock_flush.assert_not_called()
self.assertDictEqual(actual_ret, expected_ret)