Ensure connection retrieved by ID is of type ConnectionBase

datalens-tech · Dec 24, 2024 · dee3ca5 · dee3ca5
1 parent 21be0d8
commit dee3ca5
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/lib/dl_core_testing/dl_core_testing/connection.py b/lib/dl_core_testing/dl_core_testing/connection.py
@@ -84,7 +84,9 @@ def make_saved_connection(
     if conn_id is None:
         raise ValueError("Connection ID is None")
 
-    return sync_usm.get_by_id(conn_id)
+    conn = sync_usm.get_by_id(conn_id)
+    assert isinstance(conn, ConnectionBase)  # for typing
+    return conn
 
 
 def make_saved_connection_from_db(