Skip to content

Commit 10bbce4

Browse files
committed
fixing the test to use the new socket server
Signed-off-by: Jesse Jaggars <jjaggars@redhat.com>
1 parent 2f9b2f1 commit 10bbce4

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

test/integration/test_main.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -51,10 +51,10 @@ async def wait_for_time(seconds):
5151
await asyncio.sleep(seconds)
5252

5353

54-
@patch.object(receptor.protocol.BasicProtocol, 'connection_made')
55-
def test_main_node(mock_connection_made, event_loop, receptor_config):
54+
@patch('receptor.connection.sock.serve')
55+
def test_main_node(mock_sock, event_loop, receptor_config):
5656
c = receptor.Controller(receptor_config, loop=event_loop)
5757
event_loop.call_soon(event_loop.create_task, connect_port(c.receptor))
5858
c.enable_server('{}'.format(receptor_config.node_listen))
5959
c.run()
60-
mock_connection_made.assert_called_once()
60+
mock_sock.assert_called_once()

0 commit comments

Comments
 (0)