Restore states through a JSON store instead of recorder (#17270)

* Restore states through a JSON store

* Accept entity_id directly in restore state helper

* Keep states stored between runs for a limited time

* Remove warning

tests/helpers/test_storage.py

@@ -1,7 +1,8 @@
 """Tests for the storage helper."""
 import asyncio
 from datetime import timedelta
-from unittest.mock import patch
+import json
+from unittest.mock import patch, Mock
 
 import pytest
 
@@ -31,6 +32,21 @@ async def test_loading(hass, store):
     assert data == MOCK_DATA
 
 
+async def test_custom_encoder(hass):
+    """Test we can save and load data."""
+    class JSONEncoder(json.JSONEncoder):
+        """Mock JSON encoder."""
+
+        def default(self, o):
+            """Mock JSON encode method."""
+            return "9"
+
+    store = storage.Store(hass, MOCK_VERSION, MOCK_KEY, encoder=JSONEncoder)
+    await store.async_save(Mock())
+    data = await store.async_load()
+    assert data == "9"
+
+
 async def test_loading_non_existing(hass, store):
     """Test we can save and load data."""
     with patch('homeassistant.util.json.open', side_effect=FileNotFoundError):