sv/tsun-gen3-proxy
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix return type get_extra_info in FakeWriter

Browse Source
This commit is contained in:
Stefan Allius
2024-12-22 02:51:21 +01:00
parent 7878c54ecc
commit ef3ed5d287
3 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
app/tests/test_inverter_base.py
View File

@@ -58,7 +58,7 @@ class FakeWriter():
return return
def get_extra_info(self, sel: str): def get_extra_info(self, sel: str):
if sel == 'peername': if sel == 'peername':
return 'remote.intern' return ('47.1.2.3', 10000)
elif sel == 'sockname': elif sel == 'sockname':
return 'sock:1234' return 'sock:1234'
assert False assert False

2
app/tests/test_inverter_g3.py
View File

@@ -59,7 +59,7 @@ class FakeWriter():
return return
def get_extra_info(self, sel: str): def get_extra_info(self, sel: str):
if sel == 'peername': if sel == 'peername':
return 'remote.intern' return ('47.1.2.3', 10000)
elif sel == 'sockname': elif sel == 'sockname':
return 'sock:1234' return 'sock:1234'
assert False assert False

5
app/tests/test_inverter_g3p.py
View File

@@ -58,7 +58,7 @@ class FakeWriter():
return return
def get_extra_info(self, sel: str): def get_extra_info(self, sel: str):
if sel == 'peername': if sel == 'peername':
return 'remote.intern' return ('47.1.2.3', 10000)
elif sel == 'sockname': elif sel == 'sockname':
return 'sock:1234' return 'sock:1234'
assert False assert False
@@ -94,7 +94,8 @@ def patch_open_connection():
with patch.object(asyncio, 'open_connection', new_open) as conn: with patch.object(asyncio, 'open_connection', new_open) as conn:
yield conn yield conn
def test_method_calls(): def test_method_calls(config_conn):
_ = config_conn
reader = FakeReader() reader = FakeReader()
writer = FakeWriter() writer = FakeWriter()
InverterBase._registry.clear() InverterBase._registry.clear()