diff options
Diffstat (limited to 'tests/client/test_firewall.py')
| -rw-r--r-- | tests/client/test_firewall.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/tests/client/test_firewall.py b/tests/client/test_firewall.py index 7404328..6a3b4db 100644 --- a/tests/client/test_firewall.py +++ b/tests/client/test_firewall.py @@ -145,6 +145,7 @@ def test_main(mock_get_method, mock_setup_daemon, mock_rewrite_etc_hosts): (AF_INET6, 128, True, u'2404:6800:4004:80c::101f', 80, 80)], True, None, + None, '0x01'), call().setup_firewall( 1025, 1027, @@ -154,7 +155,8 @@ def test_main(mock_get_method, mock_setup_daemon, mock_rewrite_etc_hosts): (AF_INET, 32, True, u'1.2.3.66', 8080, 8080)], True, None, + None, '0x01'), - call().restore_firewall(1024, AF_INET6, True, None), - call().restore_firewall(1025, AF_INET, True, None), + call().restore_firewall(1024, AF_INET6, True, None, None), + call().restore_firewall(1025, AF_INET, True, None, None), ] |