def test_chain_present(self):
"""
Test to verify the chain is exist.
"""
ret = {"name": "salt", "changes": {}, "result": True, "comment": ""}
mock = MagicMock(side_effect=[True, False, False, False])
with patch.dict(iptables.__salt__, {"iptables.check_chain": mock}):
ret.update(
{
"comment": (
"iptables salt chain is already exist in filter table for ipv4"
)
}
)
self.assertDictEqual(iptables.chain_present("salt"), ret)
with patch.dict(iptables.__opts__, {"test": True}):
ret.update(
{
"comment": (
"iptables salt chain in filter"
" table needs to be set for ipv4"
),
"result": None,
}
)
self.assertDictEqual(iptables.chain_present("salt"), ret)
with patch.dict(iptables.__opts__, {"test": False}):
mock = MagicMock(side_effect=[True, ""])
with patch.dict(iptables.__salt__, {"iptables.new_chain": mock}):
ret.update(
{
"result": True,
"comment": (
"iptables salt chain in filter"
" table create success for ipv4"
),
"changes": {"locale": "salt"},
}
)
self.assertDictEqual(iptables.chain_present("salt"), ret)
ret.update(
{
"changes": {},
"result": False,
"comment": (
"Failed to create salt chain in filter table: for ipv4"
),
}
)
self.assertDictEqual(iptables.chain_present("salt"), ret)
def test_chain_present(self):
'''
Test to verify the chain is exist.
'''
ret = {'name': 'salt', 'changes': {}, 'result': True, 'comment': ''}
mock = MagicMock(side_effect=[True, False, False, False])
with patch.dict(iptables.__salt__, {'iptables.check_chain': mock}):
ret.update({
'comment':
'iptables salt chain is already'
' exist in filter table for ipv4'
})
self.assertDictEqual(iptables.chain_present("salt"), ret)
with patch.dict(iptables.__opts__, {'test': True}):
ret.update({
'comment': 'iptables salt chain in filter'
' table needs to be set for ipv4',
'result': None
})
self.assertDictEqual(iptables.chain_present("salt"), ret)
with patch.dict(iptables.__opts__, {'test': False}):
mock = MagicMock(side_effect=[True, ''])
with patch.dict(iptables.__salt__,
{'iptables.new_chain': mock}):
ret.update({
'result': True,
'comment': 'iptables salt chain in filter'
' table create success for ipv4',
'changes': {
'locale': 'salt'
}
})
self.assertDictEqual(iptables.chain_present('salt'), ret)
ret.update({
'changes': {},
'result':
False,
'comment':
'Failed to create salt chain'
' in filter table: for ipv4'
})
self.assertDictEqual(iptables.chain_present('salt'), ret)
def test_chain_present(self):
'''
Test to verify the chain is exist.
'''
ret = {'name': 'salt',
'changes': {},
'result': True,
'comment': ''
}
mock = MagicMock(side_effect=[True, False, False, False])
with patch.dict(iptables.__salt__, {'iptables.check_chain': mock}):
ret.update({'comment': 'iptables salt chain is already'
' exist in filter table for ipv4'})
self.assertDictEqual(iptables.chain_present("salt"), ret)
with patch.dict(iptables.__opts__, {'test': True}):
ret.update({'comment': 'iptables salt chain in filter'
' table needs to be set for ipv4',
'result': None})
self.assertDictEqual(iptables.chain_present("salt"), ret)
with patch.dict(iptables.__opts__, {'test': False}):
mock = MagicMock(side_effect=[True, ''])
with patch.dict(iptables.__salt__,
{'iptables.new_chain': mock}):
ret.update({'result': True,
'comment': 'iptables salt chain in filter'
' table create success for ipv4',
'changes': {'locale': 'salt'}})
self.assertDictEqual(iptables.chain_present('salt'), ret)
ret.update({'changes': {}, 'result': False,
'comment': 'Failed to create salt chain'
' in filter table: for ipv4'})
self.assertDictEqual(iptables.chain_present('salt'), ret)
