summaryrefslogtreecommitdiff
path: root/test/engine/test_reconnect.py
diff options
context:
space:
mode:
Diffstat (limited to 'test/engine/test_reconnect.py')
-rw-r--r--test/engine/test_reconnect.py4
1 files changed, 4 insertions, 0 deletions
diff --git a/test/engine/test_reconnect.py b/test/engine/test_reconnect.py
index 619319693..39ebcc91b 100644
--- a/test/engine/test_reconnect.py
+++ b/test/engine/test_reconnect.py
@@ -370,6 +370,9 @@ class MockReconnectTest(fixtures.TestBase):
mock_dialect = Mock()
class MyURL(URL):
+ def _get_entrypoint(self):
+ return Dialect
+
def get_dialect(self):
return Dialect
@@ -420,6 +423,7 @@ class CursorErrTest(fixtures.TestBase):
from sqlalchemy.engine import default
url = Mock(
get_dialect=lambda: default.DefaultDialect,
+ _get_entrypoint=lambda: default.DefaultDialect,
translate_connect_args=lambda: {}, query={},)
eng = testing_engine(
url, options=dict(module=dbapi, _initialize=initialize))