Teach sqlite3 about HAFakeDatetime (#78756)

pull/78780/head
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
1 changed files with 6 additions and 0 deletions

View File

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