Teach sqlite3 about HAFakeDatetime (#78756)

This commit is contained in:
Erik Montnemery 2022-09-19 23:09:50 +02:00 committed by GitHub
parent f07204ba55
commit e66f28f3f7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -7,6 +7,7 @@ from contextlib import asynccontextmanager
import functools import functools
from json import JSONDecoder, loads from json import JSONDecoder, loads
import logging import logging
import sqlite3
import ssl import ssl
import threading import threading
from typing import Any from typing import Any
@ -104,6 +105,11 @@ def pytest_runtest_setup():
freezegun.api.datetime_to_fakedatetime = ha_datetime_to_fakedatetime freezegun.api.datetime_to_fakedatetime = ha_datetime_to_fakedatetime
freezegun.api.FakeDatetime = HAFakeDatetime freezegun.api.FakeDatetime = HAFakeDatetime
def adapt_datetime(val):
return val.isoformat(" ")
sqlite3.register_adapter(HAFakeDatetime, adapt_datetime)
def ha_datetime_to_fakedatetime(datetime): def ha_datetime_to_fakedatetime(datetime):
"""Convert datetime to FakeDatetime. """Convert datetime to FakeDatetime.