28
28
from oslo_utils import timeutils
29
29
from oslo_utils import uuidutils
30
30
from oslotest import output
31
- import sqlalchemy
31
+ import sqlalchemy as sa
32
32
import testtools
33
33
34
34
from nova .compute import rpcapi as compute_rpcapi
@@ -125,17 +125,19 @@ def test_fixture_reset(self):
125
125
self .useFixture (db_fixture )
126
126
engine = main_db_api .get_engine ()
127
127
conn = engine .connect ()
128
- result = conn .execute ("select * from instance_types" )
128
+ result = conn .execute (sa . text ( "SELECT * FROM instance_types") )
129
129
rows = result .fetchall ()
130
130
self .assertEqual (0 , len (rows ), "Rows %s" % rows )
131
131
132
132
# insert a 6th instance type, column 5 below is an int id
133
133
# which has a constraint on it, so if new standard instance
134
134
# types are added you have to bump it.
135
- conn .execute ("insert into instance_types VALUES "
136
- "(NULL, NULL, NULL, 't1.test', 6, 4096, 2, 0, NULL, '87'"
137
- ", 1.0, 40, 0, 0, 1, 0)" )
138
- result = conn .execute ("select * from instance_types" )
135
+ conn .execute (sa .text (
136
+ "INSERT INTO instance_types VALUES "
137
+ "(NULL, NULL, NULL, 't1.test', 6, 4096, 2, 0, NULL, '87'"
138
+ ", 1.0, 40, 0, 0, 1, 0)"
139
+ ))
140
+ result = conn .execute (sa .text ("SELECT * FROM instance_types" ))
139
141
rows = result .fetchall ()
140
142
self .assertEqual (1 , len (rows ), "Rows %s" % rows )
141
143
@@ -145,7 +147,7 @@ def test_fixture_reset(self):
145
147
db_fixture .reset ()
146
148
engine = main_db_api .get_engine ()
147
149
conn = engine .connect ()
148
- result = conn .execute ("select * from instance_types" )
150
+ result = conn .execute (sa . text ( "SELECT * FROM instance_types") )
149
151
rows = result .fetchall ()
150
152
self .assertEqual (0 , len (rows ), "Rows %s" % rows )
151
153
@@ -156,14 +158,19 @@ def test_api_fixture_reset(self):
156
158
self .useFixture (db_fixture )
157
159
engine = api_db_api .get_engine ()
158
160
conn = engine .connect ()
159
- result = conn .execute ("select * from cell_mappings" )
161
+ result = conn .execute (sa . text ( "SELECT * FROM cell_mappings") )
160
162
rows = result .fetchall ()
161
163
self .assertEqual (0 , len (rows ), "Rows %s" % rows )
162
164
163
165
uuid = uuidutils .generate_uuid ()
164
- conn .execute ("insert into cell_mappings (uuid, name) VALUES "
165
- "('%s', 'fake-cell')" % (uuid ,))
166
- result = conn .execute ("select * from cell_mappings" )
166
+ conn .execute (
167
+ sa .text (
168
+ "INSERT INTO cell_mappings (uuid, name) VALUES (:uuid, :name)"
169
+ ),
170
+ uuid = uuid ,
171
+ name = 'fake-cell' ,
172
+ )
173
+ result = conn .execute (sa .text ("SELECT * FROM cell_mappings" ))
167
174
rows = result .fetchall ()
168
175
self .assertEqual (1 , len (rows ), "Rows %s" % rows )
169
176
@@ -173,7 +180,7 @@ def test_api_fixture_reset(self):
173
180
db_fixture .reset ()
174
181
engine = api_db_api .get_engine ()
175
182
conn = engine .connect ()
176
- result = conn .execute ("select * from cell_mappings" )
183
+ result = conn .execute (sa . text ( "SELECT * FROM cell_mappings") )
177
184
rows = result .fetchall ()
178
185
self .assertEqual (0 , len (rows ), "Rows %s" % rows )
179
186
@@ -202,9 +209,14 @@ def test_api_fixture_cleanup(self):
202
209
engine = api_db_api .get_engine ()
203
210
conn = engine .connect ()
204
211
uuid = uuidutils .generate_uuid ()
205
- conn .execute ("insert into cell_mappings (uuid, name) VALUES "
206
- "('%s', 'fake-cell')" % (uuid ,))
207
- result = conn .execute ("select * from cell_mappings" )
212
+ conn .execute (
213
+ sa .text (
214
+ "INSERT INTO cell_mappings (uuid, name) VALUES (:uuid, :name)"
215
+ ),
216
+ uuid = uuid ,
217
+ name = 'fake-cell' ,
218
+ )
219
+ result = conn .execute (sa .text ("SELECT * FROM cell_mappings" ))
208
220
rows = result .fetchall ()
209
221
self .assertEqual (1 , len (rows ), "Rows %s" % rows )
210
222
@@ -227,13 +239,13 @@ def test_flavors(self, mock_send_notification):
227
239
228
240
engine = api_db_api .get_engine ()
229
241
conn = engine .connect ()
230
- result = conn .execute ("select * from flavors" )
242
+ result = conn .execute (sa . text ( "SELECT * FROM flavors") )
231
243
rows = result .fetchall ()
232
244
self .assertEqual (0 , len (rows ), "Rows %s" % rows )
233
245
234
246
self .useFixture (fixtures .DefaultFlavorsFixture ())
235
247
236
- result = conn .execute ("select * from flavors" )
248
+ result = conn .execute (sa . text ( "SELECT * FROM flavors") )
237
249
rows = result .fetchall ()
238
250
self .assertEqual (6 , len (rows ), "Rows %s" % rows )
239
251
@@ -323,15 +335,15 @@ def test_submit_passes_through(self):
323
335
324
336
class TestBannedDBSchemaOperations (testtools .TestCase ):
325
337
def test_column (self ):
326
- column = sqlalchemy .Column ()
338
+ column = sa .Column ()
327
339
with fixtures .BannedDBSchemaOperations (['Column' ]):
328
340
self .assertRaises (exception .DBNotAllowed ,
329
341
column .drop )
330
342
self .assertRaises (exception .DBNotAllowed ,
331
343
column .alter )
332
344
333
345
def test_table (self ):
334
- table = sqlalchemy .Table ()
346
+ table = sa .Table ()
335
347
with fixtures .BannedDBSchemaOperations (['Table' ]):
336
348
self .assertRaises (exception .DBNotAllowed ,
337
349
table .drop )
0 commit comments