def test_create_connection(self, mocker):
mocker.patch('sliplib.socket.create_connection')
sliplib.socket.create_connection.reset_mock()
new_sock = socket.socket(self.family)
sliplib.socket.create_connection.return_value = new_sock
sock = SlipSocket.create_connection(self.far_address)
assert isinstance(sock, SlipSocket)
assert sock.socket is new_sock
sliplib.socket.create_connection.assert_called_once_with(
self.far_address, None, None)
def test_create_connection(self, mocker):
"""Test that create_connection gives a SlipSocket."""
new_sock_mock = mocker.Mock(spec=socket.socket(self.family),
family=self.family,
type=socket.SOCK_STREAM,
proto=0) # pylint: disable=no-member
create_connection_mock = mocker.patch(
'sliplib.socket.create_connection', return_value=new_sock_mock)
sock = SlipSocket.create_connection(self.far_address)
assert isinstance(sock, SlipSocket)
assert sock.socket is new_sock_mock
create_connection_mock.assert_called_once_with(self.far_address[0:2],
None, None)
def __init__(self, address):
self.sock = SlipSocket.create_connection(address)