def test_new_table(self):
'''
Test if it create new custom table.
'''
self.assertEqual(nftables.new_table(table=None),
'Error: table needs to be specified')
mock = MagicMock(return_value='')
with patch.dict(nftables.__salt__, {'cmd.run': mock}):
self.assertEqual(nftables.new_table(table='nat'), True)
mock = MagicMock(return_value='table ip nat')
with patch.dict(nftables.__salt__, {'cmd.run': mock}):
self.assertEqual(nftables.new_table(table='nat'),
'Error: table nat in family ipv4 already exists')
def test_new_table(self):
'''
Test if it create new custom table.
'''
self.assertEqual(nftables.new_table(table=None),
'Error: table needs to be specified')
mock = MagicMock(return_value='')
with patch.dict(nftables.__salt__, {'cmd.run': mock}):
self.assertEqual(nftables.new_table(table='nat'), True)
mock = MagicMock(return_value='table ip nat')
with patch.dict(nftables.__salt__, {'cmd.run': mock}):
self.assertEqual(nftables.new_table(table='nat'),
'Error: table nat in family ipv4 already exists')
def test_new_table(self):
"""
Test if it create new custom table.
"""
self.assertEqual(
nftables.new_table(table=None),
{"result": False, "comment": "Table needs to be specified"},
)
mock = MagicMock(return_value="")
ret = {"comment": "Table nat in family ipv4 created", "result": True}
with patch.dict(nftables.__salt__, {"cmd.run": mock}):
self.assertEqual(nftables.new_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.new_table(table="nat"), ret)
def test_new_table(self):
'''
Test if it create new custom table.
'''
self.assertEqual(nftables.new_table(table=None), {
'result': False,
'comment': 'Table needs to be specified'
})
mock = MagicMock(return_value='')
ret = {'comment': 'Table nat in family ipv4 created', 'result': True}
with patch.dict(nftables.__salt__, {'cmd.run': mock}):
self.assertEqual(nftables.new_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.new_table(table='nat'), ret)
