Add remaining unit tests

tests/test_api.py

@@ -679,3 +679,17 @@ async def test_xbee_new(conn_mck):
     assert isinstance(api, xbee_api.XBee)
     assert conn_mck.call_count == 1
     assert conn_mck.await_count == 1
+
+
+@mock.patch.object(xbee_api.XBee, "connect", return_value=mock.MagicMock())
+async def test_connection_lost(conn_mck):
+    """Test `connection_lost` propagataion."""
+    api = await xbee_api.XBee.new(mock.sentinel.application, DEVICE_CONFIG)
+    await api.connect()
+
+    app = api._app = mock.MagicMock()
+
+    err = RuntimeError()
+    api.connection_lost(err)
+
+    app.connection_lost.assert_called_once_with(err)

tests/test_application.py

@@ -866,3 +866,11 @@ async def test_routes_updated(app, device):
     assert router2.radio_details.call_count == 0
 
     app._api._at_command.assert_awaited_once_with("DB")
+
+
+async def test_watchdog(self):
+    """Test watchdog feed method."""
+    app._api._at_command = mock.AsyncMock(return_value="OK")
+    await app._watchdog_feed()
+
+    assert app._api._at_command.mock_calls == [mock.call("VR")]

zigpy_xbee/api.py

@@ -327,8 +327,6 @@ def connection_lost(self, exc: Exception) -> None:
         if self._app is not None:
             self._app.connection_lost(exc)
 
-        self.close()
-
     def close(self):
         """Close the connection."""
         if self._uart: