@@ -222,7 +222,9 @@ async def handle(route):
222
222
await page .route ("**/*/api/v1/_selectielijstklasse-choices/?zaak=*" , handle )
223
223
224
224
async def zaken_are_indexed (self , amount , ** kwargs ):
225
- return await self ._get_or_create_batch (ZaakFactory , amount , ** kwargs )
225
+ return await self ._get_or_create_batch (
226
+ ZaakFactory , amount , with_expand = True , ** kwargs
227
+ )
226
228
227
229
@sync_to_async
228
230
def _orm_get (self , model , ** kwargs ):
@@ -252,7 +254,14 @@ async def _get_or_create(self, factory, **kwargs):
252
254
return await self ._factory_create (factory , ** kwargs )
253
255
254
256
async def _get_or_create_batch (self , factory , amount , ** kwargs ):
255
- queryset = await self ._orm_filter (factory ._meta .model , ** kwargs )
257
+ # Remove any traits of the factory
258
+ orm_params = {
259
+ key : value
260
+ for key , value in kwargs .items ()
261
+ if key not in factory ._meta .parameters
262
+ }
263
+
264
+ queryset = await self ._orm_filter (factory ._meta .model , ** orm_params )
256
265
if queryset :
257
266
return queryset
258
267
return await self ._factory_create_batch (factory , amount , ** kwargs )
0 commit comments