diff --git a/py3tftp/protocols.py b/py3tftp/protocols.py index f1ac3b6..0b9a3e0 100644 --- a/py3tftp/protocols.py +++ b/py3tftp/protocols.py @@ -406,8 +406,7 @@ def datagram_received(self, data, addr): connect = self.loop.create_datagram_endpoint( lambda: protocol(data, file_handler_cls, addr, self.extra_opts), - local_addr=(self.host_interface, - 0, )) + sock=self.transport.get_extra_info('socket')) self.loop.create_task(connect) diff --git a/tests/unit/protocols_test.py b/tests/unit/protocols_test.py index 917c0f6..a033ac4 100644 --- a/tests/unit/protocols_test.py +++ b/tests/unit/protocols_test.py @@ -32,6 +32,7 @@ def test_datagram_received(self): mock_loop.create_task.return_value = True proto = TFTPServerProtocol('127.0.0.1', mock_loop, {}) + proto.transport = MagicMock() proto.datagram_received(data, ('127.0.0.1', 0,)) self.assertTrue(mock_loop.create_datagram_endpoint.called)