Fix nullability

Author: jozkee

tests/ModelContextProtocol.Tests/Client/McpClientTests.cs

@@ -56,15 +56,17 @@ public async Task CanReadServerInfo()
         Assert.NotNull(serverInfo.Icons);
         Assert.Equal(2, serverInfo.Icons.Count);
 
-        Assert.Equal("https://example.com/icon-48.png", serverInfo.Icons[0].Source);
-        Assert.Equal("image/png", serverInfo.Icons[0].MimeType);
-        Assert.Single(serverInfo.Icons[0].Sizes, "48x48");
-        Assert.Equal("light", serverInfo.Icons[0].Theme);
-
-        Assert.Equal("https://example.com/icon.svg", serverInfo.Icons[1].Source);
-        Assert.Equal("image/svg+xml", serverInfo.Icons[1].MimeType);
-        Assert.Single(serverInfo.Icons[1].Sizes, "any");
-        Assert.Equal("dark", serverInfo.Icons[1].Theme);
+        var icon0 = serverInfo.Icons[0];
+        Assert.Equal("https://example.com/icon-48.png", icon0.Source);
+        Assert.Equal("image/png", icon0.MimeType);
+        Assert.Single(icon0.Sizes!, "48x48");
+        Assert.Equal("light", icon0.Theme);
+
+        var icon1 = serverInfo.Icons[1];
+        Assert.Equal("https://example.com/icon.svg", icon1.Source);
+        Assert.Equal("image/svg+xml", icon1.MimeType);
+        Assert.Single(icon1.Sizes!, "any");
+        Assert.Equal("dark", icon1.Theme);
     }
 
     [Theory]

tests/ModelContextProtocol.Tests/Server/McpServerPromptTests.cs

@@ -506,7 +506,7 @@ public void SupportsIconsInCreateOptions()
             Icons = icons
         });
 
-        var icon = Assert.Single(prompt.ProtocolPrompt.Icons);
+        var icon = Assert.Single(prompt.ProtocolPrompt.Icons!);
         Assert.Equal("https://example.com/prompt-icon.png", icon.Source);
         Assert.Equal("image/png", icon.MimeType);
     }
@@ -516,7 +516,7 @@ public void SupportsIconSourceInAttribute()
     {
         McpServerPrompt prompt = McpServerPrompt.Create([McpServerPrompt(IconSource = "https://example.com/prompt-icon.svg")] () => "test prompt");
 
-        var icon = Assert.Single(prompt.ProtocolPrompt.Icons);
+        var icon = Assert.Single(prompt.ProtocolPrompt.Icons!);
         Assert.Equal("https://example.com/prompt-icon.svg", icon.Source);
         Assert.Null(icon.MimeType);
         Assert.Null(icon.Sizes);
@@ -535,7 +535,7 @@ public void CreateOptionsIconsOverrideAttributeIconSource_Prompt()
             Icons = optionsIcons
         });
 
-        var icon = Assert.Single(prompt.ProtocolPrompt.Icons);
+        var icon = Assert.Single(prompt.ProtocolPrompt.Icons!);
         Assert.Equal("https://example.com/override-icon.svg", icon.Source);
         Assert.Equal("image/svg+xml", icon.MimeType);
     }

tests/ModelContextProtocol.Tests/Server/McpServerResourceTests.cs

@@ -691,7 +691,7 @@ public void SupportsIconsInResourceCreateOptions()
             Icons = icons
         });
 
-        var icon = Assert.Single(resource.ProtocolResourceTemplate.Icons);
+        var icon = Assert.Single(resource.ProtocolResourceTemplate.Icons!);
         Assert.Equal("https://example.com/resource-icon.png", icon.Source);
         Assert.Equal("image/png", icon.MimeType);
     }
@@ -701,7 +701,7 @@ public void SupportsIconSourceInResourceAttribute()
     {
         McpServerResource resource = McpServerResource.Create([McpServerResource(UriTemplate = "test://resource", IconSource = "https://example.com/resource-icon.svg")] () => "test content");
 
-        var icon = Assert.Single(resource.ProtocolResourceTemplate.Icons);
+        var icon = Assert.Single(resource.ProtocolResourceTemplate.Icons!);
         Assert.Equal("https://example.com/resource-icon.svg", icon.Source);
         Assert.Null(icon.MimeType);
         Assert.Null(icon.Sizes);
@@ -720,7 +720,7 @@ public void CreateOptionsIconsOverrideAttributeIconSource_Resource()
             Icons = optionsIcons
         });
 
-        var icon = Assert.Single(resource.ProtocolResourceTemplate.Icons);
+        var icon = Assert.Single(resource.ProtocolResourceTemplate.Icons!);
         Assert.Equal("https://example.com/override-icon.svg", icon.Source);
         Assert.Equal("image/svg+xml", icon.MimeType);
     }

tests/ModelContextProtocol.Tests/Server/McpServerToolTests.cs

@@ -711,7 +711,7 @@ public void SupportsIconSourceInAttribute()
     {
         McpServerTool tool = McpServerTool.Create([McpServerTool(IconSource = "https://example.com/tool-icon.png")] () => "result");
 
-        var icon = Assert.Single(tool.ProtocolTool.Icons);
+        var icon = Assert.Single(tool.ProtocolTool.Icons!);
         Assert.Equal("https://example.com/tool-icon.png", icon.Source);
         Assert.Null(icon.MimeType);
         Assert.Null(icon.Sizes);
@@ -730,7 +730,7 @@ public void CreateOptionsIconsOverrideAttributeIconSource()
             Icons = optionsIcons
         });
 
-        var icon = Assert.Single(tool.ProtocolTool.Icons);
+        var icon = Assert.Single(tool.ProtocolTool.Icons!);
         Assert.Equal("https://example.com/override-icon.svg", icon.Source);
         Assert.Equal("image/svg+xml", icon.MimeType);
     }