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

use mocked getaddrinfo to make host name resolution stable

Browse Source
This commit is contained in:
mangano-ito 2022-02-09 21:25:42 +09:00
parent 19e2a1810d
commit e5eb5afef0
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tests/client/test_options.py
View File

@ -1,5 +1,6 @@
import socket import socket
from argparse import ArgumentTypeError as Fatal from argparse import ArgumentTypeError as Fatal
from unittest.mock import patch
import pytest import pytest
@ -116,7 +117,8 @@ def test_convert_arg_line_to_args_skips_comments():
assert parser.convert_arg_line_to_args("# whatever something") == [] assert parser.convert_arg_line_to_args("# whatever something") == []
def test_parse_subnetport_host(): @patch('sshuttle.options.socket.getaddrinfo', side_effect = _mock_getaddrinfo)
def test_parse_subnetport_host(mock_getaddrinfo):
assert set(sshuttle.options.parse_subnetport('example.com')) \ assert set(sshuttle.options.parse_subnetport('example.com')) \
== set([ == set([
(socket.AF_INET6, '2606:2800:220:1:248:1893:25c8:1946', 128, 0, 0), (socket.AF_INET6, '2606:2800:220:1:248:1893:25c8:1946', 128, 0, 0),
@ -124,7 +126,8 @@ def test_parse_subnetport_host():
]) ])
def test_parse_subnetport_host_with_port(): @patch('sshuttle.options.socket.getaddrinfo', side_effect = _mock_getaddrinfo)
def test_parse_subnetport_host_with_port(mock_getaddrinfo):
assert set(sshuttle.options.parse_subnetport('example.com:80')) \ assert set(sshuttle.options.parse_subnetport('example.com:80')) \
== set([ == set([
(socket.AF_INET6, '2606:2800:220:1:248:1893:25c8:1946', 128, 80, 80), (socket.AF_INET6, '2606:2800:220:1:248:1893:25c8:1946', 128, 80, 80),