Add unit tests (#133)

* Add initial test & add pipeline

* Add very basic YAML config tests

* Add coverage reporting

* Add some webserver & template loading tests

* Add test cases for the helpers

* Implement initial OIDC server tests

* Test codestore & discovery checker

* Test basics of the config flow

* Add test for the HA auth provider

* Cleaned up tests & test injection

tests/test_hass_webserver.py

@@ -0,0 +1,151 @@
+"""Tests for the registered webpages"""
+
+import os
+from auth_oidc.config.const import (
+    DISCOVERY_URL,
+    CLIENT_ID,
+    FEATURES,
+    FEATURES_DISABLE_FRONTEND_INJECTION,
+)
+import pytest
+
+from homeassistant.core import HomeAssistant
+from homeassistant.setup import async_setup_component
+from homeassistant.components.http import StaticPathConfig, DOMAIN as HTTP_DOMAIN
+
+from custom_components.auth_oidc import DOMAIN
+
+
+async def setup(hass: HomeAssistant, enable_frontend_changes: bool = None):
+    mock_config = {
+        DOMAIN: {
+            CLIENT_ID: "dummy",
+            DISCOVERY_URL: "https://example.com/.well-known/openid-configuration",
+            FEATURES: {
+                FEATURES_DISABLE_FRONTEND_INJECTION: not enable_frontend_changes
+            },
+        }
+    }
+
+    if enable_frontend_changes is None:
+        del mock_config[DOMAIN][FEATURES][FEATURES_DISABLE_FRONTEND_INJECTION]
+
+    result = await async_setup_component(hass, DOMAIN, mock_config)
+    assert result
+
+
+@pytest.mark.asyncio
+async def test_welcome_page_registration(hass: HomeAssistant, hass_client):
+    """Test that welcome page is present if frontend changes are disabled."""
+
+    await setup(hass, enable_frontend_changes=False)
+
+    client = await hass_client()
+    resp = await client.get("/auth/oidc/welcome", allow_redirects=False)
+    assert resp.status == 200
+
+
+@pytest.mark.asyncio
+async def test_welcome_page_registration_with_changes(hass: HomeAssistant, hass_client):
+    """Test that welcome page is redirect if frontend changes are enabled."""
+
+    await setup(hass, enable_frontend_changes=True)
+
+    client = await hass_client()
+    resp = await client.get("/auth/oidc/welcome", allow_redirects=False)
+    assert resp.status == 307
+
+
+@pytest.mark.asyncio
+async def test_redirect_page_registration(hass: HomeAssistant, hass_client):
+    """Test that redirect page shows OIDC misconfiguration error if OIDC server is not reachable."""
+
+    await setup(hass)
+
+    client = await hass_client()
+    resp = await client.get("/auth/oidc/redirect", allow_redirects=False)
+    assert resp.status == 200
+    text = await resp.text()
+    assert "Integration is misconfigured" in text
+
+    resp2 = await client.post("/auth/oidc/redirect", allow_redirects=False)
+    assert resp2.status == 200
+
+
+@pytest.mark.asyncio
+async def test_callback_registration(hass: HomeAssistant, hass_client):
+    """Test that callback page is reachable."""
+
+    await setup(hass)
+
+    client = await hass_client()
+    resp = await client.get("/auth/oidc/callback", allow_redirects=False)
+    assert resp.status == 200
+
+
+@pytest.mark.asyncio
+async def test_finish_registration(hass: HomeAssistant, hass_client):
+    """Test that finish page is reachable."""
+
+    await setup(hass)
+
+    client = await hass_client()
+    resp = await client.get("/auth/oidc/finish", allow_redirects=False)
+    assert resp.status == 200
+    text = await resp.text()
+
+    # Should miss the code parameter if called without it
+    assert "Missing code" in text
+
+    resp2 = await client.get("/auth/oidc/finish?code=123456", allow_redirects=False)
+    assert resp2.status == 200
+    text2 = await resp2.text()
+    assert "Missing code" not in text2
+    assert "123456" in text2
+
+
+@pytest.mark.asyncio
+async def test_finish_post(hass: HomeAssistant, hass_client):
+    """Test that finish page works with POST."""
+
+    await setup(hass)
+    client = await hass_client()
+    resp = await client.post("/auth/oidc/finish", data={}, allow_redirects=False)
+    assert resp.status == 500
+
+    resp2 = await client.post(
+        "/auth/oidc/finish", data={"code": "456888"}, allow_redirects=False
+    )
+    assert resp2.status == 302
+    assert resp2.headers["Location"] == "/?storeToken=true"
+    assert resp2.cookies["auth_oidc_code"].value == "456888"
+
+
+# Test the frontend injection
+@pytest.mark.asyncio
+async def test_frontend_injection(hass: HomeAssistant, hass_client):
+    """Test that frontend injection works."""
+
+    # Because there is no frontend in the test setup,
+    # we'll have to fake /auth/authorize for the changes to register
+    await async_setup_component(hass, HTTP_DOMAIN, {})
+
+    mock_html_path = os.path.join(os.path.dirname(__file__), "mocks", "auth_page.html")
+    await hass.http.async_register_static_paths(
+        [
+            StaticPathConfig(
+                "/auth/authorize",
+                mock_html_path,
+                cache_headers=False,
+            )
+        ]
+    )
+
+    await setup(hass, enable_frontend_changes=True)
+
+    client = await hass_client()
+    resp = await client.get("/auth/authorize", allow_redirects=False)
+    assert resp.status == 200
+    text = await resp.text()
+
+    assert "<script src='/auth/oidc/static/injection.js" in text