Fix unit tests

tests/test_ash.py

@@ -86,14 +86,21 @@ def send_reset(self) -> None:
 
 
 class FakeTransport:
-    def __init__(self, receiver):
+    def __init__(self, receiver) -> None:
         self.receiver = receiver
-        self.paused = False
+        self.paused: bool = False
+        self.closing: bool = False
 
     def write(self, data: bytes) -> None:
         if not self.paused:
             self.receiver.data_received(data)
 
+    def close(self) -> None:
+        self.closing = True
+
+    def is_closing(self) -> bool:
+        return self.closing
+
 
 class FakeTransportOneByteAtATime(FakeTransport):
     def write(self, data: bytes) -> None:
@@ -317,6 +324,7 @@ async def test_sequence():
     loop = asyncio.get_running_loop()
     ezsp = MagicMock()
     transport = MagicMock()
+    transport.is_closing.return_value = False
 
     protocol = ash.AshProtocol(ezsp)
     protocol._write_frame = MagicMock(wraps=protocol._write_frame)
@@ -408,6 +416,7 @@ async def test_ash_protocol_startup(caplog):
 
     ezsp = MagicMock()
     transport = MagicMock()
+    transport.is_closing.return_value = False
 
     protocol = ash.AshProtocol(ezsp)
     protocol._write_frame = MagicMock(wraps=protocol._write_frame)

tests/test_uart.py

@@ -188,6 +188,7 @@ def test_eof_received(gw):
 async def test_connection_lost_reset_error_propagation(monkeypatch):
     app = MagicMock()
     transport = MagicMock()
+    transport.is_closing.return_value = False
 
     async def mockconnect(loop, protocol_factory, **kwargs):
         protocol = protocol_factory()