Skip to content

Commit 6a9a396

Browse files
authored
Fix qb tests using the wrong model. (#959)
The qb tests use `orm_qb` not `orm`.
1 parent 840845f commit 6a9a396

1 file changed

Lines changed: 16 additions & 16 deletions

File tree

tests/test_qb.py

Lines changed: 16 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ class TestQueryBuilder(tb.ModelTestCase):
3434
def test_implicit_select_01(self):
3535
# Schema Set
3636

37-
from models.orm import default
37+
from models.orm_qb import default
3838

3939
# mypy complains if this doesn't have type annotation???
4040
users: list[default.User] = self.client.query(default.User)
@@ -103,7 +103,7 @@ def test_implicit_select_01(self):
103103
def test_implicit_select_02(self):
104104
# Schema Set + filter
105105

106-
from models.orm import default
106+
from models.orm_qb import default
107107

108108
users = self.client.query(default.User.filter(name="Alice"))
109109
self._assertObjectsWithFields(
@@ -125,7 +125,7 @@ def test_implicit_select_02(self):
125125
def test_implicit_select_03(self):
126126
# Schema Set + Path
127127

128-
from models.orm import default
128+
from models.orm_qb import default
129129

130130
authors = self.client.query(default.Post.author)
131131

@@ -166,7 +166,7 @@ def test_implicit_select_03(self):
166166
def test_implicit_select_04(self):
167167
# Schema Set + Filter + Path
168168

169-
from models.orm import default
169+
from models.orm_qb import default
170170

171171
authors = self.client.query(
172172
default.Post.filter(body="I'm Alice").author
@@ -191,7 +191,7 @@ def test_implicit_select_04(self):
191191
def test_implicit_select_05(self):
192192
# Schema Set + Path + Filter
193193

194-
from models.orm import default
194+
from models.orm_qb import default
195195

196196
authors = self.client.query(default.Post.author.filter(name="Alice"))
197197

@@ -214,7 +214,7 @@ def test_implicit_select_05(self):
214214
def test_implicit_select_06(self):
215215
# Schema Set + Filter + Path + Filter
216216

217-
from models.orm import default
217+
from models.orm_qb import default
218218

219219
authors = self.client.query(
220220
default.Post.filter(body="I'm Alice").author.filter(name="Alice")
@@ -238,7 +238,7 @@ def test_implicit_select_06(self):
238238

239239
def test_implicit_select_07(self):
240240
# Ensure select without a schema path doesn't use a splat
241-
from models.orm import default
241+
from models.orm_qb import default
242242

243243
groups = self.client.query(
244244
default.UserGroup.select(users=lambda x: x.users).filter(
@@ -252,7 +252,7 @@ def test_implicit_select_07(self):
252252

253253
def test_implicit_select_08(self):
254254
# In shape: Schema Set
255-
from models.orm import default
255+
from models.orm_qb import default
256256

257257
groups = self.client.query(
258258
default.UserGroup.select(users=default.User).filter(name="green")
@@ -323,7 +323,7 @@ def test_implicit_select_08(self):
323323

324324
def test_implicit_select_09(self):
325325
# In shape: Schema Set
326-
from models.orm import default
326+
from models.orm_qb import default
327327

328328
groups = self.client.query(
329329
default.UserGroup.select(
@@ -351,7 +351,7 @@ def test_implicit_select_09(self):
351351

352352
def test_implicit_select_10(self):
353353
# In shape: Schema Set + Path
354-
from models.orm import default
354+
from models.orm_qb import default
355355

356356
groups = self.client.query(
357357
default.UserGroup.select(users=default.Post.author).filter(
@@ -397,7 +397,7 @@ def test_implicit_select_10(self):
397397

398398
def test_implicit_select_11(self):
399399
# In shape: Schema Set + Filter + Path
400-
from models.orm import default
400+
from models.orm_qb import default
401401

402402
groups = self.client.query(
403403
default.UserGroup.select(
@@ -425,7 +425,7 @@ def test_implicit_select_11(self):
425425

426426
def test_implicit_select_12(self):
427427
# In shape: Schema Set + Filter + Path
428-
from models.orm import default
428+
from models.orm_qb import default
429429

430430
groups = self.client.query(
431431
default.UserGroup.select(
@@ -453,7 +453,7 @@ def test_implicit_select_12(self):
453453

454454
def test_implicit_select_13(self):
455455
# In shape: Schema Set + Filter + Path
456-
from models.orm import default
456+
from models.orm_qb import default
457457

458458
groups = self.client.query(
459459
default.UserGroup.select(
@@ -1351,21 +1351,21 @@ def test_qb_array_agg_01(self):
13511351

13521352
def test_qb_cast_scalar_01(self):
13531353
# scalar to scalar
1354-
from models.orm import std
1354+
from models.orm_qb import std
13551355

13561356
result = self.client.get(std.str.cast(std.int64(1)))
13571357
self.assertEqual(result, "1")
13581358

13591359
def test_qb_cast_scalar_02(self):
13601360
# enum to scalar
1361-
from models.orm import default, std
1361+
from models.orm_qb import default, std
13621362

13631363
result = self.client.get(std.str.cast(default.Color.Red))
13641364
self.assertEqual(result, "Red")
13651365

13661366
def test_qb_cast_scalar_03(self):
13671367
# scalar to enum
1368-
from models.orm import default, std
1368+
from models.orm_qb import default, std
13691369

13701370
result = self.client.get(default.Color.cast(std.str("Red")))
13711371
self.assertEqual(result, default.Color.Red)

0 commit comments

Comments
 (0)