move StremPtr instances into Inverter class
This commit is contained in:
@@ -4,10 +4,24 @@ import asyncio
|
||||
|
||||
from itertools import count
|
||||
from mock import patch
|
||||
from app.src.async_stream import StreamPtr
|
||||
from app.src.async_stream import AsyncStream, AsyncIfcImpl
|
||||
from app.src.gen3.connection_g3 import ConnectionG3Server
|
||||
from app.src.gen3.talent import Talent
|
||||
|
||||
|
||||
class FakeInverter():
|
||||
async def async_publ_mqtt(self) -> None:
|
||||
pass # dummy funcion
|
||||
|
||||
async def async_create_remote(self, inv_prot: str, conn_class) -> None:
|
||||
pass # dummy function
|
||||
|
||||
def __init__ (self):
|
||||
self.remote = StreamPtr(None)
|
||||
self.local = StreamPtr(None)
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def patch_async_init():
|
||||
with patch.object(AsyncStream, '__init__') as conn:
|
||||
@@ -71,8 +85,8 @@ def test_method_calls(patch_talent_init, patch_healthy, patch_async_close, patch
|
||||
writer = FakeWriter()
|
||||
id_str = "id_string"
|
||||
addr = ('proxy.local', 10000)
|
||||
conn = ConnectionG3Server(reader, writer, addr,
|
||||
rstream= None, id_str=id_str)
|
||||
conn = ConnectionG3Server(FakeInverter(), reader, writer, addr,
|
||||
id_str=id_str)
|
||||
assert 5 == conn._ifc.get_conn_no()
|
||||
spy2.assert_called_once_with(conn, True, conn._ifc, id_str)
|
||||
conn.healthy()
|
||||
|
||||
Reference in New Issue
Block a user