diff --git a/backend/btrixcloud/basecrawls.py b/backend/btrixcloud/basecrawls.py index 9435a620..7837aafb 100644 --- a/backend/btrixcloud/basecrawls.py +++ b/backend/btrixcloud/basecrawls.py @@ -486,6 +486,7 @@ def init_base_crawls_api(app, mdb, users, crawl_manager, orgs, user_dep): name: Optional[str] = None, state: Optional[str] = None, description: Optional[str] = None, + collectionId: Optional[UUID4] = None, sortBy: Optional[str] = "finished", sortDirection: Optional[int] = -1, ): @@ -495,6 +496,7 @@ def init_base_crawls_api(app, mdb, users, crawl_manager, orgs, user_dep): userid=userid, name=name, description=description, + collection_id=collectionId, states=states, page_size=pageSize, page=page, diff --git a/backend/test/test_uploads.py b/backend/test/test_uploads.py index f9f12cef..60e71b7d 100644 --- a/backend/test/test_uploads.py +++ b/backend/test/test_uploads.py @@ -134,7 +134,6 @@ def test_collection_uploads(admin_auth_headers, default_org_id): "name": "My Test Coll", }, ) - print(r.json()) assert r.status_code == 200 data = r.json() coll_id = data["id"] @@ -151,6 +150,17 @@ def test_collection_uploads(admin_auth_headers, default_org_id): assert len(results["items"]) == 1 assert results["items"][0]["id"] == upload_id + # Test all crawls filtered by collection + r = requests.get( + f"{API_PREFIX}/orgs/{default_org_id}/all-crawls?collectionId={coll_id}", + headers=admin_auth_headers, + ) + + results = r.json() + + assert len(results["items"]) == 1 + assert results["items"][0]["id"] == upload_id + # Delete Collection r = requests.delete( f"{API_PREFIX}/orgs/{default_org_id}/collections/{coll_id}",