diff --git a/pymongosql/__init__.py b/pymongosql/__init__.py index d8940b9..c7b0e18 100644 --- a/pymongosql/__init__.py +++ b/pymongosql/__init__.py @@ -6,7 +6,7 @@ if TYPE_CHECKING: from .connection import Connection -__version__: str = "0.4.4" +__version__: str = "0.4.5" # Globals https://www.python.org/dev/peps/pep-0249/#globals apilevel: str = "2.0" diff --git a/pyproject.toml b/pyproject.toml index d23eb3a..5857c80 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -58,6 +58,7 @@ Documentation = "https://github.com/passren/PyMongoSQL/wiki" [project.entry-points."sqlalchemy.dialects"] mongodb = "pymongosql.sqlalchemy_mongodb.sqlalchemy_dialect:PyMongoSQLDialect" "mongodb+srv" = "pymongosql.sqlalchemy_mongodb.sqlalchemy_dialect:PyMongoSQLDialect" +"mongodb.srv" = "pymongosql.sqlalchemy_mongodb.sqlalchemy_dialect:PyMongoSQLDialect" [tool.black] line-length = 120 diff --git a/tests/test_sqlalchemy_dialect.py b/tests/test_sqlalchemy_dialect.py index 91a0530..1954193 100644 --- a/tests/test_sqlalchemy_dialect.py +++ b/tests/test_sqlalchemy_dialect.py @@ -658,3 +658,14 @@ def test_dialect_registration(self): except ImportError: # Skip if SQLAlchemy registry is not available self.skipTest("SQLAlchemy registry not available") + + def test_srv_dialect_lookup(self): + """Test that mongodb.srv resolves correctly (mongodb+srv:// URLs).""" + if not HAS_SQLALCHEMY: + self.skipTest("SQLAlchemy not available") + + from sqlalchemy.dialects import registry + + loaded = registry.load("mongodb.srv") + self.assertIsNotNone(loaded) + self.assertEqual(loaded.name, "mongodb")