extern/sshuttle
1
1
Fork 0
mirror of https://github.com/sshuttle/sshuttle.git synced 2025-05-08 18:14:28 +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

9
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,11 +62,9 @@ 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')
) as (mock_setup_daemon, mock_get_method):
stdin, stdout = setup_daemon() stdin, stdout = setup_daemon()
mock_setup_daemon.return_value = stdin, stdout mock_setup_daemon.return_value = stdin, stdout