Fix saving the changed data from option flow (#137)

Author: DeerMaximum

custom_components/ta_cmi/config_flow.py

@@ -1,21 +1,21 @@
 """Config flow for Technische Alternative C.M.I. integration."""
 from __future__ import annotations
 
+import time
 from copy import deepcopy
 from datetime import timedelta
-import time
 from typing import Any
 
+import homeassistant.helpers.config_validation as cv
+import voluptuous as vol
 from aiohttp import ClientSession
 from homeassistant import config_entries
 from homeassistant.const import CONF_HOST, CONF_PASSWORD, CONF_USERNAME
 from homeassistant.core import callback
 from homeassistant.data_entry_flow import FlowResult
 from homeassistant.exceptions import HomeAssistantError
 from homeassistant.helpers.aiohttp_client import async_get_clientsession
-import homeassistant.helpers.config_validation as cv
 from ta_cmi import CMI, ApiError, Device, InvalidCredentialsError, RateLimitError
-import voluptuous as vol
 
 from . import custom_sleep
 from .const import (
@@ -349,9 +349,15 @@ async def async_step_init(
                     errors["base"] = "unknown"
                 else:
                     self.data[CONF_HOST] = user_input[CONF_HOST]
-                    return self.async_create_entry(title="", data=self.data)
+                    self.hass.config_entries.async_update_entry(
+                        self.config_entry, data=self.data
+                    )
+                    return self.async_create_entry(title="", data={})
             else:
-                return self.async_create_entry(title="", data=self.data)
+                self.hass.config_entries.async_update_entry(
+                    self.config_entry, data=self.data
+                )
+                return self.async_create_entry(title="", data={})
 
         return self.async_show_form(
             step_id="init",

tests/test_config_flow.py

@@ -572,7 +572,7 @@ async def test_options_flow_init_no_ip(hass: HomeAssistant) -> None:
         )
 
         assert result["type"] == FlowResultType.CREATE_ENTRY
-        assert dict(config_entry.options) == DUMMY_CONFIG_ENTRY_UPDATED
+        assert dict(config_entry.data) == DUMMY_CONFIG_ENTRY_UPDATED
 
 
 @pytest.mark.asyncio
@@ -607,7 +607,7 @@ async def test_options_flow_init(hass: HomeAssistant) -> None:
         )
 
         assert result["type"] == FlowResultType.CREATE_ENTRY
-        assert dict(config_entry.options) == DUMMY_CONFIG_ENTRY_UPDATED_IP
+        assert dict(config_entry.data) == DUMMY_CONFIG_ENTRY_UPDATED_IP
 
 
 @pytest.mark.asyncio
@@ -640,7 +640,7 @@ async def test_options_flow_ip_change_invalid_auth(hass: HomeAssistant) -> None:
         assert result["type"] == FlowResultType.FORM
         assert result["step_id"] == "init"
         assert result["errors"] == {"base": "invalid_auth"}
-        assert dict(config_entry.options) == {}
+        assert dict(config_entry.data) == DUMMY_CONFIG_ENTRY
 
 
 @pytest.mark.asyncio