aspnetcore/test/Microsoft.AspNet.Razor.Test/CodeGeneration/CodeGenTestHost.cs

27 lines
818 B
C#

// Copyright (c) .NET Foundation. All rights reserved.
// Licensed under the Apache License, Version 2.0. See License.txt in the project root for license information.
using Microsoft.AspNet.Razor.CodeGeneration;
namespace Microsoft.AspNet.Razor.Test.Generator
{
public class CodeGenTestHost : RazorEngineHost
{
public CodeGenTestHost(RazorCodeLanguage language)
: base(language)
{
}
public override CodeGenerator DecorateCodeGenerator(CodeGenerator incomingBuilder, CodeGeneratorContext context)
{
if (incomingBuilder is CodeGenTestCodeGenerator)
{
return incomingBuilder;
}
else
{
return new CodeGenTestCodeGenerator(context);
}
}
}
}