Add unit test for GraphServerHandler::handler_post_node_expansion()

Author: ericfe-google

bigquery_magics/graph_server.py

@@ -228,7 +228,7 @@ def handle_post_node_expansion(self):
             # - request: Dict with node details (uid, node_labels, node_properties, direction, edge_label)
             self.do_data_response(
                 execute_node_expansion(
-                    params_str=data.get("params"), request=data.get("request")
+                    params=data.get("params"), request=data.get("request")
                 )
             )
         except BaseException as e:

tests/unit/test_graph_server.py

@@ -416,6 +416,30 @@ def test_post_query(self):
         )
         self.assertIsNone(response_data["schema"])
 
+    @pytest.mark.skipif(
+        graph_visualization is None, reason="Requires `spanner-graph-notebook`"
+    )
+    def test_post_node_expansion(self):
+        self.assertTrue(self.server_thread.is_alive())
+        route = graph_server.graph_server.build_route(
+            graph_server.GraphServer.endpoints["post_node_expansion"]
+        )
+        request = {
+            "request": {
+                "uid": "test_uid",
+                "node_labels": ["label1, label2"],
+                "node_properites": {},
+                "direction": "INCOMING",
+                "edge_label": None,
+            },
+            "params": "{}",
+        }
+        response = requests.post(route, json={"params": json.dumps(request)})
+        self.assertEqual(response.status_code, 200)
+        self.assertEqual(
+            response.json(), {"error": "Node expansion not yet implemented"}
+        )
+
 
 def test_stop_server_never_started():
     graph_server.graph_server.stop_server()