Add WS API for creating a Thread network (#88830)

* Add WS API for creating a Thread network

* Add tests

tests/components/otbr/test_websocket_api.py

@@ -96,3 +96,139 @@ async def test_get_info_fetch_fails(
     assert msg["id"] == 5
     assert not msg["success"]
     assert msg["error"]["code"] == "get_dataset_failed"
+
+
+async def test_create_network(
+    hass: HomeAssistant,
+    aioclient_mock: AiohttpClientMocker,
+    otbr_config_entry,
+    websocket_client,
+) -> None:
+    """Test create network."""
+
+    with patch(
+        "python_otbr_api.OTBR.create_active_dataset"
+    ) as create_dataset_mock, patch(
+        "python_otbr_api.OTBR.set_enabled"
+    ) as set_enabled_mock:
+        await websocket_client.send_json(
+            {
+                "id": 5,
+                "type": "otbr/create_network",
+            }
+        )
+
+        msg = await websocket_client.receive_json()
+        assert msg["id"] == 5
+        assert msg["success"]
+        assert msg["result"] is None
+
+    create_dataset_mock.assert_called_once_with(
+        python_otbr_api.models.OperationalDataSet(network_name="home-assistant")
+    )
+    assert len(set_enabled_mock.mock_calls) == 2
+    assert set_enabled_mock.mock_calls[0][1][0] is False
+    assert set_enabled_mock.mock_calls[1][1][0] is True
+
+
+async def test_create_network_no_entry(
+    hass: HomeAssistant,
+    aioclient_mock: AiohttpClientMocker,
+    hass_ws_client: WebSocketGenerator,
+) -> None:
+    """Test create network."""
+    await async_setup_component(hass, "otbr", {})
+    websocket_client = await hass_ws_client(hass)
+    await websocket_client.send_json(
+        {
+            "id": 5,
+            "type": "otbr/create_network",
+        }
+    )
+
+    msg = await websocket_client.receive_json()
+    assert msg["id"] == 5
+    assert not msg["success"]
+    assert msg["error"]["code"] == "not_loaded"
+
+
+async def test_get_info_fetch_fails_1(
+    hass: HomeAssistant,
+    aioclient_mock: AiohttpClientMocker,
+    otbr_config_entry,
+    websocket_client,
+) -> None:
+    """Test create network."""
+    await async_setup_component(hass, "otbr", {})
+
+    with patch(
+        "python_otbr_api.OTBR.set_enabled",
+        side_effect=python_otbr_api.OTBRError,
+    ):
+        await websocket_client.send_json(
+            {
+                "id": 5,
+                "type": "otbr/create_network",
+            }
+        )
+        msg = await websocket_client.receive_json()
+
+    assert msg["id"] == 5
+    assert not msg["success"]
+    assert msg["error"]["code"] == "set_enabled_failed"
+
+
+async def test_get_info_fetch_fails_2(
+    hass: HomeAssistant,
+    aioclient_mock: AiohttpClientMocker,
+    otbr_config_entry,
+    websocket_client,
+) -> None:
+    """Test create network."""
+    await async_setup_component(hass, "otbr", {})
+
+    with patch(
+        "python_otbr_api.OTBR.set_enabled",
+    ), patch(
+        "python_otbr_api.OTBR.create_active_dataset",
+        side_effect=python_otbr_api.OTBRError,
+    ):
+        await websocket_client.send_json(
+            {
+                "id": 5,
+                "type": "otbr/create_network",
+            }
+        )
+        msg = await websocket_client.receive_json()
+
+    assert msg["id"] == 5
+    assert not msg["success"]
+    assert msg["error"]["code"] == "create_active_dataset_failed"
+
+
+async def test_get_info_fetch_fails_3(
+    hass: HomeAssistant,
+    aioclient_mock: AiohttpClientMocker,
+    otbr_config_entry,
+    websocket_client,
+) -> None:
+    """Test create network."""
+    await async_setup_component(hass, "otbr", {})
+
+    with patch(
+        "python_otbr_api.OTBR.set_enabled",
+        side_effect=[None, python_otbr_api.OTBRError],
+    ), patch(
+        "python_otbr_api.OTBR.create_active_dataset",
+    ):
+        await websocket_client.send_json(
+            {
+                "id": 5,
+                "type": "otbr/create_network",
+            }
+        )
+        msg = await websocket_client.receive_json()
+
+    assert msg["id"] == 5
+    assert not msg["success"]
+    assert msg["error"]["code"] == "set_enabled_failed"