def test_check_table(self):
'''
Test if it check for the existence of a table
'''
self.assertEqual(nftables.check_table(),
'Error: table needs to be specified')
mock = MagicMock(return_value='')
with patch.dict(nftables.__salt__, {'cmd.run': mock}):
self.assertFalse(nftables.check_table(table='nat'))
mock = MagicMock(return_value='table ip nat')
with patch.dict(nftables.__salt__, {'cmd.run': mock}):
self.assertTrue(nftables.check_table(table='nat'))
def test_check_table(self):
'''
Test if it check for the existence of a table
'''
self.assertEqual(nftables.check_table(),
'Error: table needs to be specified')
mock = MagicMock(return_value='')
with patch.dict(nftables.__salt__, {'cmd.run': mock}):
self.assertFalse(nftables.check_table(table='nat'))
mock = MagicMock(return_value='table ip nat')
with patch.dict(nftables.__salt__, {'cmd.run': mock}):
self.assertTrue(nftables.check_table(table='nat'))
def test_check_table(self):
"""
Test if it check for the existence of a table
"""
self.assertEqual(
nftables.check_table(),
{"result": False, "comment": "Table needs to be specified"},
)
mock = MagicMock(return_value="")
ret = {"comment": "Table nat in family ipv4 does not exist", "result": False}
with patch.dict(nftables.__salt__, {"cmd.run": mock}):
self.assertEqual(nftables.check_table(table="nat"), ret)
mock = MagicMock(return_value="table ip nat")
ret = {"comment": "Table nat in family ipv4 exists", "result": True}
with patch.dict(nftables.__salt__, {"cmd.run": mock}):
self.assertEqual(nftables.check_table(table="nat"), ret)
def test_check_table(self):
'''
Test if it check for the existence of a table
'''
self.assertEqual(nftables.check_table(),
{'result': False,
'comment': 'Table needs to be specified'})
mock = MagicMock(return_value='')
ret = {'comment': 'Table nat in family ipv4 does not exist',
'result': False}
with patch.dict(nftables.__salt__, {'cmd.run': mock}):
self.assertEqual(nftables.check_table(table='nat'), ret)
mock = MagicMock(return_value='table ip nat')
ret = {'comment': 'Table nat in family ipv4 exists',
'result': True}
with patch.dict(nftables.__salt__, {'cmd.run': mock}):
self.assertEqual(nftables.check_table(table='nat'), ret)
