Fix depth and last approved filter test that wasn't updated

webrecorder · Feb 6, 2025 · 268c7cb · 268c7cb
1 parent 0d7c74c
commit 268c7cb
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 9 deletions.
diff --git a/backend/btrixcloud/pages.py b/backend/btrixcloud/pages.py
@@ -546,10 +546,8 @@ async def list_pages(
         if is_seed in (True, False):
             query["isSeed"] = is_seed
 
-        # Check that field matches and value is an int to avoid 0
-        # from returning null results
-        if depth:
-            query["$and"] = [{"depth": depth}, {"depth": {"type": 16}}]
+        if isinstance(depth, int):
+            query["depth"] = depth
 
         if reviewed:
             query["$or"] = [
@@ -699,10 +697,8 @@ async def list_collection_pages(
         if is_seed in (True, False):
             query["isSeed"] = is_seed
 
-        # Check that field matches and value is an int to avoid 0
-        # from returning null results
-        if depth:
-            query["$and"] = [{"depth": depth}, {"depth": {"type": 16}}]
+        if isinstance(depth, int):
+            query["depth"] = depth
 
         aggregate = [{"$match": query}]
 

diff --git a/backend/test/test_run_crawl.py b/backend/test/test_run_crawl.py
@@ -953,7 +953,7 @@ def test_crawl_pages_qa_filters(crawler_auth_headers, default_org_id, crawler_cr
         headers=crawler_auth_headers,
     )
     assert r.status_code == 200
-    assert r.json()["total"] == 0
+    assert r.json()["total"] == 2
 
 
 def test_re_add_crawl_pages(crawler_auth_headers, default_org_id, crawler_crawl_id):