extern/sshuttle
1
1
Fork 0
mirror of https://github.com/sshuttle/sshuttle.git synced 2025-05-09 10:34:33 +02:00
Code Issues Packages Projects Releases Wiki Activity

Don't use nested.

Browse Source 
Is Python 2.x only. Not supported under Python 3.x.
This commit is contained in:
Brian May 2015-11-17 09:32:40 +11:00
parent cf0aaa7134
commit 9d443e4155
1 changed files with 32 additions and 35 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

67
sshuttle/tests/test_firewall.py
View File

@ -1,5 +1,4 @@
from mock import Mock, patch, call from mock import Mock, patch, call
from contextlib import nested
import io import io
import os import os
import os.path import os.path
@ -63,39 +62,37 @@ def test_rewrite_etc_hosts():
assert filecmp.cmp("tmp/hosts.orig", "tmp/hosts", shallow=False) is True assert filecmp.cmp("tmp/hosts.orig", "tmp/hosts", shallow=False) is True
def test_main(): @patch('sshuttle.firewall.setup_daemon')
with nested( @patch('sshuttle.firewall.get_method')
patch('sshuttle.firewall.setup_daemon'), def test_main(mock_get_method, mock_setup_daemon):
patch('sshuttle.firewall.get_method') stdin, stdout = setup_daemon()
) as (mock_setup_daemon, mock_get_method): mock_setup_daemon.return_value = stdin, stdout
stdin, stdout = setup_daemon()
mock_setup_daemon.return_value = stdin, stdout
sshuttle.firewall.main("test", False) sshuttle.firewall.main("test", False)
stdout.mock_calls == [ stdout.mock_calls == [
call.write('READY test\n'), call.write('READY test\n'),
call.flush(), call.flush(),
call.write('STARTED\n'), call.write('STARTED\n'),
call.flush() call.flush()
] ]
mock_setup_daemon.mock_calls == [call()] mock_setup_daemon.mock_calls == [call()]
mock_get_method.mock_calls == [ mock_get_method.mock_calls == [
call('test'), call('test'),
call().setup_firewall( call().setup_firewall(
1024, 1026, 1024, 1026,
[(10, u'2404:6800:4004:80c::33')], [(10, u'2404:6800:4004:80c::33')],
10, 10,
[(10, 64, False, u'2404:6800:4004:80c::'), [(10, 64, False, u'2404:6800:4004:80c::'),
(10, 128, True, u'2404:6800:4004:80c::101f')], (10, 128, True, u'2404:6800:4004:80c::101f')],
True), True),
call().setup_firewall( call().setup_firewall(
1025, 1027, 1025, 1027,
[(2, u'1.2.3.33')], [(2, u'1.2.3.33')],
2, 2,
[(2, 24, False, u'1.2.3.0'), (2, 32, True, u'1.2.3.66')], [(2, 24, False, u'1.2.3.0'), (2, 32, True, u'1.2.3.66')],
True), True),
call().setup_firewall()(), call().setup_firewall()(),
call().setup_firewall(1024, 0, [], 10, [], True), call().setup_firewall(1024, 0, [], 10, [], True),
call().setup_firewall(1025, 0, [], 2, [], True), call().setup_firewall(1025, 0, [], 2, [], True),
] ]