def test_table_zero():
"""
Tests clearing all the statistics of a table.
"""
ret = {}
ret["stderr"] = "42 addresses has been cleared"
ret["retcode"] = 0
mock_cmd = MagicMock(return_value=ret)
with patch.dict(pf.__salt__, {"cmd.run_all": mock_cmd}):
assert pf.table("zero", table="bad_hosts")["changes"]
def test_table_no_changes():
"""
Tests a table command that has no changes.
"""
ret = {}
ret["stderr"] = "0/1 addresses expired."
ret["retcode"] = 0
mock_cmd = MagicMock(return_value=ret)
with patch.dict(pf.__salt__, {"cmd.run_all": mock_cmd}):
assert not pf.table("expire", table="bad_hosts", number=300)["changes"]
def test_table(self):
'''
Tests a regular table command.
'''
ret = {}
ret['stderr'] = '42 addresses deleted'
ret['retcode'] = 0
mock_cmd = MagicMock(return_value=ret)
with patch.dict(pf.__salt__, {'cmd.run_all': mock_cmd}):
self.assertTrue(pf.table('flush', table='bad_hosts')['changes'])
def test_table():
"""
Tests a regular table command.
"""
ret = {}
ret["stderr"] = "42 addresses deleted"
ret["retcode"] = 0
mock_cmd = MagicMock(return_value=ret)
with patch.dict(pf.__salt__, {"cmd.run_all": mock_cmd}):
assert pf.table("flush", table="bad_hosts")["changes"]
def test_table_show():
"""
Tests showing table contents.
"""
ret = {}
ret["stdout"] = "1.2.3.4\n5.6.7.8"
ret["retcode"] = 0
expected = ["1.2.3.4", "5.6.7.8"]
mock_cmd = MagicMock(return_value=ret)
with patch.dict(pf.__salt__, {"cmd.run_all": mock_cmd}):
assert pf.table("show", table="bad_hosts")["comment"] == expected
def test_table_test_address():
"""
Tests testing addresses in a table.
"""
ret = {}
ret["stderr"] = "1/2 addressess match."
ret["retcode"] = 0
mock_cmd = MagicMock(return_value=ret)
with patch.dict(pf.__salt__, {"cmd.run_all": mock_cmd}):
assert pf.table("test", table="bad_hosts",
addresses=["1.2.3.4"])["matches"]
def test_table_no_changes(self):
'''
Tests a table command that has no changes.
'''
ret = {}
ret['stderr'] = '0/1 addresses expired.'
ret['retcode'] = 0
mock_cmd = MagicMock(return_value=ret)
with patch.dict(pf.__salt__, {'cmd.run_all': mock_cmd}):
self.assertFalse(
pf.table('expire', table='bad_hosts', number=300)['changes'])
def test_table_expire(self):
'''
Tests the table expire command.
'''
ret = {}
ret['stderr'] = '1/1 addresses expired.'
ret['retcode'] = 0
mock_cmd = MagicMock(return_value=ret)
with patch.dict(pf.__salt__, {'cmd.run_all': mock_cmd}):
self.assertTrue(
pf.table('expire', table='bad_hosts', number=300)['changes'])
def test_table_expire(self):
"""
Tests the table expire command.
"""
ret = {}
ret["stderr"] = "1/1 addresses expired."
ret["retcode"] = 0
mock_cmd = MagicMock(return_value=ret)
with patch.dict(pf.__salt__, {"cmd.run_all": mock_cmd}):
self.assertTrue(
pf.table("expire", table="bad_hosts", number=300)["changes"])
def test_table_delete_addresses():
"""
Tests deleting addresses in a table.
"""
ret = {}
ret["stderr"] = "2/2 addressess deleted."
ret["retcode"] = 0
mock_cmd = MagicMock(return_value=ret)
with patch.dict(pf.__salt__, {"cmd.run_all": mock_cmd}):
assert pf.table("delete",
table="bad_hosts",
addresses=["1.2.3.4", "5.6.7.8"])["changes"]
def test_table_show(self):
'''
Tests showing table contents.
'''
ret = {}
ret['stdout'] = '1.2.3.4\n5.6.7.8'
ret['retcode'] = 0
expected = ['1.2.3.4', '5.6.7.8']
mock_cmd = MagicMock(return_value=ret)
with patch.dict(pf.__salt__, {'cmd.run_all': mock_cmd}):
self.assertListEqual(
pf.table('show', table='bad_hosts')['comment'], expected)
def test_table_test_address(self):
'''
Tests testing addresses in a table.
'''
ret = {}
ret['stderr'] = '1/2 addressess match.'
ret['retcode'] = 0
mock_cmd = MagicMock(return_value=ret)
with patch.dict(pf.__salt__, {'cmd.run_all': mock_cmd}):
self.assertTrue(
pf.table('test', table='bad_hosts',
addresses=['1.2.3.4'])['matches'])
def test_table_add_addresses(self):
'''
Tests adding addresses to a table.
'''
ret = {}
ret['stderr'] = '2/2 addressess added.'
ret['retcode'] = 0
mock_cmd = MagicMock(return_value=ret)
with patch.dict(pf.__salt__, {'cmd.run_all': mock_cmd}):
self.assertTrue(
pf.table('add',
table='bad_hosts',
addresses=['1.2.3.4', '5.6.7.8'])['changes'])
def test_table_add_addresses(self):
"""
Tests adding addresses to a table.
"""
ret = {}
ret["stderr"] = "2/2 addressess added."
ret["retcode"] = 0
mock_cmd = MagicMock(return_value=ret)
with patch.dict(pf.__salt__, {"cmd.run_all": mock_cmd}):
self.assertTrue(
pf.table("add",
table="bad_hosts",
addresses=["1.2.3.4", "5.6.7.8"])["changes"])
