@@ -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