def test_delete_table(self):
'''
Test if it delete custom table.
'''
self.assertEqual(nftables.delete_table(table=None),
{'result': False,
'comment': 'Table needs to be specified'})
mock_ret = {'result': False,
'comment': 'Table nat in family ipv4 does not exist'}
with patch('salt.modules.nftables.check_table',
MagicMock(return_value=mock_ret)):
ret = nftables.delete_table(table='nat')
self.assertEqual(ret,
{'result': False,
'comment': 'Table nat in family ipv4 does not exist'})
mock = MagicMock(return_value='table ip nat')
with patch.dict(nftables.__salt__, {'cmd.run': mock}), \
patch('salt.modules.nftables.check_table',
MagicMock(return_value={'result': True,
'comment': ''})):
self.assertEqual(nftables.delete_table(table='nat'),
{'comment': 'Table nat in family ipv4 could not be deleted',
'result': False})
mock = MagicMock(return_value='')
with patch.dict(nftables.__salt__, {'cmd.run': mock}), \
patch('salt.modules.nftables.check_table',
MagicMock(return_value={'result': True,
'comment': ''})):
self.assertEqual(nftables.delete_table(table='nat'),
{'comment': 'Table nat in family ipv4 deleted',
'result': True})
def test_delete_table(self):
"""
Test if it delete custom table.
"""
self.assertEqual(
nftables.delete_table(table=None),
{
"result": False,
"comment": "Table needs to be specified"
},
)
mock_ret = {
"result": False,
"comment": "Table nat in family ipv4 does not exist",
}
with patch("salt.modules.nftables.check_table",
MagicMock(return_value=mock_ret)):
ret = nftables.delete_table(table="nat")
self.assertEqual(
ret,
{
"result": False,
"comment": "Table nat in family ipv4 does not exist"
},
)
mock = MagicMock(return_value="table ip nat")
with patch.dict(nftables.__salt__, {"cmd.run": mock}), patch(
"salt.modules.nftables.check_table",
MagicMock(return_value={
"result": True,
"comment": ""
}),
):
self.assertEqual(
nftables.delete_table(table="nat"),
{
"comment": "Table nat in family ipv4 could not be deleted",
"result": False,
},
)
mock = MagicMock(return_value="")
with patch.dict(nftables.__salt__, {"cmd.run": mock}), patch(
"salt.modules.nftables.check_table",
MagicMock(return_value={
"result": True,
"comment": ""
}),
):
self.assertEqual(
nftables.delete_table(table="nat"),
{
"comment": "Table nat in family ipv4 deleted",
"result": True
},
)
def test_delete_table(self):
'''
Test if it delete custom table.
'''
self.assertEqual(nftables.delete_table(table=None),
'Error: table needs to be specified')
mock = MagicMock(return_value='')
with patch.dict(nftables.__salt__, {'cmd.run': mock}):
self.assertEqual(nftables.delete_table(table='nat'),
'Error: table nat in family ipv4 does not exist')
mock = MagicMock(return_value='table nat')
with patch.dict(nftables.__salt__, {'cmd.run': mock}):
self.assertEqual(nftables.delete_table(table='nat'), 'table nat')
def test_delete_table(self):
'''
Test if it delete custom table.
'''
self.assertEqual(nftables.delete_table(table=None),
'Error: table needs to be specified')
mock = MagicMock(return_value='')
with patch.dict(nftables.__salt__, {'cmd.run': mock}):
self.assertEqual(nftables.delete_table(table='nat'),
'Error: table nat in family ipv4 does not exist')
mock = MagicMock(return_value='table ip nat')
with patch.dict(nftables.__salt__, {'cmd.run': mock}):
self.assertEqual(nftables.delete_table(table='nat'), 'table ip nat')
