1212# See the License for the specific language governing permissions and
1313# limitations under the License.
1414
15- from sqlalchemy import create_engine
1615from sqlalchemy .orm import Session
1716from sqlalchemy .testing import eq_ , is_instance_of
1817from google .cloud .spanner_v1 import (
19- FixedSizePool ,
2018 ResultSet ,
21- BatchCreateSessionsRequest ,
19+ CreateSessionRequest ,
2220 ExecuteSqlRequest ,
2321 CommitRequest ,
2422 BeginTransactionRequest ,
@@ -45,10 +43,7 @@ def test_create_table(self):
4543""" ,
4644 ResultSet (),
4745 )
48- engine = create_engine (
49- "spanner:///projects/p/instances/i/databases/d" ,
50- connect_args = {"client" : self .client , "pool" : FixedSizePool (size = 10 )},
51- )
46+ engine = self .create_engine ()
5247 Base .metadata .create_all (engine )
5348 requests = self .database_admin_service .requests
5449 eq_ (1 , len (requests ))
@@ -74,10 +69,7 @@ def test_create_auto_increment_table(self):
7469""" ,
7570 ResultSet (),
7671 )
77- engine = create_engine (
78- "spanner:///projects/p/instances/i/databases/d" ,
79- connect_args = {"client" : self .client , "pool" : FixedSizePool (size = 10 )},
80- )
72+ engine = self .create_engine ()
8173 engine .dialect .use_auto_increment = True
8274 Base .metadata .create_all (engine )
8375 requests = self .database_admin_service .requests
@@ -103,10 +95,7 @@ def test_create_table_with_specific_sequence_kind(self):
10395""" ,
10496 ResultSet (),
10597 )
106- engine = create_engine (
107- "spanner:///projects/p/instances/i/databases/d" ,
108- connect_args = {"client" : self .client , "pool" : FixedSizePool (size = 10 )},
109- )
98+ engine = self .create_engine ()
11099 engine .dialect .default_sequence_kind = "non_existing_kind"
111100 Base .metadata .create_all (engine )
112101 requests = self .database_admin_service .requests
@@ -126,10 +115,7 @@ def test_insert_row(self):
126115 from test .mockserver_tests .auto_increment_model import Singer
127116
128117 self .add_insert_result ("INSERT INTO singers (name) VALUES (@a0) THEN RETURN id" )
129- engine = create_engine (
130- "spanner:///projects/p/instances/i/databases/d" ,
131- connect_args = {"client" : self .client , "pool" : FixedSizePool (size = 10 )},
132- )
118+ engine = self .create_engine ()
133119
134120 with Session (engine ) as session :
135121 singer = Singer (name = "Test" )
@@ -141,7 +127,7 @@ def test_insert_row(self):
141127 # Verify the requests that we got.
142128 requests = self .spanner_service .requests
143129 eq_ (4 , len (requests ))
144- is_instance_of (requests [0 ], BatchCreateSessionsRequest )
130+ is_instance_of (requests [0 ], CreateSessionRequest )
145131 is_instance_of (requests [1 ], BeginTransactionRequest )
146132 is_instance_of (requests [2 ], ExecuteSqlRequest )
147133 is_instance_of (requests [3 ], CommitRequest )
@@ -152,10 +138,7 @@ def test_insert_row_with_pk_value(self):
152138 # SQLAlchemy should not use a THEN RETURN clause when a value for the
153139 # primary key has been set on the model.
154140 add_update_count ("INSERT INTO singers (id, name) VALUES (@a0, @a1)" , 1 )
155- engine = create_engine (
156- "spanner:///projects/p/instances/i/databases/d" ,
157- connect_args = {"client" : self .client , "pool" : FixedSizePool (size = 10 )},
158- )
141+ engine = self .create_engine ()
159142
160143 with Session (engine ) as session :
161144 # Manually specify a value for the primary key.
0 commit comments