@@ -54,33 +54,12 @@ def test_rest_registry_server_initializes_correctly(
5454 assert {"BearerAuth" : []} in openapi_schema ["security" ]
5555
5656
57- def test_routes_registered_in_app (mock_store_and_registry ):
58- from fastapi . routing import APIRoute
57+ def test_routes_registered_in_app ():
58+ from feast . api . registry . rest import register_all_routes
5959
60- store , _ = mock_store_and_registry
60+ app = MagicMock ()
61+ grpc_handler = MagicMock ()
62+ server = MagicMock ()
63+ register_all_routes (app , grpc_handler , server )
6164
62- server = RestRegistryServer (store )
63- route_paths = [
64- route .path for route in server .app .routes if isinstance (route , APIRoute )
65- ]
66- assert "/feature_services" in route_paths
67- assert "/entities" in route_paths
68- assert "/projects" in route_paths
69- assert "/data_sources" in route_paths
70- assert "/saved_datasets" in route_paths
71- assert "/permissions" in route_paths
72- assert "/lineage/registry" in route_paths
73- assert "/lineage/objects/{object_type}/{object_name}" in route_paths
74- assert "/lineage/complete" in route_paths
75- assert "/entities/all" in route_paths
76- assert "/feature_views/all" in route_paths
77- assert "/data_sources/all" in route_paths
78- assert "/feature_services/all" in route_paths
79- assert "/saved_datasets/all" in route_paths
80- assert "/lineage/registry/all" in route_paths
81- assert "/lineage/complete/all" in route_paths
82- assert "/features" in route_paths
83- assert "/features/all" in route_paths
84- assert "/features/{feature_view}/{name}" in route_paths
85- assert "/metrics/resource_counts" in route_paths
86- assert "/metrics/recently_visited" in route_paths
65+ assert app .include_router .call_count == 13
0 commit comments