working on inspection

Author: fgregg

pytest_flask_sqlalchemy/fixtures.py

@@ -93,7 +93,10 @@ def _engine(pytestconfig, request, _transaction, mocker):
 
     # Make sure that any attempts to call `connect()` simply return a
     # reference to the open connection
-    engine = mocker.MagicMock(spec=sa.engine.Engine)
+    class EngineMock(mocker.MagickMock):
+        pass
+
+    engine = EngineMock(spec=sa.engine.Engine)
 
     engine.connect.return_value = connection
 
@@ -107,7 +110,7 @@ def _engine(pytestconfig, request, _transaction, mocker):
 
     # Calls to execution_options should return this mocked engine
     engine.execution_options.return_value = engine
-    sa.inspection._self_inspects(engine)
+    sa.inspection._registrar[EngineMock] = sa.engine.Inspector._engine_insp
 
     # References to `Engine.dialect` should redirect to the Connection (this
     # is primarily useful for the `autoload` flag in SQLAlchemy, which references