def test_delete(self):
'''
Test to delete a rule to a chain
'''
ret = {'name': 'salt',
'changes': {},
'result': None,
'comment': ''}
mock = MagicMock(return_value=[])
with patch.object(nftables, '_STATE_INTERNAL_KEYWORDS', mock):
mock = MagicMock(return_value='a')
with patch.dict(nftables.__salt__, {'nftables.build_rule': mock}):
mock = MagicMock(side_effect=[False, True, True, True])
with patch.dict(nftables.__salt__, {'nftables.check': mock}):
ret.update({'comment': 'nftables rule for salt'
' already absent for ipv4 (a)',
'result': True})
self.assertDictEqual(nftables.delete('salt',
table='', chain=''),
ret)
with patch.dict(nftables.__opts__, {'test': True}):
ret.update({'result': None,
'comment': 'nftables rule for salt needs'
' to be deleted for ipv4 (a)'})
self.assertDictEqual(nftables.delete('salt',
table='',
chain=''), ret)
with patch.dict(nftables.__opts__, {'test': False}):
mock = MagicMock(side_effect=[True, False])
with patch.dict(nftables.__salt__,
{'nftables.delete': mock}):
ret.update({'result': True,
'changes': {'locale': 'salt'},
'comment': 'Delete nftables rule'
' for salt a'})
self.assertDictEqual(nftables.delete('salt',
table='',
chain='',
position=''),
ret)
ret.update({'result': False,
'changes': {},
'comment': 'Failed to delete nftables'
' rule for salt.\nAttempted rule was a'
})
self.assertDictEqual(nftables.delete('salt',
table='',
chain='',
position=''),
ret)
def test_delete(self):
'''
Test to delete a rule to a chain
'''
ret = {'name': 'salt',
'changes': {},
'result': None,
'comment': ''}
mock = MagicMock(return_value=[])
with patch.object(nftables, '_STATE_INTERNAL_KEYWORDS', mock):
mock = MagicMock(return_value='a')
with patch.dict(nftables.__salt__, {'nftables.build_rule': mock}):
mock = MagicMock(side_effect=[False, True, True, True])
with patch.dict(nftables.__salt__, {'nftables.check': mock}):
ret.update({'comment': 'nftables rule for salt'
' already absent for ipv4 (a)',
'result': True})
self.assertDictEqual(nftables.delete('salt',
table='', chain=''),
ret)
with patch.dict(nftables.__opts__, {'test': True}):
ret.update({'result': None,
'comment': 'nftables rule for salt needs'
' to be deleted for ipv4 (a)'})
self.assertDictEqual(nftables.delete('salt',
table='',
chain=''), ret)
with patch.dict(nftables.__opts__, {'test': False}):
mock = MagicMock(side_effect=[True, False])
with patch.dict(nftables.__salt__,
{'nftables.delete': mock}):
ret.update({'result': True,
'changes': {'locale': 'salt'},
'comment': 'Delete nftables rule'
' for salt a'})
self.assertDictEqual(nftables.delete('salt',
table='',
chain='',
position=''),
ret)
ret.update({'result': False,
'changes': {},
'comment': 'Failed to delete nftables'
' rule for salt.\nAttempted rule was a'
})
self.assertDictEqual(nftables.delete('salt',
table='',
chain='',
position=''),
ret)
def test_delete():
"""
Test to delete a rule to a chain
"""
ret = {"name": "salt", "changes": {}, "result": None, "comment": ""}
mock = MagicMock(return_value=[])
with patch.object(nftables, "_STATE_INTERNAL_KEYWORDS", mock):
mock = MagicMock(return_value={
"result": True,
"comment": "",
"rule": "a"
})
with patch.dict(nftables.__salt__, {"nftables.build_rule": mock}):
mock = MagicMock(side_effect=[
{
"result": False,
"comment": ""
},
{
"result": True,
"comment": ""
},
{
"result": True,
"comment": ""
},
{
"result": True,
"comment": ""
},
])
with patch.dict(nftables.__salt__, {"nftables.check": mock}):
ret.update({
"comment":
"nftables rule for salt already absent for ipv4 (a)",
"result": True,
})
assert nftables.delete("salt", table="", chain="") == ret
with patch.dict(nftables.__opts__, {"test": True}):
ret.update({
"result":
None,
"comment":
"nftables rule for salt needs to be deleted for ipv4 (a)",
})
assert nftables.delete("salt", table="", chain="") == ret
with patch.dict(nftables.__opts__, {"test": False}):
mock = MagicMock(side_effect=[
{
"result": True,
"comment": ""
},
{
"result": False,
"comment": ""
},
])
with patch.dict(nftables.__salt__,
{"nftables.delete": mock}):
ret.update({
"result":
True,
"changes": {
"locale": "salt"
},
"comment":
"Delete nftables rule for salt a",
})
assert (nftables.delete("salt",
table="",
chain="",
position="") == ret)
ret.update({
"result":
False,
"changes": {},
"comment":
"Failed to delete nftables rule for"
" salt.\nAttempted rule was a",
})
assert (nftables.delete("salt",
table="",
chain="",
position="") == ret)