Fix tests

core/web/apiv2/graph.py

@@ -170,9 +170,9 @@ async def match(request: AnalysisRequest) -> AnalysisResponse:
         processed_relationships = set()
 
         if request.fetch_neighbors:
-            vertices, edges, _ = db_observable.neighbors()
-            # Get neighboring entities and relationships.
-            for edge in edges:
+            vertices, paths, _ = db_observable.neighbors()
+            for path in paths:
+                edge = path[0]  # neighbors only returns 1 hop max by default
                 if edge.id in processed_relationships:
                     continue
 

tests/apiv2/graph.py

@@ -66,7 +66,7 @@ def test_get_neighbors_tag(self):
         data = response.json()
         self.assertEqual(response.status_code, 200, data)
         self.assertEqual(len(data["vertices"]), 2)
-        self.assertEqual(data["total"], 2)
+        self.assertEqual(data["total"], 1)
 
         self.assertIn(self.entity1.extended_id, data["vertices"])
 

tests/apiv2/observables.py

@@ -163,9 +163,9 @@ def test_tag_observable(self):
         tag_relationships = data["tags"][f'observables/{observable_id}']
         self.assertEqual(len(tag_relationships), 2, data)
         self.assertIn("tag1", tag_relationships)
-        self.assertEquals(tag_relationships["tag1"]["source"], f'observables/{observable_id}')
+        self.assertEqual(tag_relationships["tag1"]["source"], f'observables/{observable_id}')
         self.assertIn("tag2", tag_relationships)
-        self.assertEquals(tag_relationships["tag2"]["source"], f'observables/{observable_id}')
+        self.assertEqual(tag_relationships["tag2"]["source"], f'observables/{observable_id}')
 
         response = client.post(
             f"/api/v2/tags/search", json={"name": "tag1", "count": 1, "page": 0}