Commit 7ba5f5e 1 parent 6de5a8a commit 7ba5f5e Copy full SHA for 7ba5f5e
File tree 1 file changed +16
-2
lines changed
1 file changed +16
-2
lines changed Original file line number Diff line number Diff line change 12
12
import os
13
13
import io
14
14
import time
15
+ from typing import Any
15
16
import arrow
16
17
import celery
17
18
import shutil
@@ -494,12 +495,25 @@ def format_items(items):
494
495
return ",\n " .join (output )
495
496
496
497
498
+ async def delete_entries_for (context , resource : str ) -> None :
499
+ """
500
+ Attempts to remove all items from the resources MongoDB and/or Elastic.
501
+ First tries with async, otherwise it falls back to sync resources.
502
+ """
503
+
504
+ async_app = context .app .async_app
505
+
506
+ try :
507
+ await async_app .resources .get_resource_service (resource ).delete_many ({})
508
+ except KeyError :
509
+ get_resource_service (resource ).delete_action ()
510
+
511
+
497
512
@given ('empty "{resource}"' )
498
513
@async_run_until_complete
499
514
async def step_impl_given_empty (context , resource ):
500
515
if not is_user_resource (resource ):
501
- async with context .app .test_request_context (context .app .config ["URL_PREFIX" ]):
502
- get_resource_service (resource ).delete_action ()
516
+ await delete_entries_for (context , resource )
503
517
504
518
505
519
@given ('"{resource}"' )
You can’t perform that action at this time.
0 commit comments