Skip to content

Fix qb tests using the wrong model. #959

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
dnwpark merged 1 commit into from
Oct 24, 2025
Merged

Fix qb tests using the wrong model. #959

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
32 changes: 16 additions & 16 deletions tests/test_qb.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -34,7 +34,7 @@ class TestQueryBuilder(tb.ModelTestCase):
def test_implicit_select_01(self):
# Schema Set

from models.orm import default
from models.orm_qb import default

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

from models.orm import default
from models.orm_qb import default

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

from models.orm import default
from models.orm_qb import default

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

Expand Down Expand Up @@ -166,7 +166,7 @@ def test_implicit_select_03(self):
def test_implicit_select_04(self):
# Schema Set + Filter + Path

from models.orm import default
from models.orm_qb import default

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

from models.orm import default
from models.orm_qb import default

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

Expand All @@ -214,7 +214,7 @@ def test_implicit_select_05(self):
def test_implicit_select_06(self):
# Schema Set + Filter + Path + Filter

from models.orm import default
from models.orm_qb import default

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

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

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

def test_implicit_select_08(self):
# In shape: Schema Set
from models.orm import default
from models.orm_qb import default

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

def test_implicit_select_09(self):
# In shape: Schema Set
from models.orm import default
from models.orm_qb import default

groups = self.client.query(
default.UserGroup.select(
Expand Down Expand Up @@ -351,7 +351,7 @@ def test_implicit_select_09(self):

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

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

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

groups = self.client.query(
default.UserGroup.select(
Expand Down Expand Up @@ -425,7 +425,7 @@ def test_implicit_select_11(self):

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

groups = self.client.query(
default.UserGroup.select(
Expand Down Expand Up @@ -453,7 +453,7 @@ def test_implicit_select_12(self):

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

groups = self.client.query(
default.UserGroup.select(
Expand Down Expand Up @@ -1351,21 +1351,21 @@ def test_qb_array_agg_01(self):

def test_qb_cast_scalar_01(self):
# scalar to scalar
from models.orm import std
from models.orm_qb import std

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

def test_qb_cast_scalar_02(self):
# enum to scalar
from models.orm import default, std
from models.orm_qb import default, std

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

def test_qb_cast_scalar_03(self):
# scalar to enum
from models.orm import default, std
from models.orm_qb import default, std

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