modify test files

Author: yzAiden

test/backend/app/test_idata_app.py

@@ -515,31 +515,30 @@ def test_router_prefix(self):
         assert router.prefix == "/idata"
 
     def test_routes_registered(self):
-        """Test that all routes are registered."""
-        app = _build_app()
-        routes = [route.path for route in app.routes]
-
-        assert "/idata/knowledge-space" in routes
-        assert "/idata/datasets" in routes
+        """Test that all routes are registered on the router."""
+        # Get routes directly from the router
+        route_paths = [route.path for route in router.routes]
+        
+        # The routes already include the router prefix
+        assert "/idata/knowledge-space" in route_paths
+        assert "/idata/datasets" in route_paths
 
     def test_router_methods(self):
         """Test that routes have correct HTTP methods."""
-        app = _build_app()
-
-        # Find routes by path
+        # Check routes directly from the router
         knowledge_space_route = None
         datasets_route = None
-
-        for route in app.routes:
-            if hasattr(route, 'path'):
-                if route.path == "/idata/knowledge-space":
-                    knowledge_space_route = route
-                elif route.path == "/idata/datasets":
-                    datasets_route = route
-
+        
+        for route in router.routes:
+            # Match against the full path including prefix
+            if route.path == "/idata/knowledge-space":
+                knowledge_space_route = route
+            elif route.path == "/idata/datasets":
+                datasets_route = route
+                
         assert knowledge_space_route is not None
         assert datasets_route is not None
-
-        # Check HTTP methods
-        assert "GET" in [method for method in knowledge_space_route.methods]
-        assert "GET" in [method for method in datasets_route.methods]
+        
+        # Check HTTP methods (APIRoute has 'methods' attribute)
+        assert "GET" in knowledge_space_route.methods
+        assert "GET" in datasets_route.methods

test/backend/app/test_northbound_base_app.py

@@ -274,12 +274,14 @@ def test_cors_middleware_configuration(self):
 
     def test_router_inclusion(self):
         """The main northbound router should be included."""
-        routes = [route.path for route in app.routes]
+        from fastapi.routing import APIRoute
+        routes = [route.path for route in app.routes if isinstance(route, APIRoute)]
         self.assertIn("/dummy", routes)
 
     def test_a2a_router_inclusion(self):
         """A2A router should be registered under /nb/a2a."""
-        routes = [route.path for route in app.routes]
+        from fastapi.routing import APIRoute
+        routes = [route.path for route in app.routes if isinstance(route, APIRoute)]
         self.assertIn("/nb/a2a/{endpoint_id}/.well-known/agent-card.json", routes)
         self.assertIn("/nb/a2a/{endpoint_id}/v1", routes)
         self.assertIn("/nb/a2a/{endpoint_id}/message:send", routes)

test/backend/services/test_data_process_service.py

@@ -361,7 +361,7 @@ async def read(self):
             Image.new('L', (9, 9), color=10),
         ]
 
-        result = asyncio.run(self.service._load_image(session, "http://example.com/a.bin"))
+        result = asyncio.run(self.service._load_image(session, "https://example.com/a.bin"))
 
         self.assertEqual(result.mode, 'RGB')
         temp_file.write.assert_called_once_with(b"not-directly-readable")
@@ -396,14 +396,14 @@ async def read(self):
             Image.new('RGBA', (9, 9), color=(1, 2, 3, 4)),
         ]
 
-        result = asyncio.run(self.service._load_image(session, "http://example.com/a.bin"))
+        result = asyncio.run(self.service._load_image(session, "https://example.com/a.bin"))
 
         self.assertEqual(result.mode, 'RGB')
         mock_unlink.assert_called_once_with("temp-image.bin")
 
     def test_load_image_svg_is_filtered(self):
         """SVG URLs are skipped before any HTTP request."""
-        result = asyncio.run(self.service._load_image(MagicMock(), "http://example.com/a.svg"))
+        result = asyncio.run(self.service._load_image(MagicMock(), "https://example.com/a.svg"))
 
         self.assertIsNone(result)