Skip to content
Snippets Groups Projects
Commit 76f581eb2dc1 authored by Luke Bakken's avatar Luke Bakken
Browse files

Fix tests

parent 93696e6c0d9b
No related branches found
No related tags found
No related merge requests found
...@@ -112,7 +112,7 @@ ...@@ -112,7 +112,7 @@
def test_tornado_connection_timeout(self): def test_tornado_connection_timeout(self):
with self.assertRaises(exceptions.AMQPConnectionError) as err_ctx: with self.assertRaises(exceptions.AMQPConnectionError) as err_ctx:
with mock.patch( with mock.patch(
'pika.TornadoConnection' 'pika.adapters.tornado_connection.TornadoConnection'
'._create_tcp_connection_socket', '._create_tcp_connection_socket',
return_value=mock.Mock( return_value=mock.Mock(
spec_set=socket.socket, spec_set=socket.socket,
...@@ -131,7 +131,7 @@ ...@@ -131,7 +131,7 @@
def test_twisted_connection_timeout(self): def test_twisted_connection_timeout(self):
with self.assertRaises(exceptions.AMQPConnectionError) as err_ctx: with self.assertRaises(exceptions.AMQPConnectionError) as err_ctx:
with mock.patch( with mock.patch(
'pika.TwistedConnection._create_tcp_connection_socket', 'pika.adapters.twisted_connection.TwistedConnection._create_tcp_connection_socket',
return_value=mock.Mock( return_value=mock.Mock(
spec_set=socket.socket, spec_set=socket.socket,
connect=mock.Mock( connect=mock.Mock(
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment