Skip to content

Commit 5214fe5

Browse files
committed
fix(sqlserver): add profile fixtures and pydantic validated model to query and basic providers
- Add setup_profile_fixtures and setup_async_profile_fixtures to QueryProvider - Add profiles to cleanup table lists
1 parent dca3f58 commit 5214fe5

1 file changed

Lines changed: 16 additions & 2 deletions

File tree

‎tests/providers/query.py‎

Lines changed: 16 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -235,6 +235,13 @@ def setup_mapped_models(self, scenario_name: str) -> Tuple[Type[ActiveRecord], T
235235
(MappedComment, "comments")
236236
], scenario_name)
237237

238+
def setup_profile_fixtures(self, scenario_name: str) -> Tuple[Type[ActiveRecord], Type[ActiveRecord]]:
239+
Profile = User.get_relation('profile').get_related_model(User)
240+
return self._setup_multiple_models([
241+
(User, "users"),
242+
(Profile, "profiles"),
243+
], scenario_name)
244+
238245
async def setup_async_order_fixtures(self, scenario_name: str) -> Tuple[Type[ActiveRecord], Type[ActiveRecord], Type[ActiveRecord]]:
239246
from rhosocial.activerecord.testsuite.feature.query.fixtures.async_models import AsyncUser, AsyncOrder, AsyncOrderItem
240247
return await self._setup_multiple_models_async([
@@ -302,6 +309,13 @@ async def setup_async_mapped_models(self, scenario_name: str) -> Tuple[Type[Acti
302309
(AsyncMappedComment, "comments")
303310
], scenario_name)
304311

312+
async def setup_async_profile_fixtures(self, scenario_name: str) -> Tuple[Type[ActiveRecord], Type[ActiveRecord]]:
313+
from rhosocial.activerecord.testsuite.feature.query.fixtures.async_models import AsyncUser, AsyncProfile
314+
return await self._setup_multiple_models_async([
315+
(AsyncUser, "users"),
316+
(AsyncProfile, "profiles"),
317+
], scenario_name)
318+
305319
async def cleanup_after_test_async(self, scenario_name: str) -> None:
306320
from rhosocial.activerecord.backend.impl.sqlserver import AsyncSQLServerBackend
307321
from rhosocial.activerecord.backend.expression.statements import DropTableExpression
@@ -311,7 +325,7 @@ async def cleanup_after_test_async(self, scenario_name: str) -> None:
311325
if not is_async:
312326
continue
313327
try:
314-
for table_name in ['comments', 'order_items', 'posts', 'orders', 'users', 'json_users', 'nodes', 'extended_order_items', 'extended_orders', 'searchable_items']:
328+
for table_name in ['comments', 'order_items', 'posts', 'orders', 'users', 'json_users', 'nodes', 'extended_order_items', 'extended_orders', 'searchable_items', 'profiles']:
315329
try:
316330
expr = DropTableExpression(backend_instance.dialect, table_name, if_exists=True)
317331
await backend_instance.execute(*expr.to_sql())
@@ -336,7 +350,7 @@ def cleanup_after_test(self, scenario_name: str):
336350
from rhosocial.activerecord.backend.expression.statements import DropTableExpression
337351
for backend_instance in self._active_backends:
338352
try:
339-
for table_name in ['comments', 'order_items', 'posts', 'orders', 'users', 'json_users', 'nodes', 'extended_order_items', 'extended_orders', 'searchable_items']:
353+
for table_name in ['comments', 'order_items', 'posts', 'orders', 'users', 'json_users', 'nodes', 'extended_order_items', 'extended_orders', 'searchable_items', 'profiles']:
340354
try:
341355
expr = DropTableExpression(backend_instance.dialect, table_name, if_exists=True)
342356
backend_instance.execute(*expr.to_sql())

0 commit comments

Comments
 (0)