Add Azure DevOps Integration (#33765)

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

tests/components/azure_devops/__init__.py

@@ -0,0 +1 @@
+"""Tests for the Azure DevOps integration."""

tests/components/azure_devops/test_config_flow.py

@@ -0,0 +1,257 @@
+"""Test the Azure DevOps config flow."""
+from aioazuredevops.core import DevOpsProject
+import aiohttp
+
+from homeassistant import config_entries, data_entry_flow
+from homeassistant.components.azure_devops.const import (
+    CONF_ORG,
+    CONF_PAT,
+    CONF_PROJECT,
+    DOMAIN,
+)
+from homeassistant.core import HomeAssistant
+
+from tests.async_mock import patch
+from tests.common import MockConfigEntry
+
+FIXTURE_REAUTH_INPUT = {CONF_PAT: "abc123"}
+FIXTURE_USER_INPUT = {CONF_ORG: "random", CONF_PROJECT: "project", CONF_PAT: "abc123"}
+
+UNIQUE_ID = "random_project"
+
+
+async def test_show_user_form(hass: HomeAssistant) -> None:
+    """Test that the setup form is served."""
+    result = await hass.config_entries.flow.async_init(
+        DOMAIN, context={"source": config_entries.SOURCE_USER}
+    )
+
+    assert result["type"] == data_entry_flow.RESULT_TYPE_FORM
+    assert result["step_id"] == "user"
+
+
+async def test_authorization_error(hass: HomeAssistant) -> None:
+    """Test we show user form on Azure DevOps authorization error."""
+    with patch(
+        "homeassistant.components.azure_devops.config_flow.DevOpsClient.authorize",
+        return_value=False,
+    ):
+        result = await hass.config_entries.flow.async_init(
+            DOMAIN, context={"source": config_entries.SOURCE_USER}
+        )
+
+        assert result["type"] == data_entry_flow.RESULT_TYPE_FORM
+        assert result["step_id"] == "user"
+
+        result2 = await hass.config_entries.flow.async_configure(
+            result["flow_id"], FIXTURE_USER_INPUT,
+        )
+        await hass.async_block_till_done()
+
+        assert result2["type"] == data_entry_flow.RESULT_TYPE_FORM
+        assert result2["step_id"] == "user"
+        assert result2["errors"] == {"base": "authorization_error"}
+
+
+async def test_reauth_authorization_error(hass: HomeAssistant) -> None:
+    """Test we show user form on Azure DevOps authorization error."""
+    with patch(
+        "homeassistant.components.azure_devops.config_flow.DevOpsClient.authorize",
+        return_value=False,
+    ):
+        result = await hass.config_entries.flow.async_init(
+            DOMAIN, context={"source": "reauth"}, data=FIXTURE_USER_INPUT
+        )
+
+        assert result["type"] == data_entry_flow.RESULT_TYPE_FORM
+        assert result["step_id"] == "reauth"
+
+        result2 = await hass.config_entries.flow.async_configure(
+            result["flow_id"], FIXTURE_REAUTH_INPUT,
+        )
+        await hass.async_block_till_done()
+
+        assert result2["type"] == data_entry_flow.RESULT_TYPE_FORM
+        assert result2["step_id"] == "reauth"
+        assert result2["errors"] == {"base": "authorization_error"}
+
+
+async def test_connection_error(hass: HomeAssistant) -> None:
+    """Test we show user form on Azure DevOps connection error."""
+    with patch(
+        "homeassistant.components.azure_devops.config_flow.DevOpsClient.authorize",
+        side_effect=aiohttp.ClientError,
+    ):
+        result = await hass.config_entries.flow.async_init(
+            DOMAIN, context={"source": config_entries.SOURCE_USER}
+        )
+
+        assert result["type"] == data_entry_flow.RESULT_TYPE_FORM
+        assert result["step_id"] == "user"
+
+        result2 = await hass.config_entries.flow.async_configure(
+            result["flow_id"], FIXTURE_USER_INPUT,
+        )
+        await hass.async_block_till_done()
+
+        assert result2["type"] == data_entry_flow.RESULT_TYPE_FORM
+        assert result2["step_id"] == "user"
+        assert result2["errors"] == {"base": "connection_error"}
+
+
+async def test_reauth_connection_error(hass: HomeAssistant) -> None:
+    """Test we show user form on Azure DevOps connection error."""
+    with patch(
+        "homeassistant.components.azure_devops.config_flow.DevOpsClient.authorize",
+        side_effect=aiohttp.ClientError,
+    ):
+        result = await hass.config_entries.flow.async_init(
+            DOMAIN, context={"source": "reauth"}, data=FIXTURE_USER_INPUT
+        )
+
+        assert result["type"] == data_entry_flow.RESULT_TYPE_FORM
+        assert result["step_id"] == "reauth"
+
+        result2 = await hass.config_entries.flow.async_configure(
+            result["flow_id"], FIXTURE_REAUTH_INPUT,
+        )
+        await hass.async_block_till_done()
+
+        assert result2["type"] == data_entry_flow.RESULT_TYPE_FORM
+        assert result2["step_id"] == "reauth"
+        assert result2["errors"] == {"base": "connection_error"}
+
+
+async def test_project_error(hass: HomeAssistant) -> None:
+    """Test we show user form on Azure DevOps connection error."""
+    with patch(
+        "homeassistant.components.azure_devops.config_flow.DevOpsClient.authorized",
+        return_value=True,
+    ), patch(
+        "homeassistant.components.azure_devops.config_flow.DevOpsClient.authorize",
+    ), patch(
+        "homeassistant.components.azure_devops.config_flow.DevOpsClient.get_project",
+        return_value=None,
+    ):
+        result = await hass.config_entries.flow.async_init(
+            DOMAIN, context={"source": config_entries.SOURCE_USER}
+        )
+
+        assert result["type"] == data_entry_flow.RESULT_TYPE_FORM
+        assert result["step_id"] == "user"
+
+        result2 = await hass.config_entries.flow.async_configure(
+            result["flow_id"], FIXTURE_USER_INPUT,
+        )
+        await hass.async_block_till_done()
+
+        assert result2["type"] == data_entry_flow.RESULT_TYPE_FORM
+        assert result2["step_id"] == "user"
+        assert result2["errors"] == {"base": "project_error"}
+
+
+async def test_reauth_project_error(hass: HomeAssistant) -> None:
+    """Test we show user form on Azure DevOps project error."""
+    with patch(
+        "homeassistant.components.azure_devops.config_flow.DevOpsClient.authorize",
+    ), patch(
+        "homeassistant.components.azure_devops.config_flow.DevOpsClient.authorized",
+        return_value=True,
+    ), patch(
+        "homeassistant.components.azure_devops.config_flow.DevOpsClient.get_project",
+        return_value=None,
+    ):
+        result = await hass.config_entries.flow.async_init(
+            DOMAIN, context={"source": "reauth"}, data=FIXTURE_USER_INPUT
+        )
+
+        assert result["type"] == data_entry_flow.RESULT_TYPE_FORM
+        assert result["step_id"] == "reauth"
+
+        result2 = await hass.config_entries.flow.async_configure(
+            result["flow_id"], FIXTURE_REAUTH_INPUT,
+        )
+        await hass.async_block_till_done()
+
+        assert result2["type"] == data_entry_flow.RESULT_TYPE_FORM
+        assert result2["step_id"] == "reauth"
+        assert result2["errors"] == {"base": "project_error"}
+
+
+async def test_reauth_flow(hass: HomeAssistant) -> None:
+    """Test reauth works."""
+    with patch(
+        "homeassistant.components.azure_devops.config_flow.DevOpsClient.authorize",
+        return_value=False,
+    ):
+        mock_config = MockConfigEntry(
+            domain=DOMAIN, unique_id=UNIQUE_ID, data=FIXTURE_USER_INPUT
+        )
+        mock_config.add_to_hass(hass)
+
+        result = await hass.config_entries.flow.async_init(
+            DOMAIN, context={"source": "reauth"}, data=FIXTURE_USER_INPUT
+        )
+
+        assert result["type"] == data_entry_flow.RESULT_TYPE_FORM
+        assert result["step_id"] == "reauth"
+        assert result["errors"] == {"base": "authorization_error"}
+
+    with patch(
+        "homeassistant.components.azure_devops.config_flow.DevOpsClient.authorize",
+    ), patch(
+        "homeassistant.components.azure_devops.config_flow.DevOpsClient.authorized",
+        return_value=True,
+    ), patch(
+        "homeassistant.components.azure_devops.config_flow.DevOpsClient.get_project",
+        return_value=DevOpsProject(
+            "abcd-abcd-abcd-abcd", FIXTURE_USER_INPUT[CONF_PROJECT]
+        ),
+    ):
+        result2 = await hass.config_entries.flow.async_configure(
+            result["flow_id"], FIXTURE_REAUTH_INPUT,
+        )
+        await hass.async_block_till_done()
+
+        assert result2["type"] == data_entry_flow.RESULT_TYPE_ABORT
+        assert result2["reason"] == "reauth_successful"
+
+
+async def test_full_flow_implementation(hass: HomeAssistant) -> None:
+    """Test registering an integration and finishing flow works."""
+    with patch(
+        "homeassistant.components.azure_devops.async_setup", return_value=True
+    ) as mock_setup, patch(
+        "homeassistant.components.azure_devops.async_setup_entry", return_value=True,
+    ) as mock_setup_entry, patch(
+        "homeassistant.components.azure_devops.config_flow.DevOpsClient.authorized",
+        return_value=True,
+    ), patch(
+        "homeassistant.components.azure_devops.config_flow.DevOpsClient.authorize",
+    ), patch(
+        "homeassistant.components.azure_devops.config_flow.DevOpsClient.get_project",
+        return_value=DevOpsProject(
+            "abcd-abcd-abcd-abcd", FIXTURE_USER_INPUT[CONF_PROJECT]
+        ),
+    ):
+        result = await hass.config_entries.flow.async_init(
+            DOMAIN, context={"source": config_entries.SOURCE_USER}
+        )
+
+        assert result["type"] == data_entry_flow.RESULT_TYPE_FORM
+        assert result["step_id"] == "user"
+
+        result2 = await hass.config_entries.flow.async_configure(
+            result["flow_id"], FIXTURE_USER_INPUT,
+        )
+        await hass.async_block_till_done()
+        assert len(mock_setup.mock_calls) == 1
+        assert len(mock_setup_entry.mock_calls) == 1
+
+        assert result2["type"] == data_entry_flow.RESULT_TYPE_CREATE_ENTRY
+        assert (
+            result2["title"]
+            == f"{FIXTURE_USER_INPUT[CONF_ORG]}/{FIXTURE_USER_INPUT[CONF_PROJECT]}"
+        )
+        assert result2["data"][CONF_ORG] == FIXTURE_USER_INPUT[CONF_ORG]
+        assert result2["data"][CONF_PROJECT] == FIXTURE_USER_INPUT[CONF_PROJECT]