5
5
6
6
from __future__ import annotations
7
7
8
- import snowflake . connector
8
+ import uuid
9
9
10
10
11
11
def exe0 (cnx , sql ):
12
12
return cnx .cursor ().execute (sql )
13
13
14
14
15
- def _run_autocommit_off (cnx , db_parameters ):
15
+ def _run_autocommit_off (cnx , name ):
16
16
"""Runs autocommit off test.
17
17
18
18
Args:
@@ -21,7 +21,7 @@ def _run_autocommit_off(cnx, db_parameters):
21
21
"""
22
22
23
23
def exe (cnx , sql ):
24
- return cnx .cursor ().execute (sql .format (name = db_parameters [ " name" ] ))
24
+ return cnx .cursor ().execute (sql .format (name = name ))
25
25
26
26
exe (
27
27
cnx ,
@@ -89,7 +89,7 @@ def exe(cnx, sql):
89
89
assert res [0 ] == 2
90
90
91
91
92
- def _run_autocommit_on (cnx , db_parameters ):
92
+ def _run_autocommit_on (cnx , name ):
93
93
"""Run autocommit on test.
94
94
95
95
Args:
@@ -98,7 +98,7 @@ def _run_autocommit_on(cnx, db_parameters):
98
98
"""
99
99
100
100
def exe (cnx , sql ):
101
- return cnx .cursor ().execute (sql .format (name = db_parameters [ " name" ] ))
101
+ return cnx .cursor ().execute (sql .format (name = name ))
102
102
103
103
exe (
104
104
cnx ,
@@ -116,16 +116,18 @@ def exe(cnx, sql):
116
116
assert res [0 ] == 4
117
117
118
118
119
- def test_autocommit_attribute (conn_cnx , db_parameters ):
119
+ def test_autocommit_attribute (conn_cnx ):
120
120
"""Tests autocommit attribute.
121
121
122
122
Args:
123
123
conn_cnx: The database connection context.
124
124
db_parameters: Database parameters.
125
125
"""
126
126
127
+ name = "python_tests_" + str (uuid .uuid4 ()).replace ("-" , "_" )
128
+
127
129
def exe (cnx , sql ):
128
- return cnx .cursor ().execute (sql .format (name = db_parameters [ " name" ] ))
130
+ return cnx .cursor ().execute (sql .format (name = name ))
129
131
130
132
with conn_cnx () as cnx :
131
133
exe (
@@ -136,9 +138,9 @@ def exe(cnx, sql):
136
138
)
137
139
try :
138
140
cnx .autocommit (False )
139
- _run_autocommit_off (cnx , db_parameters )
141
+ _run_autocommit_off (cnx , name )
140
142
cnx .autocommit (True )
141
- _run_autocommit_on (cnx , db_parameters )
143
+ _run_autocommit_on (cnx , name )
142
144
finally :
143
145
exe (
144
146
cnx ,
@@ -148,25 +150,18 @@ def exe(cnx, sql):
148
150
)
149
151
150
152
151
- def test_autocommit_parameters (db_parameters ):
153
+ def test_autocommit_parameters (conn_cnx ):
152
154
"""Tests autocommit parameter.
153
155
154
156
Args:
155
157
db_parameters: Database parameters.
156
158
"""
159
+ name = "python_tests_" + str (uuid .uuid4 ()).replace ("-" , "_" )
157
160
158
161
def exe (cnx , sql ):
159
- return cnx .cursor ().execute (sql .format (name = db_parameters ["name" ]))
160
-
161
- with snowflake .connector .connect (
162
- user = db_parameters ["user" ],
163
- password = db_parameters ["password" ],
164
- host = db_parameters ["host" ],
165
- port = db_parameters ["port" ],
166
- account = db_parameters ["account" ],
167
- protocol = db_parameters ["protocol" ],
168
- schema = db_parameters ["schema" ],
169
- database = db_parameters ["database" ],
162
+ return cnx .cursor ().execute (sql .format (name = name ))
163
+
164
+ with conn_cnx (
170
165
autocommit = False ,
171
166
) as cnx :
172
167
exe (
@@ -175,20 +170,12 @@ def exe(cnx, sql):
175
170
CREATE TABLE {name} (c1 boolean)
176
171
""" ,
177
172
)
178
- _run_autocommit_off (cnx , db_parameters )
179
-
180
- with snowflake .connector .connect (
181
- user = db_parameters ["user" ],
182
- password = db_parameters ["password" ],
183
- host = db_parameters ["host" ],
184
- port = db_parameters ["port" ],
185
- account = db_parameters ["account" ],
186
- protocol = db_parameters ["protocol" ],
187
- schema = db_parameters ["schema" ],
188
- database = db_parameters ["database" ],
173
+ _run_autocommit_off (cnx , name )
174
+
175
+ with conn_cnx (
189
176
autocommit = True ,
190
177
) as cnx :
191
- _run_autocommit_on (cnx , db_parameters )
178
+ _run_autocommit_on (cnx , name )
192
179
exe (
193
180
cnx ,
194
181
"""
0 commit comments