chore: add region tags to connection code in tests (#113)

* chore: add region tags to connection code in tests

* linting

Author: shubha-rajan

tests/system/test_pg8000_connection.py

@@ -25,6 +25,9 @@
 table_name = f"books_{uuid.uuid4().hex}"
 
 
+# [START cloud_sql_connector_postgres_pg8000]
+# The Cloud SQL Python Connector can be used along with SQLAlchemy using the
+# 'creator' argument to 'create_engine'
 def init_connection_engine() -> sqlalchemy.engine.Engine:
     def getconn() -> pg8000.dbapi.Connection:
         conn: pg8000.dbapi.Connection = connector.connect(
@@ -44,6 +47,9 @@ def getconn() -> pg8000.dbapi.Connection:
     return engine
 
 
+# [END cloud_sql_connector_postgres_pg8000]
+
+
 @pytest.fixture(name="pool")
 def setup() -> Generator:
     pool = init_connection_engine()

tests/system/test_pymysql_connection.py

@@ -25,6 +25,9 @@
 table_name = f"books_{uuid.uuid4().hex}"
 
 
+# [START cloud_sql_connector_mysql_pymysql]
+# The Cloud SQL Python Connector can be used along with SQLAlchemy using the
+# 'creator' argument to 'create_engine'
 def init_connection_engine() -> sqlalchemy.engine.Engine:
     def getconn() -> pymysql.connections.Connection:
         conn: pymysql.connections.Connection = connector.connect(
@@ -43,6 +46,9 @@ def getconn() -> pymysql.connections.Connection:
     return engine
 
 
+# [END cloud_sql_connector_mysql_pymysql]
+
+
 @pytest.fixture(name="pool")
 def setup() -> Generator:
     pool = init_connection_engine()

tests/system/test_pytds_connection.py

@@ -25,6 +25,10 @@
 table_name = f"books_{uuid.uuid4().hex}"
 
 
+# [START cloud_sql_connector_mysql_pytds]
+# The Cloud SQL Python Connector can be used along with SQLAlchemy using the
+# 'creator' argument to 'create_engine'. To use SQLAlchemy with pytds, include
+# 'sqlalchemy-pytds` in your dependencies
 def init_connection_engine() -> sqlalchemy.engine.Engine:
     def getconn() -> pytds.Connection:
         conn = connector.connect(
@@ -44,6 +48,9 @@ def getconn() -> pytds.Connection:
     return engine
 
 
+# [END cloud_sql_connector_mysql_pytds]
+
+
 @pytest.fixture(name="pool")
 def setup() -> Generator:
     pool = init_connection_engine()