extern/sshuttle
1
1
Fork 0
mirror of https://github.com/sshuttle/sshuttle.git synced 2024-12-01 20:34:16 +01:00
Code Issues Packages Projects Releases Wiki Activity
7fc33c0020
sshuttle/tests/client/test_helpers.py

197 lines
6.1 KiB
Python
Raw Normal View History

Add more tests.
2015-11-17 03:08:12 +01:00
import io
import socket
Fix tests on Macos Swap hardcoded AF_INET(6) values for Python-provided values as they differ between Darwin and Linux (30 vs 10 for AF_INET6 for instance).
2017-09-24 14:11:26 +02:00
from socket import AF_INET, AF_INET6
Mock socket bind to avoid depending on local IPs being available in test box
2017-11-07 00:00:06 +01:00
import errno
Add more tests.
2015-11-17 03:08:12 +01:00
Fix/pep8 (#277) * re-organized imports according to pep8 * fixed all remaining pep8 issues * moved common config into setup.cfg, additionally test `tests` * removed --select=X -- the errors selected where by default not in flake8's --ignore list so effectively had no effect * update .travis.yml to reflect changes in tox.ini * make travis just use tox in order to avoid code duplaction * replace py.test with pytest * fixed .travis.yml * try different pypy toxenv * hopefully fixed testenv for pypy * added pypy basepython, removed unused python2.6 * install dev package before testing (fixes missing coverage) * fixed empty exception pass blocks with noqa * Added dummy log message on empty try-except-pass blocks to make dodacy happy :( * Replaced Exception with BaseException
2019-02-10 23:59:13 +01:00
from mock import patch, call
Add more tests.
2015-11-17 03:08:12 +01:00
import sshuttle.helpers
@patch('sshuttle.helpers.logprefix', new='prefix: ')
@patch('sshuttle.helpers.sys.stdout')
@patch('sshuttle.helpers.sys.stderr')
def test_log(mock_stderr, mock_stdout):
sshuttle.helpers.log("message")
Add another test.
2015-12-15 04:23:42 +01:00
sshuttle.helpers.log("abc")
Print PF rules used. Also support multiline debug output better.
2015-12-13 23:21:15 +01:00
sshuttle.helpers.log("message 1\n")
sshuttle.helpers.log("message 2\nline2\nline3\n")
sshuttle.helpers.log("message 3\nline2\nline3")
Add more tests.
2015-11-17 03:08:12 +01:00
assert mock_stdout.mock_calls == [
call.flush(),
Print PF rules used. Also support multiline debug output better.
2015-12-13 23:21:15 +01:00
call.flush(),
call.flush(),
call.flush(),
Add another test.
2015-12-15 04:23:42 +01:00
call.flush(),
Add more tests.
2015-11-17 03:08:12 +01:00
]
assert mock_stderr.mock_calls == [
Refactor debug, log and Fatal messages. This commit rewrites the log() function so that it will append a newline at the end of the message if none is present. It doesn't make sense to print a log message without a newline since the next log message (which will write a prefix) expects to be starting at the beginning of a line. Although it isn't strictly necessary, this commit also removes any newlines at the ends of messages. If I missed any, including the newline at the end of the message will continue to work as it did before. Previously, some calls were missing the newline at the end even though including it was necessary for subsequent messages to appear correctly. This code also cleans up some redundant prefixes. The log() method will prepend the prefix and the different processes should set their prefix as soon as they start. Multiline messages are still supported (although the prefix for the additional lines was changed to match the length of the prefix used for the first line).
2020-12-29 18:58:44 +01:00
call.write('prefix: message\n'),
Add more tests.
2015-11-17 03:08:12 +01:00
call.flush(),
Refactor debug, log and Fatal messages. This commit rewrites the log() function so that it will append a newline at the end of the message if none is present. It doesn't make sense to print a log message without a newline since the next log message (which will write a prefix) expects to be starting at the beginning of a line. Although it isn't strictly necessary, this commit also removes any newlines at the ends of messages. If I missed any, including the newline at the end of the message will continue to work as it did before. Previously, some calls were missing the newline at the end even though including it was necessary for subsequent messages to appear correctly. This code also cleans up some redundant prefixes. The log() method will prepend the prefix and the different processes should set their prefix as soon as they start. Multiline messages are still supported (although the prefix for the additional lines was changed to match the length of the prefix used for the first line).
2020-12-29 18:58:44 +01:00
call.write('prefix: abc\n'),
Add another test.
2015-12-15 04:23:42 +01:00
call.flush(),
Print PF rules used. Also support multiline debug output better.
2015-12-13 23:21:15 +01:00
call.write('prefix: message 1\n'),
call.flush(),
call.write('prefix: message 2\n'),
Refactor debug, log and Fatal messages. This commit rewrites the log() function so that it will append a newline at the end of the message if none is present. It doesn't make sense to print a log message without a newline since the next log message (which will write a prefix) expects to be starting at the beginning of a line. Although it isn't strictly necessary, this commit also removes any newlines at the ends of messages. If I missed any, including the newline at the end of the message will continue to work as it did before. Previously, some calls were missing the newline at the end even though including it was necessary for subsequent messages to appear correctly. This code also cleans up some redundant prefixes. The log() method will prepend the prefix and the different processes should set their prefix as soon as they start. Multiline messages are still supported (although the prefix for the additional lines was changed to match the length of the prefix used for the first line).
2020-12-29 18:58:44 +01:00
call.write(' line2\n'),
call.write(' line3\n'),
Print PF rules used. Also support multiline debug output better.
2015-12-13 23:21:15 +01:00
call.flush(),
call.write('prefix: message 3\n'),
Refactor debug, log and Fatal messages. This commit rewrites the log() function so that it will append a newline at the end of the message if none is present. It doesn't make sense to print a log message without a newline since the next log message (which will write a prefix) expects to be starting at the beginning of a line. Although it isn't strictly necessary, this commit also removes any newlines at the ends of messages. If I missed any, including the newline at the end of the message will continue to work as it did before. Previously, some calls were missing the newline at the end even though including it was necessary for subsequent messages to appear correctly. This code also cleans up some redundant prefixes. The log() method will prepend the prefix and the different processes should set their prefix as soon as they start. Multiline messages are still supported (although the prefix for the additional lines was changed to match the length of the prefix used for the first line).
2020-12-29 18:58:44 +01:00
call.write(' line2\n'),
call.write(' line3\n'),
Print PF rules used. Also support multiline debug output better.
2015-12-13 23:21:15 +01:00
call.flush(),
Add more tests.
2015-11-17 03:08:12 +01:00
]
@patch('sshuttle.helpers.logprefix', new='prefix: ')
@patch('sshuttle.helpers.verbose', new=1)
@patch('sshuttle.helpers.sys.stdout')
@patch('sshuttle.helpers.sys.stderr')
def test_debug1(mock_stderr, mock_stdout):
sshuttle.helpers.debug1("message")
assert mock_stdout.mock_calls == [
call.flush(),
]
assert mock_stderr.mock_calls == [
Refactor debug, log and Fatal messages. This commit rewrites the log() function so that it will append a newline at the end of the message if none is present. It doesn't make sense to print a log message without a newline since the next log message (which will write a prefix) expects to be starting at the beginning of a line. Although it isn't strictly necessary, this commit also removes any newlines at the ends of messages. If I missed any, including the newline at the end of the message will continue to work as it did before. Previously, some calls were missing the newline at the end even though including it was necessary for subsequent messages to appear correctly. This code also cleans up some redundant prefixes. The log() method will prepend the prefix and the different processes should set their prefix as soon as they start. Multiline messages are still supported (although the prefix for the additional lines was changed to match the length of the prefix used for the first line).
2020-12-29 18:58:44 +01:00
call.write('prefix: message\n'),
Add more tests.
2015-11-17 03:08:12 +01:00
call.flush(),
]
@patch('sshuttle.helpers.logprefix', new='prefix: ')
@patch('sshuttle.helpers.verbose', new=0)
@patch('sshuttle.helpers.sys.stdout')
@patch('sshuttle.helpers.sys.stderr')
def test_debug1_nop(mock_stderr, mock_stdout):
sshuttle.helpers.debug1("message")
assert mock_stdout.mock_calls == []
assert mock_stderr.mock_calls == []
@patch('sshuttle.helpers.logprefix', new='prefix: ')
@patch('sshuttle.helpers.verbose', new=2)
@patch('sshuttle.helpers.sys.stdout')
@patch('sshuttle.helpers.sys.stderr')
def test_debug2(mock_stderr, mock_stdout):
sshuttle.helpers.debug2("message")
assert mock_stdout.mock_calls == [
call.flush(),
]
assert mock_stderr.mock_calls == [
Refactor debug, log and Fatal messages. This commit rewrites the log() function so that it will append a newline at the end of the message if none is present. It doesn't make sense to print a log message without a newline since the next log message (which will write a prefix) expects to be starting at the beginning of a line. Although it isn't strictly necessary, this commit also removes any newlines at the ends of messages. If I missed any, including the newline at the end of the message will continue to work as it did before. Previously, some calls were missing the newline at the end even though including it was necessary for subsequent messages to appear correctly. This code also cleans up some redundant prefixes. The log() method will prepend the prefix and the different processes should set their prefix as soon as they start. Multiline messages are still supported (although the prefix for the additional lines was changed to match the length of the prefix used for the first line).
2020-12-29 18:58:44 +01:00
call.write('prefix: message\n'),
Add more tests.
2015-11-17 03:08:12 +01:00
call.flush(),
]
@patch('sshuttle.helpers.logprefix', new='prefix: ')
@patch('sshuttle.helpers.verbose', new=1)
@patch('sshuttle.helpers.sys.stdout')
@patch('sshuttle.helpers.sys.stderr')
def test_debug2_nop(mock_stderr, mock_stdout):
sshuttle.helpers.debug2("message")
assert mock_stdout.mock_calls == []
assert mock_stderr.mock_calls == []
@patch('sshuttle.helpers.logprefix', new='prefix: ')
@patch('sshuttle.helpers.verbose', new=3)
@patch('sshuttle.helpers.sys.stdout')
@patch('sshuttle.helpers.sys.stderr')
def test_debug3(mock_stderr, mock_stdout):
sshuttle.helpers.debug3("message")
assert mock_stdout.mock_calls == [
call.flush(),
]
assert mock_stderr.mock_calls == [
Refactor debug, log and Fatal messages. This commit rewrites the log() function so that it will append a newline at the end of the message if none is present. It doesn't make sense to print a log message without a newline since the next log message (which will write a prefix) expects to be starting at the beginning of a line. Although it isn't strictly necessary, this commit also removes any newlines at the ends of messages. If I missed any, including the newline at the end of the message will continue to work as it did before. Previously, some calls were missing the newline at the end even though including it was necessary for subsequent messages to appear correctly. This code also cleans up some redundant prefixes. The log() method will prepend the prefix and the different processes should set their prefix as soon as they start. Multiline messages are still supported (although the prefix for the additional lines was changed to match the length of the prefix used for the first line).
2020-12-29 18:58:44 +01:00
call.write('prefix: message\n'),
Add more tests.
2015-11-17 03:08:12 +01:00
call.flush(),
]
@patch('sshuttle.helpers.logprefix', new='prefix: ')
@patch('sshuttle.helpers.verbose', new=2)
@patch('sshuttle.helpers.sys.stdout')
@patch('sshuttle.helpers.sys.stderr')
def test_debug3_nop(mock_stderr, mock_stdout):
sshuttle.helpers.debug3("message")
assert mock_stdout.mock_calls == []
assert mock_stderr.mock_calls == []
@patch('sshuttle.helpers.open', create=True)
def test_resolvconf_nameservers(mock_open):
mock_open.return_value = io.StringIO(u"""
# Generated by NetworkManager
search pri
nameserver 192.168.1.1
nameserver 192.168.2.1
nameserver 192.168.3.1
nameserver 192.168.4.1
nameserver 2404:6800:4004:80c::1
nameserver 2404:6800:4004:80c::2
nameserver 2404:6800:4004:80c::3
nameserver 2404:6800:4004:80c::4
""")
Make server and client handle resolv.conf differently. The server should just read from resolv.conf to find DNS servers to use. This restores this behavior after the previous commit changed it. The client now reads both /etc/resolv.conf and /run/systemd/resolve/resolv.conf. The latter is required to more reliably intercept regular DNS requests that systemd-resolved makes.
2020-11-04 02:27:57 +01:00
ns = sshuttle.helpers.resolvconf_nameservers(False)
Add more tests.
2015-11-17 03:08:12 +01:00
assert ns == [
Fix tests on Macos Swap hardcoded AF_INET(6) values for Python-provided values as they differ between Darwin and Linux (30 vs 10 for AF_INET6 for instance).
2017-09-24 14:11:26 +02:00
(AF_INET, u'192.168.1.1'), (AF_INET, u'192.168.2.1'),
(AF_INET, u'192.168.3.1'), (AF_INET, u'192.168.4.1'),
(AF_INET6, u'2404:6800:4004:80c::1'),
(AF_INET6, u'2404:6800:4004:80c::2'),
(AF_INET6, u'2404:6800:4004:80c::3'),
(AF_INET6, u'2404:6800:4004:80c::4')
Add more tests.
2015-11-17 03:08:12 +01:00
]
@patch('sshuttle.helpers.open', create=True)
def test_resolvconf_random_nameserver(mock_open):
mock_open.return_value = io.StringIO(u"""
# Generated by NetworkManager
search pri
nameserver 192.168.1.1
nameserver 192.168.2.1
nameserver 192.168.3.1
nameserver 192.168.4.1
nameserver 2404:6800:4004:80c::1
nameserver 2404:6800:4004:80c::2
nameserver 2404:6800:4004:80c::3
nameserver 2404:6800:4004:80c::4
""")
Make server and client handle resolv.conf differently. The server should just read from resolv.conf to find DNS servers to use. This restores this behavior after the previous commit changed it. The client now reads both /etc/resolv.conf and /run/systemd/resolve/resolv.conf. The latter is required to more reliably intercept regular DNS requests that systemd-resolved makes.
2020-11-04 02:27:57 +01:00
ns = sshuttle.helpers.resolvconf_random_nameserver(False)
Add more tests.
2015-11-17 03:08:12 +01:00
assert ns in [
Fix tests on Macos Swap hardcoded AF_INET(6) values for Python-provided values as they differ between Darwin and Linux (30 vs 10 for AF_INET6 for instance).
2017-09-24 14:11:26 +02:00
(AF_INET, u'192.168.1.1'), (AF_INET, u'192.168.2.1'),
(AF_INET, u'192.168.3.1'), (AF_INET, u'192.168.4.1'),
(AF_INET6, u'2404:6800:4004:80c::1'),
(AF_INET6, u'2404:6800:4004:80c::2'),
(AF_INET6, u'2404:6800:4004:80c::3'),
(AF_INET6, u'2404:6800:4004:80c::4')
Add more tests.
2015-11-17 03:08:12 +01:00
]
Mock socket bind to avoid depending on local IPs being available in test box
2017-11-07 00:00:06 +01:00
@patch('sshuttle.helpers.socket.socket.bind')
def test_islocal(mock_bind):
bind_error = socket.error(errno.EADDRNOTAVAIL)
mock_bind.side_effect = [None, bind_error, None, bind_error]
Fix tests on Macos Swap hardcoded AF_INET(6) values for Python-provided values as they differ between Darwin and Linux (30 vs 10 for AF_INET6 for instance).
2017-09-24 14:11:26 +02:00
assert sshuttle.helpers.islocal("127.0.0.1", AF_INET)
assert not sshuttle.helpers.islocal("192.0.2.1", AF_INET)
assert sshuttle.helpers.islocal("::1", AF_INET6)
assert not sshuttle.helpers.islocal("2001:db8::1", AF_INET6)
Add more tests.
2015-11-17 03:08:12 +01:00
def test_family_ip_tuple():
assert sshuttle.helpers.family_ip_tuple("127.0.0.1") \
Fix tests on Macos Swap hardcoded AF_INET(6) values for Python-provided values as they differ between Darwin and Linux (30 vs 10 for AF_INET6 for instance).
2017-09-24 14:11:26 +02:00
== (AF_INET, "127.0.0.1")
Add more tests.
2015-11-17 03:08:12 +01:00
assert sshuttle.helpers.family_ip_tuple("192.168.2.6") \
Fix tests on Macos Swap hardcoded AF_INET(6) values for Python-provided values as they differ between Darwin and Linux (30 vs 10 for AF_INET6 for instance).
2017-09-24 14:11:26 +02:00
== (AF_INET, "192.168.2.6")
Add more tests.
2015-11-17 03:08:12 +01:00
assert sshuttle.helpers.family_ip_tuple("::1") \
Fix tests on Macos Swap hardcoded AF_INET(6) values for Python-provided values as they differ between Darwin and Linux (30 vs 10 for AF_INET6 for instance).
2017-09-24 14:11:26 +02:00
== (AF_INET6, "::1")
Add more tests.
2015-11-17 03:08:12 +01:00
assert sshuttle.helpers.family_ip_tuple("2404:6800:4004:80c::1") \
Fix tests on Macos Swap hardcoded AF_INET(6) values for Python-provided values as they differ between Darwin and Linux (30 vs 10 for AF_INET6 for instance).
2017-09-24 14:11:26 +02:00
== (AF_INET6, "2404:6800:4004:80c::1")
Add more tests.
2015-11-17 03:08:12 +01:00
def test_family_to_string():
Fix tests on Macos Swap hardcoded AF_INET(6) values for Python-provided values as they differ between Darwin and Linux (30 vs 10 for AF_INET6 for instance).
2017-09-24 14:11:26 +02:00
assert sshuttle.helpers.family_to_string(AF_INET) == "AF_INET"
assert sshuttle.helpers.family_to_string(AF_INET6) == "AF_INET6"
remove version_info based branching
2020-06-03 04:53:46 +02:00
expected = 'AddressFamily.AF_UNIX'
Fix for Python3.5.
2015-11-17 03:12:41 +01:00
assert sshuttle.helpers.family_to_string(socket.AF_UNIX) == expected
Reference in New Issue Copy Permalink