Lint

Author: erikjohnston

synapse/storage/engines/sqlite.py

@@ -86,6 +86,7 @@ def on_new_connection(self, db_conn: "LoggingDatabaseConnection") -> None:
 
         if self._is_in_memory:
             if self._prepped_conn is not None:
+                assert isinstance(db_conn.conn, sqlite3.Connection)
                 self._prepped_conn.backup(db_conn.conn)
             else:
                 # In memory databases need to be rebuilt each time. Ideally we'd

tests/server.py

@@ -16,6 +16,7 @@
 import logging
 import os
 import os.path
+import sqlite3
 import time
 import uuid
 import warnings
@@ -106,6 +107,8 @@
 # the type of thing that can be passed into `make_request` in the headers list
 CustomHeaderType = Tuple[Union[str, bytes], Union[str, bytes]]
 
+PREPPED_DB_CONN: Optional[LoggingDatabaseConnection] = None
+
 
 class TimedOutException(Exception):
     """
@@ -832,11 +835,6 @@ class TestHomeServer(HomeServer):
     DATASTORE_CLASS = DataStore  # type: ignore[assignment]
 
 
-import sqlite3
-
-PREPPED_DB_CONN = None
-
-
 def setup_test_homeserver(
     cleanup_func: Callable[[Callable[[], None]], None],
     name: str = "test",