Google assistant: disconnect user agent when not found in google (#48233)

tests/components/cloud/test_google_config.py

@@ -1,5 +1,5 @@
 """Test the Cloud Google Config."""
-from unittest.mock import AsyncMock, Mock, patch
+from unittest.mock import Mock, patch
 
 import pytest
 
@@ -41,21 +41,19 @@ async def test_google_update_report_state(mock_conf, hass, cloud_prefs):
     assert len(mock_report_state.mock_calls) == 1
 
 
-async def test_sync_entities(aioclient_mock, hass, cloud_prefs):
+async def test_sync_entities(mock_conf, hass, cloud_prefs):
     """Test sync devices."""
-    config = CloudGoogleConfig(
-        hass,
-        GACTIONS_SCHEMA({}),
-        "mock-user-id",
-        cloud_prefs,
-        Mock(auth=Mock(async_check_token=AsyncMock())),
-    )
+    await mock_conf.async_initialize()
+    await mock_conf.async_connect_agent_user("mock-user-id")
+
+    assert len(mock_conf._store.agent_user_ids) == 1
 
     with patch(
         "hass_nabucasa.cloud_api.async_google_actions_request_sync",
         return_value=Mock(status=HTTP_NOT_FOUND),
     ) as mock_request_sync:
-        assert await config.async_sync_entities("user") == HTTP_NOT_FOUND
+        assert await mock_conf.async_sync_entities("mock-user-id") == HTTP_NOT_FOUND
+        assert len(mock_conf._store.agent_user_ids) == 0
         assert len(mock_request_sync.mock_calls) == 1
 
 
@@ -165,7 +163,29 @@ async def test_google_entity_registry_sync(hass, mock_cloud_login, cloud_prefs):
 
         assert len(mock_sync.mock_calls) == 3
 
+
+async def test_sync_google_when_started(hass, mock_cloud_login, cloud_prefs):
+    """Test Google config syncs on init."""
+    config = CloudGoogleConfig(
+        hass, GACTIONS_SCHEMA({}), "mock-user-id", cloud_prefs, hass.data["cloud"]
+    )
     with patch.object(config, "async_sync_entities_all") as mock_sync:
+        await config.async_initialize()
+        await config.async_connect_agent_user("mock-user-id")
+        assert len(mock_sync.mock_calls) == 1
+
+
+async def test_sync_google_on_home_assistant_start(hass, mock_cloud_login, cloud_prefs):
+    """Test Google config syncs when home assistant started."""
+    config = CloudGoogleConfig(
+        hass, GACTIONS_SCHEMA({}), "mock-user-id", cloud_prefs, hass.data["cloud"]
+    )
+    hass.state = CoreState.starting
+    with patch.object(config, "async_sync_entities_all") as mock_sync:
+        await config.async_initialize()
+        await config.async_connect_agent_user("mock-user-id")
+        assert len(mock_sync.mock_calls) == 0
+
         hass.bus.async_fire(EVENT_HOMEASSISTANT_STARTED)
         await hass.async_block_till_done()
         assert len(mock_sync.mock_calls) == 1