Fix sync context in icloud (#77582)

Co-authored-by: Martin Hjelmare <marhje52@gmail.com>

tests/components/icloud/test_init.py

@@ -0,0 +1,45 @@
+"""Tests for the iCloud config flow."""
+from unittest.mock import Mock, patch
+
+import pytest
+
+from homeassistant.components.icloud.const import DOMAIN
+from homeassistant.config_entries import ConfigEntryState
+from homeassistant.core import HomeAssistant
+
+from .const import MOCK_CONFIG, USERNAME
+
+from tests.common import MockConfigEntry
+
+
+@pytest.fixture(name="service_2fa")
+def mock_controller_2fa_service():
+    """Mock a successful 2fa service."""
+    with patch(
+        "homeassistant.components.icloud.account.PyiCloudService"
+    ) as service_mock:
+        service_mock.return_value.requires_2fa = True
+        service_mock.return_value.requires_2sa = True
+        service_mock.return_value.validate_2fa_code = Mock(return_value=True)
+        service_mock.return_value.is_trusted_session = False
+        yield service_mock
+
+
+@pytest.mark.usefixtures("service_2fa")
+async def test_setup_2fa(hass: HomeAssistant) -> None:
+    """Test that invalid login triggers reauth flow."""
+    config_entry = MockConfigEntry(
+        domain=DOMAIN, data=MOCK_CONFIG, entry_id="test", unique_id=USERNAME
+    )
+    config_entry.add_to_hass(hass)
+
+    assert config_entry.state is ConfigEntryState.NOT_LOADED
+    assert not hass.config_entries.flow.async_progress()
+
+    await hass.config_entries.async_setup(config_entry.entry_id)
+    await hass.async_block_till_done()
+
+    assert config_entry.state is ConfigEntryState.LOADED
+    in_progress_flows = hass.config_entries.flow.async_progress()
+    assert len(in_progress_flows) == 1
+    assert in_progress_flows[0]["context"]["unique_id"] == config_entry.unique_id