format & update tests

Author: emma-sg

backend/btrixcloud/crawlconfigs.py

@@ -1422,9 +1422,9 @@ async def get_crawl_configs(
 
     @router.get("/tags", response_model=List[str], deprecated=True)
     async def get_crawl_config_tags(org: Organization = Depends(org_viewer_dep)):
-        '''
+        """
         Deprecated - prefer /api/orgs/{oid}/crawlconfigs/tagCounts instead.
-        '''
+        """
         return await ops.get_crawl_config_tags(org)
 
     @router.get("/tagCounts", response_model=CrawlConfigTags)

backend/btrixcloud/models.py

@@ -593,6 +593,7 @@ class CrawlConfigTagCount(BaseModel):
     tag: str
     count: int
 
+
 # ============================================================================
 class CrawlConfigTags(BaseModel):
     """Response model for crawlconfig tags"""

backend/test/test_crawl_config_tags.py

@@ -50,6 +50,22 @@ def test_get_config_by_tag_1(admin_auth_headers, default_org_id):
     assert sorted(data) == ["tag-1", "tag-2", "wr-test-1", "wr-test-2"]
 
 
+def test_get_config_by_tag_counts_1(admin_auth_headers, default_org_id):
+    r = requests.get(
+        f"{API_PREFIX}/orgs/{default_org_id}/crawlconfigs/tagCounts",
+        headers=admin_auth_headers,
+    )
+    data = r.json()
+    assert data == {
+        "tags": [
+            {"tag": "tag-1", "count": 1},
+            {"tag": "tag-2", "count": 1},
+            {"tag": "wr-test-1", "count": 1},
+            {"tag": "wr-test-2", "count": 1},
+        ]
+    }
+
+
 def test_create_new_config_2(admin_auth_headers, default_org_id):
     r = requests.post(
         f"{API_PREFIX}/orgs/{default_org_id}/crawlconfigs/",
@@ -84,6 +100,24 @@ def test_get_config_by_tag_2(admin_auth_headers, default_org_id):
     ]
 
 
+def test_get_config_by_tag_counts_2(admin_auth_headers, default_org_id):
+    r = requests.get(
+        f"{API_PREFIX}/orgs/{default_org_id}/crawlconfigs/tagCounts",
+        headers=admin_auth_headers,
+    )
+    data = r.json()
+    assert data == {
+        "tags": [
+            {"tag": "tag-0", "count": 1},
+            {"tag": "tag-1", "count": 1},
+            {"tag": "tag-2", "count": 1},
+            {"tag": "tag-3", "count": 1},
+            {"tag": "wr-test-1", "count": 1},
+            {"tag": "wr-test-2", "count": 1},
+        ]
+    }
+
+
 def test_get_config_2(admin_auth_headers, default_org_id):
     r = requests.get(
         f"{API_PREFIX}/orgs/{default_org_id}/crawlconfigs/{new_cid_2}",