Fix startup broken by #577 (#578)

* Include missing calls to `write_config` and `register_endpoints`

* Extend unit tests to ensure config is written and endpoints are added

bellows/zigbee/application.py

@@ -25,6 +25,7 @@
 
 import bellows
 from bellows.config import (
+    CONF_EZSP_CONFIG,
     CONF_EZSP_POLICIES,
     CONF_PARAM_MAX_WATCHDOG_FAILURES,
     CONF_USE_THREAD,
@@ -142,6 +143,8 @@ async def connect(self) -> None:
             ezsp.close()
             raise
 
+        # Writing config is required here because network info can't be loaded otherwise
+        await ezsp.write_config(self.config[CONF_EZSP_CONFIG])
         self._ezsp = ezsp
 
     async def _ensure_network_running(self) -> bool:
@@ -169,6 +172,7 @@ async def _ensure_network_running(self) -> bool:
     async def start_network(self):
         ezsp = self._ezsp
 
+        await self.register_endpoints()
         await self._ensure_network_running()
 
         if await repairs.fix_invalid_tclk_partner_ieee(ezsp):

tests/test_application.py

@@ -64,6 +64,7 @@ def ezsp_mock(ieee):
     mock_ezsp.readCounters = AsyncMock(return_value=[[0] * 10])
     mock_ezsp.readAndClearCounters = AsyncMock(return_value=[[0] * 10])
     mock_ezsp.setPolicy = AsyncMock(return_value=[0])
+    mock_ezsp.addEndpoint = AsyncMock(return_value=[t.EmberStatus.SUCCESS])
     mock_ezsp.get_board_info = AsyncMock(
         return_value=("Mock Manufacturer", "Mock board", "Mock version")
     )
@@ -220,6 +221,9 @@ def form_network():
 
     with p1, p2 as multicast_mock:
         await app.startup(auto_form=auto_form)
+
+    assert ezsp_mock.write_config.call_count == 1
+    assert ezsp_mock.addEndpoint.call_count >= 2
     assert multicast_mock.await_count == 1