main commit

venv/lib/python3.12/site-packages/alembic/testing/assertions.py

@@ -8,7 +8,6 @@ from typing import Dict
 
 from sqlalchemy import exc as sa_exc
 from sqlalchemy.engine import default
-from sqlalchemy.engine import URL
 from sqlalchemy.testing.assertions import _expect_warnings
 from sqlalchemy.testing.assertions import eq_  # noqa
 from sqlalchemy.testing.assertions import is_  # noqa
@@ -18,6 +17,8 @@ from sqlalchemy.testing.assertions import is_true  # noqa
 from sqlalchemy.testing.assertions import ne_  # noqa
 from sqlalchemy.util import decorator
 
+from ..util import sqla_compat
+
 
 def _assert_proper_exception_context(exception):
     """assert that any exception we're catching does not have a __context__
@@ -73,9 +74,7 @@ class _ErrorContainer:
 
 
 @contextlib.contextmanager
-def _expect_raises(
-    except_cls, msg=None, check_context=False, text_exact=False
-):
+def _expect_raises(except_cls, msg=None, check_context=False):
     ec = _ErrorContainer()
     if check_context:
         are_we_already_in_a_traceback = sys.exc_info()[0]
@@ -86,10 +85,7 @@ def _expect_raises(
         ec.error = err
         success = True
         if msg is not None:
-            if text_exact:
-                assert str(err) == msg, f"{msg} != {err}"
-            else:
-                assert re.search(msg, str(err), re.UNICODE), f"{msg} !~ {err}"
+            assert re.search(msg, str(err), re.UNICODE), f"{msg} !~ {err}"
         if check_context and not are_we_already_in_a_traceback:
             _assert_proper_exception_context(err)
         print(str(err).encode("utf-8"))
@@ -102,12 +98,8 @@ def expect_raises(except_cls, check_context=True):
     return _expect_raises(except_cls, check_context=check_context)
 
 
-def expect_raises_message(
-    except_cls, msg, check_context=True, text_exact=False
-):
-    return _expect_raises(
-        except_cls, msg=msg, check_context=check_context, text_exact=text_exact
-    )
+def expect_raises_message(except_cls, msg, check_context=True):
+    return _expect_raises(except_cls, msg=msg, check_context=check_context)
 
 
 def eq_ignore_whitespace(a, b, msg=None):
@@ -126,7 +118,7 @@ def _get_dialect(name):
     if name is None or name == "default":
         return default.DefaultDialect()
     else:
-        d = URL.create(name).get_dialect()()
+        d = sqla_compat._create_url(name).get_dialect()()
 
         if name == "postgresql":
             d.implicit_returning = True
@@ -167,10 +159,6 @@ def emits_python_deprecation_warning(*messages):
     return decorate
 
 
-def expect_deprecated(*messages, **kw):
-    return _expect_warnings(DeprecationWarning, messages, **kw)
-
-
 def expect_sqlalchemy_deprecated(*messages, **kw):
     return _expect_warnings(sa_exc.SADeprecationWarning, messages, **kw)