File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 1+ import asyncio
2+ import socket
3+ from unittest .mock import patch
4+
15import pytest
6+
27import receptor
38from receptor .config import ReceptorConfig
49from receptor .receptor import Receptor
5- import socket
6- import asyncio
7- from unittest .mock import patch
810
911
1012@pytest .fixture
@@ -37,7 +39,7 @@ async def connect_port(receptor_obj):
3739 n = 5
3840 while n :
3941 sock = socket .socket (socket .AF_INET , socket .SOCK_STREAM )
40- node , port = receptor_obj .config .node_listen .split (":" )
42+ node , port = receptor_obj .config .node_listen [ 0 ] .split (":" )
4143 result = sock .connect_ex ((node , int (port )))
4244 if result != 0 :
4345 await asyncio .sleep (1 )
@@ -55,6 +57,6 @@ async def wait_for_time(seconds):
5557def test_main_node (mock_sock , event_loop , receptor_config ):
5658 c = receptor .Controller (receptor_config , loop = event_loop )
5759 event_loop .call_soon (event_loop .create_task , connect_port (c .receptor ))
58- c .enable_server ('{}' . format ( receptor_config .node_listen ) )
60+ c .enable_server (receptor_config .node_listen )
5961 c .run ()
6062 mock_sock .assert_called_once ()
You can’t perform that action at this time.
0 commit comments