From 416636fa9b41e55d63a358fc997ad80174222a46 Mon Sep 17 00:00:00 2001 From: vieira Date: Mon, 6 Nov 2017 23:00:06 +0000 Subject: Mock socket bind to avoid depending on local IPs being available in test box --- sshuttle/tests/client/test_helpers.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/sshuttle/tests/client/test_helpers.py b/sshuttle/tests/client/test_helpers.py index 67c6682..4c7c9d6 100644 --- a/sshuttle/tests/client/test_helpers.py +++ b/sshuttle/tests/client/test_helpers.py @@ -2,6 +2,7 @@ from mock import patch, call import sys import io import socket +import errno import sshuttle.helpers @@ -162,7 +163,11 @@ nameserver 2404:6800:4004:80c::4 ] -def test_islocal(): +@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] + assert sshuttle.helpers.islocal("127.0.0.1", socket.AF_INET) assert not sshuttle.helpers.islocal("192.0.2.1", socket.AF_INET) assert sshuttle.helpers.islocal("::1", socket.AF_INET6) -- cgit v1.2.3