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"])