refactor: Connection now stores a direct reference to its Pool object instead of just the pool ID.

Author: bhatt4982

spannerlib/wrappers/spannerlib-python/spannerlib-python/google/cloud/spannerlib/connection.py

@@ -41,20 +41,20 @@ def __init__(self, oid: int, pool: "Pool") -> None:
                 created.
         """
         super().__init__(pool.spannerlib, oid)
-        self._pool_id = pool.oid
+        self._pool = pool
 
     @property
-    def pool_id(self) -> int:
-        """Returns the pool ID associated with this connection."""
-        return self._pool_id
+    def pool(self) -> "Pool":
+        """Returns the pool associated with this connection."""
+        return self._pool
 
     def _close_lib_object(self) -> None:
         """Internal method to close the pool in the Go library."""
         try:
             logger.info("Closing connection ID: %d", self.oid)
             # Call the Go library function to close the connection.
             with self.spannerlib.close_connection(
-                self.pool_id, self.oid
+                self.pool.oid, self.oid
             ) as msg:
                 msg.bind_library(self.spannerlib)
                 msg.raise_if_error()

spannerlib/wrappers/spannerlib-python/spannerlib-python/tests/unit/test_connection.py

@@ -92,14 +92,14 @@ def test_initialization(self, mock_pool, mock_spanner_lib):
         assert conn.oid == 50
 
         # Verify specific connection attributes
-        assert conn.pool_id == 999
+        assert conn.pool.oid == 999
 
-    def test_pool_id_property_is_read_only(self, connection):
-        """Ensure pool_id cannot be overwritten accidentally."""
-        assert connection.pool_id == 999
+    def test_pool_property_is_read_only(self, connection):
+        """Ensure pool cannot be overwritten accidentally."""
+        assert connection.pool.oid == 999
 
         with pytest.raises(AttributeError):
-            connection.pool_id = 888
+            connection.pool = None
 
     # -------------------------------------------------------------------------
     # Test Methods: Lifecycle & Cleanup

spannerlib/wrappers/spannerlib-python/spannerlib-python/tests/unit/test_pool.py

@@ -208,7 +208,7 @@ def test_create_connection_success(
         mock_lib_instance.create_connection.assert_called_once_with(100)
         assert isinstance(conn, Connection)
         assert conn.oid == 200
-        assert conn.pool_id == 100
+        assert conn.pool.oid == 100
         assert conn.spannerlib == mock_lib_instance
 
     def test_create_connection_pool_closed(self, mock_lib_instance):