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)