Add config flow for cloudflare (#41167)

* add config flow for cloudflare * Create const.py * work on flow. * remove const. * lint. * Apply suggestions from code review Co-authored-by: J. Nick Koston <nick@koston.org> * Update config_flows.py * Update homeassistant/components/cloudflare/strings.json * Apply suggestions from code review * Apply suggestions from code review * Apply suggestions from code review * Update strings.json * Apply suggestions from code review * Update __init__.py * Update __init__.py Co-authored-by: J. Nick Koston <nick@koston.org>
2025-12-25 05:26:47 +00:00 · 2020-10-21 09:09:00 -05:00
parent e203896638
commit d8577a1550
14 changed files with 692 additions and 49 deletions
--- a/tests/components/cloudflare/init.py
+++ b/tests/components/cloudflare/init.py
@@ -0,0 +1,101 @@
+"""Tests for the Cloudflare integration."""
+from typing import List
+
+from pycfdns import CFRecord
+
+from homeassistant.components.cloudflare.const import CONF_RECORDS, DOMAIN
+from homeassistant.const import CONF_API_TOKEN, CONF_ZONE
+
+from tests.async_mock import AsyncMock, patch
+from tests.common import MockConfigEntry
+
+ENTRY_CONFIG = {
+    CONF_API_TOKEN: "mock-api-token",
+    CONF_ZONE: "mock.com",
+    CONF_RECORDS: ["ha.mock.com", "homeassistant.mock.com"],
+}
+
+ENTRY_OPTIONS = {}
+
+USER_INPUT = {
+    CONF_API_TOKEN: "mock-api-token",
+}
+
+USER_INPUT_ZONE = {CONF_ZONE: "mock.com"}
+
+USER_INPUT_RECORDS = {CONF_RECORDS: ["ha.mock.com", "homeassistant.mock.com"]}
+
+MOCK_ZONE = "mock.com"
+MOCK_ZONE_ID = "mock-zone-id"
+MOCK_ZONE_RECORDS = [
+    {
+        "id": "zone-record-id",
+        "type": "A",
+        "name": "ha.mock.com",
+        "proxied": True,
+        "content": "127.0.0.1",
+    },
+    {
+        "id": "zone-record-id-2",
+        "type": "A",
+        "name": "homeassistant.mock.com",
+        "proxied": True,
+        "content": "127.0.0.1",
+    },
+    {
+        "id": "zone-record-id-3",
+        "type": "A",
+        "name": "mock.com",
+        "proxied": True,
+        "content": "127.0.0.1",
+    },
+]
+
+
+async def init_integration(
+    hass,
+    *,
+    data: dict = ENTRY_CONFIG,
+    options: dict = ENTRY_OPTIONS,
+) -> MockConfigEntry:
+    """Set up the Cloudflare integration in Home Assistant."""
+    entry = MockConfigEntry(domain=DOMAIN, data=data, options=options)
+    entry.add_to_hass(hass)
+
+    await hass.config_entries.async_setup(entry.entry_id)
+    await hass.async_block_till_done()
+
+    return entry
+
+
+def _get_mock_cfupdate(
+    zone: str = MOCK_ZONE,
+    zone_id: str = MOCK_ZONE_ID,
+    records: List = MOCK_ZONE_RECORDS,
+):
+    client = AsyncMock()
+
+    zone_records = [record["name"] for record in records]
+    cf_records = [CFRecord(record) for record in records]
+
+    client.get_zones = AsyncMock(return_value=[zone])
+    client.get_zone_records = AsyncMock(return_value=zone_records)
+    client.get_record_info = AsyncMock(return_value=cf_records)
+    client.get_zone_id = AsyncMock(return_value=zone_id)
+    client.update_records = AsyncMock(return_value=None)
+
+    return client
+
+
+def _patch_async_setup(return_value=True):
+    return patch(
+        "homeassistant.components.cloudflare.async_setup",
+        return_value=return_value,
+    )
+
+
+def _patch_async_setup_entry(return_value=True):
+    return patch(
+        "homeassistant.components.cloudflare.async_setup_entry",
+        return_value=return_value,
+    )