From 0bbe0aed83e8b9efa28250625749f87844a3ac33 Mon Sep 17 00:00:00 2001 From: Dan Schaper Date: Tue, 24 Jan 2017 20:09:10 -0800 Subject: [PATCH] Last of the IPTables tests, new test rig is operating. --- test/test_automated_install.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/test/test_automated_install.py b/test/test_automated_install.py index 14b2add2..211364ed 100644 --- a/test/test_automated_install.py +++ b/test/test_automated_install.py @@ -163,8 +163,9 @@ def test_configureFirewall_IPTables_enabled_not_exist_no_errors(Pihole): expected_stdout = 'Installing new IPTables firewall rulesets' assert expected_stdout in configureFirewall.stdout firewall_calls = Pihole.run('cat /var/log/iptables').stdout - # Only check the first rule, since iptables returns a 1 from the mock command. assert 'iptables -I INPUT 1 -p tcp -m tcp --dport 80 -j ACCEPT' in firewall_calls + assert 'iptables -I INPUT 1 -p tcp -m tcp --dport 53 -j ACCEPT' in firewall_calls + assert 'iptables -I INPUT 1 -p udp -m udp --dport 53 -j ACCEPT' in firewall_calls # Helper functions def mock_command(script, args, container):