windows: support automatic nameserver detection for --dns option

This commit is contained in:
nom3ad
2024-05-01 20:08:15 +05:30
committed by Brian May
parent 51287dc4db
commit b826ae6b91
3 changed files with 39 additions and 12 deletions

View File

@ -143,7 +143,7 @@ nameserver 2404:6800:4004:80c::4
@patch('sshuttle.helpers.open', create=True)
def test_resolvconf_random_nameserver(mock_open):
def test_get_random_nameserver(mock_open):
mock_open.return_value = io.StringIO(u"""
# Generated by NetworkManager
search pri
@ -156,7 +156,7 @@ nameserver 2404:6800:4004:80c::2
nameserver 2404:6800:4004:80c::3
nameserver 2404:6800:4004:80c::4
""")
ns = sshuttle.helpers.resolvconf_random_nameserver(False)
ns = sshuttle.helpers.get_random_nameserver()
assert ns in [
(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'),