Unit test UART callbacks

zigpy · May 2, 2024 · 61658a5 · 61658a5
1 parent 2746492
commit 61658a5
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/tests/test_uart.py b/tests/test_uart.py
@@ -8,7 +8,7 @@
 from bellows import uart
 import bellows.types as t
 
-from .async_mock import AsyncMock, MagicMock, patch, sentinel
+from .async_mock import AsyncMock, MagicMock, call, patch, sentinel
 
 
 @pytest.mark.parametrize("flow_control", ["software", "hardware"])
@@ -239,3 +239,13 @@ async def test_wait_for_startup_reset_failure(gw):
         await asyncio.wait_for(gw.wait_for_startup_reset(), 0.01)
 
     assert gw._startup_reset_future is None
+
+
+async def test_callbacks(gw):
+    gw.data_received(b"some ezsp packet")
+    assert gw._application.frame_received.mock_calls == [call(b"some ezsp packet")]
+
+    gw.error_received(t.NcpResetCode.RESET_SOFTWARE)
+    assert gw._application.enter_failed_state.mock_calls == [
+        call(t.NcpResetCode.RESET_SOFTWARE)
+    ]