Add test cases for version injection

Author: passren

pymongosql/connection.py

@@ -11,6 +11,7 @@
 from pymongo.driver_info import DriverInfo
 from pymongo.errors import ConnectionFailure
 
+from . import __version__
 from .common import BaseCursor
 from .cursor import Cursor
 from .error import DatabaseError, OperationalError
@@ -20,7 +21,7 @@
 try:
     _VERSION = _get_version("pymongosql")
 except Exception:
-    _VERSION = None
+    _VERSION = __version__
 
 _DRIVER_INFO = DriverInfo(name="PyMongoSQL", version=_VERSION)
 

tests/test_connection.py

@@ -1,7 +1,7 @@
 # -*- coding: utf-8 -*-
 import pytest
 
-from pymongosql.connection import Connection
+from pymongosql.connection import _VERSION, Connection
 from pymongosql.cursor import Cursor
 from pymongosql.error import OperationalError
 from tests.conftest import TEST_DB, TEST_URI
@@ -26,6 +26,12 @@ def test_connection_init_with_basic_params(self, conn):
             assert conn.port == 27017
             assert conn.database_name == "test_db"
             assert conn.is_connected
+
+        # Verify driver name and version are correctly configured
+        assert _VERSION is not None
+        assert conn._pymongo_params.get("driver").name == "PyMongoSQL"
+        assert conn._pymongo_params.get("driver").version == _VERSION
+
         conn.close()
 
     def test_connection_with_connect_false(self):