diff --git a/src/Microsoft.AspNetCore.Mvc.Razor.Extensions/ViewComponentTagHelperPass.cs b/src/Microsoft.AspNetCore.Mvc.Razor.Extensions/ViewComponentTagHelperPass.cs index 64218e7d02..76b025e1a4 100644 --- a/src/Microsoft.AspNetCore.Mvc.Razor.Extensions/ViewComponentTagHelperPass.cs +++ b/src/Microsoft.AspNetCore.Mvc.Razor.Extensions/ViewComponentTagHelperPass.cs @@ -53,6 +53,7 @@ namespace Microsoft.AspNetCore.Mvc.Razor.Extensions RazorIRBuilder.Create(statement) .Add(new RazorIRToken() { + Kind = RazorIRToken.TokenKind.CSharp, Content = writer.Builder.ToString() }); diff --git a/test/Microsoft.AspNetCore.Mvc.Razor.Extensions.Test/ViewComponentTagHelperPassTest.cs b/test/Microsoft.AspNetCore.Mvc.Razor.Extensions.Test/ViewComponentTagHelperPassTest.cs index be377e9b23..9dc91db1d1 100644 --- a/test/Microsoft.AspNetCore.Mvc.Razor.Extensions.Test/ViewComponentTagHelperPassTest.cs +++ b/test/Microsoft.AspNetCore.Mvc.Razor.Extensions.Test/ViewComponentTagHelperPassTest.cs @@ -122,6 +122,7 @@ public class __Generated__TagCloudViewComponentTagHelper : Microsoft.AspNetCore. ", tokenNode.Content, ignoreLineEndingDifferences: true); + Assert.Equal(tokenNode.Kind, RazorIRToken.TokenKind.CSharp); } [Fact] @@ -194,6 +195,7 @@ public class __Generated__TagCloudViewComponentTagHelper : Microsoft.AspNetCore. ", tokenNode.Content, ignoreLineEndingDifferences: true); + Assert.Equal(tokenNode.Kind, RazorIRToken.TokenKind.CSharp); } [Fact] @@ -278,6 +280,7 @@ public class __Generated__TagCloudViewComponentTagHelper : Microsoft.AspNetCore. ", tokenNode.Content, ignoreLineEndingDifferences: true); + Assert.Equal(tokenNode.Kind, RazorIRToken.TokenKind.CSharp); } private RazorCodeDocument CreateDocument(string content)