|  | 
|  | 1 | +using Microsoft.Extensions.DependencyInjection; | 
|  | 2 | +using ModelContextProtocol.Protocol.Types; | 
|  | 3 | +using ModelContextProtocol.Server; | 
|  | 4 | + | 
|  | 5 | +namespace ModelContextProtocol.Tests.Server; | 
|  | 6 | +public class McpServerResourceTests | 
|  | 7 | +{ | 
|  | 8 | +    [Fact] | 
|  | 9 | +    public void CanCreateServerWithResourceTemplates() | 
|  | 10 | +    { | 
|  | 11 | +        var services = new ServiceCollection(); | 
|  | 12 | + | 
|  | 13 | +        services.AddMcpServer() | 
|  | 14 | +            .WithStdioServerTransport() | 
|  | 15 | +            .WithListResourceTemplatesHandler((ctx, ct) => | 
|  | 16 | +            { | 
|  | 17 | +                return Task.FromResult(new ListResourceTemplatesResult | 
|  | 18 | +                { | 
|  | 19 | +                    ResourceTemplates = | 
|  | 20 | +                    [ | 
|  | 21 | +                        new ResourceTemplate { Name = "Static Resource", Description = "A static resource with a numeric ID", UriTemplate = "test://static/resource/{id}" } | 
|  | 22 | +                    ] | 
|  | 23 | +                }); | 
|  | 24 | +            }) | 
|  | 25 | +            .WithReadResourceHandler((ctx, ct) => | 
|  | 26 | +            { | 
|  | 27 | +                return Task.FromResult(new ReadResourceResult | 
|  | 28 | +                { | 
|  | 29 | +                    Contents = [new ResourceContents | 
|  | 30 | +                    { | 
|  | 31 | +                        Uri = ctx.Params!.Uri!, | 
|  | 32 | +                        Text = "Static Resource", | 
|  | 33 | +                        MimeType = "text/plain", | 
|  | 34 | +                    }] | 
|  | 35 | +                }); | 
|  | 36 | +            }); | 
|  | 37 | + | 
|  | 38 | +        var provider = services.BuildServiceProvider(); | 
|  | 39 | + | 
|  | 40 | +        provider.GetRequiredService<IMcpServer>(); | 
|  | 41 | +    } | 
|  | 42 | + | 
|  | 43 | +    [Fact] | 
|  | 44 | +    public void CanCreateServerWithResources() | 
|  | 45 | +    { | 
|  | 46 | +        var services = new ServiceCollection(); | 
|  | 47 | + | 
|  | 48 | +        services.AddMcpServer() | 
|  | 49 | +            .WithStdioServerTransport() | 
|  | 50 | +            .WithListResourcesHandler((ctx, ct) => | 
|  | 51 | +            { | 
|  | 52 | +                return Task.FromResult(new ListResourcesResult | 
|  | 53 | +                { | 
|  | 54 | +                    Resources = | 
|  | 55 | +                    [ | 
|  | 56 | +                        new Resource { Name = "Static Resource", Description = "A static resource with a numeric ID", Uri = "test://static/resource/foo.txt" } | 
|  | 57 | +                    ] | 
|  | 58 | +                }); | 
|  | 59 | +            }) | 
|  | 60 | +            .WithReadResourceHandler((ctx, ct) => | 
|  | 61 | +            { | 
|  | 62 | +                return Task.FromResult(new ReadResourceResult | 
|  | 63 | +                { | 
|  | 64 | +                    Contents = [new ResourceContents | 
|  | 65 | +                    { | 
|  | 66 | +                        Uri = ctx.Params!.Uri!, | 
|  | 67 | +                        Text = "Static Resource", | 
|  | 68 | +                        MimeType = "text/plain", | 
|  | 69 | +                    }] | 
|  | 70 | +                }); | 
|  | 71 | +            }); | 
|  | 72 | + | 
|  | 73 | +        var provider = services.BuildServiceProvider(); | 
|  | 74 | + | 
|  | 75 | +        provider.GetRequiredService<IMcpServer>(); | 
|  | 76 | +    } | 
|  | 77 | + | 
|  | 78 | +    [Fact] | 
|  | 79 | +    public void CreatingReadHandlerWithNoListHandlerFails() | 
|  | 80 | +    { | 
|  | 81 | +        var services = new ServiceCollection(); | 
|  | 82 | +        services.AddMcpServer() | 
|  | 83 | +            .WithStdioServerTransport() | 
|  | 84 | +            .WithReadResourceHandler((ctx, ct) => | 
|  | 85 | +            { | 
|  | 86 | +                return Task.FromResult(new ReadResourceResult | 
|  | 87 | +                { | 
|  | 88 | +                    Contents = [new ResourceContents | 
|  | 89 | +                    { | 
|  | 90 | +                        Uri = ctx.Params!.Uri!, | 
|  | 91 | +                        Text = "Static Resource", | 
|  | 92 | +                        MimeType = "text/plain", | 
|  | 93 | +                    }] | 
|  | 94 | +                }); | 
|  | 95 | +            }); | 
|  | 96 | +        var sp = services.BuildServiceProvider(); | 
|  | 97 | +        Assert.Throws<McpServerException>(() => sp.GetRequiredService<IMcpServer>()); | 
|  | 98 | +    } | 
|  | 99 | +} | 
0 commit comments