csharp-sdk/tests/ModelContextProtocol.Tests/Protocol/ClientCapabilitiesTests.cs at b19aefda8cce5f5525b7eb8786501b5dd0fa71fc · modelcontextprotocol/csharp-sdk · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
using ModelContextProtocol.Protocol;
using System.Text.Json;

namespace ModelContextProtocol.Tests.Protocol;

public static class ClientCapabilitiesTests
{
    [Fact]
    public static void ExtensionsProperty_SerializationRoundTrip()
    {
        // Arrange - Use raw JSON instead of objects for source generation compatibility
        string json = """
            {
                "extensions": {
                    "io.modelcontextprotocol/oauth-client-credentials": {},
                    "io.modelcontextprotocol/test-extension": {
                        "setting1": "value1",
                        "setting2": 42
                    }
                }
            }
            """;

        // Act - Deserialize from JSON
        var deserialized = JsonSerializer.Deserialize<ClientCapabilities>(json, McpJsonUtilities.DefaultOptions);

        // Assert
        Assert.NotNull(deserialized);
        Assert.NotNull(deserialized.Extensions);
        Assert.Equal(2, deserialized.Extensions.Count);
        Assert.True(deserialized.Extensions.ContainsKey("io.modelcontextprotocol/oauth-client-credentials"));
        Assert.True(deserialized.Extensions.ContainsKey("io.modelcontextprotocol/test-extension"));

        // Act - Serialize back to JSON
        string roundtrippedJson = JsonSerializer.Serialize(deserialized, McpJsonUtilities.DefaultOptions);

        // Assert - Deserialize again to verify
        var deserialized2 = JsonSerializer.Deserialize<ClientCapabilities>(roundtrippedJson, McpJsonUtilities.DefaultOptions);
        Assert.NotNull(deserialized2);
        Assert.NotNull(deserialized2.Extensions);
        Assert.Equal(2, deserialized2.Extensions.Count);
    }

    [Fact]
    public static void ExtensionsProperty_DeserializesCorrectly()
    {
        // Arrange
        string json = """
            {
                "extensions": {
                    "io.modelcontextprotocol/test": {}
                }
            }
            """;

        // Act
        var capabilities = JsonSerializer.Deserialize<ClientCapabilities>(json, McpJsonUtilities.DefaultOptions);

        // Assert
        Assert.NotNull(capabilities);
        Assert.NotNull(capabilities.Extensions);
        Assert.Single(capabilities.Extensions);
        Assert.True(capabilities.Extensions.ContainsKey("io.modelcontextprotocol/test"));
    }

    [Fact]
    public static void WithoutExtensions_DeserializesWithNullExtensions()
    {
        // Arrange
        string json = "{}";

        // Act
        var capabilities = JsonSerializer.Deserialize<ClientCapabilities>(json, McpJsonUtilities.DefaultOptions);

        // Assert
        Assert.NotNull(capabilities);
        Assert.Null(capabilities.Extensions);
    }

    [Fact]
    public static void WithEmptyExtensions_DeserializesAsEmptyDictionary()
    {
        // Arrange
        string json = """
            {
                "extensions": {}
            }
            """;

        // Act
        var capabilities = JsonSerializer.Deserialize<ClientCapabilities>(json, McpJsonUtilities.DefaultOptions);

        // Assert
        Assert.NotNull(capabilities);
        Assert.NotNull(capabilities.Extensions);
        Assert.Empty(capabilities.Extensions);
    }

    [Fact]
    public static void ExtensionsWithComplexValues_RoundTrips()
    {
        // Arrange
        string json = """
            {
                "extensions": {
                    "io.modelcontextprotocol/complex": {
                        "stringValue": "test",
                        "numberValue": 123,
                        "boolValue": true,
                        "arrayValue": [1, 2, 3]
                    }
                }
            }
            """;

        // Act - Deserialize from JSON
        var deserialized = JsonSerializer.Deserialize<ClientCapabilities>(json, McpJsonUtilities.DefaultOptions);

        // Assert
        Assert.NotNull(deserialized);
        Assert.NotNull(deserialized.Extensions);
        Assert.Single(deserialized.Extensions);
        Assert.True(deserialized.Extensions.ContainsKey("io.modelcontextprotocol/complex"));

        // Verify the complex value can be accessed as JsonElement
        var complexValue = deserialized.Extensions["io.modelcontextprotocol/complex"];
        Assert.NotNull(complexValue);

        // Act - Serialize back to JSON
        string roundtrippedJson = JsonSerializer.Serialize(deserialized, McpJsonUtilities.DefaultOptions);

        // Assert - Verify it can deserialize again
        var deserialized2 = JsonSerializer.Deserialize<ClientCapabilities>(roundtrippedJson, McpJsonUtilities.DefaultOptions);
        Assert.NotNull(deserialized2);
        Assert.NotNull(deserialized2.Extensions);
        Assert.Single(deserialized2.Extensions);
    }
}