Adding some more descriptive relationships to the MCP inspector (#828)

Author: aaronpowell

src/CommunityToolkit.Aspire.Hosting.McpInspector/McpInspectorResourceBuilderExtensions.cs

@@ -198,7 +198,13 @@ public static IResourceBuilder<McpInspectorResource> WithMcpServer<TResource>(
 
         builder.Resource.AddMcpServer(mcpServer.Resource, isDefault, transportType, path);
 
-        mcpServer.WithRelationship(builder.Resource, "InspectedBy");
+        mcpServer.WithRelationship(builder.Resource, "Inspected By");
+        builder.WithRelationship(mcpServer.Resource, "Inspecting");
+
+        if (isDefault)
+        {
+            builder.WithRelationship(mcpServer.Resource, "Default Inspected Server");
+        }
 
         return builder;
     }

tests/CommunityToolkit.Aspire.Hosting.McpInspector.Tests/McpInspectorResourceBuilderExtensionsTests.cs

@@ -319,11 +319,44 @@ public void WithMcpServerCreatesResourceRelationshipAnnotations()
         var inspectorResource = Assert.Single(appModel.Resources.OfType<McpInspectorResource>());
         var serverResource = appModel.Resources.Single(r => r.Name == "mcpServer");
 
-        // The inspector and/or server should have a resource relationship annotation linking them.
-        var serverHasRelationship = serverResource.TryGetAnnotationsOfType<ResourceRelationshipAnnotation>(out var serverRelationships);
-        var inspectorHasRelationship = inspectorResource.TryGetAnnotationsOfType<ResourceRelationshipAnnotation>(out var inspectorRelationships);
+        // The server should have an "Inspected By" relationship pointing at the inspector
+        Assert.True(serverResource.TryGetAnnotationsOfType<ResourceRelationshipAnnotation>(out var serverRelationships));
+        var serverRelationship = Assert.Single(serverRelationships);
+        Assert.Same(inspectorResource, serverRelationship.Resource);
+        Assert.Equal("Inspected By", serverRelationship.Type);
+
+        // The inspector should have relationships back to the server: "Inspecting" and (since isDefault=true) "Default Inspected Server"
+        Assert.True(inspectorResource.TryGetAnnotationsOfType<ResourceRelationshipAnnotation>(out var inspectorRelationships));
+        Assert.Contains(inspectorRelationships, r => r.Resource == serverResource && r.Type == "Inspecting");
+        Assert.Contains(inspectorRelationships, r => r.Resource == serverResource && r.Type == "Default Inspected Server");
+    }
+
+    [Fact]
+    public void WithMcpServer_NonDefaultDoesNotAddDefaultInspectedServerRelationship()
+    {
+        // Arrange
+        var appBuilder = DistributedApplication.CreateBuilder();
+
+        // Create mock MCP server resources
+        var mockServer1 = appBuilder.AddProject<Projects.CommunityToolkit_Aspire_Hosting_McpInspector_McpServer>("mcpServer1");
+        var mockServer2 = appBuilder.AddProject<Projects.CommunityToolkit_Aspire_Hosting_McpInspector_McpServer>("mcpServer2");
+
+        // Act: first is default, second is not
+        var inspector = appBuilder.AddMcpInspector("inspector")
+            .WithMcpServer(mockServer1, isDefault: true)
+            .WithMcpServer(mockServer2, isDefault: false);
+
+        using var app = appBuilder.Build();
+
+        // Assert
+        var appModel = app.Services.GetRequiredService<DistributedApplicationModel>();
+
+        var inspectorResource = Assert.Single(appModel.Resources.OfType<McpInspectorResource>());
+        var serverResource2 = appModel.Resources.Single(r => r.Name == "mcpServer2");
 
-        Assert.True(serverHasRelationship || inspectorHasRelationship, "Expected a ResourceRelationshipAnnotation on either the server or inspector resource.");
+        // Inspector should have relationships but should NOT have "Default Inspected Server" for serverResource2
+        Assert.True(inspectorResource.TryGetAnnotationsOfType<ResourceRelationshipAnnotation>(out var inspectorRelationships));
+        Assert.DoesNotContain(inspectorRelationships, r => r.Resource == serverResource2 && r.Type == "Default Inspected Server");
     }
 
     [Fact]