105 lines
3.9 KiB
C#
105 lines
3.9 KiB
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 System.Linq;
|
|
using System.Threading.Tasks;
|
|
using Microsoft.AspNetCore.Razor.Language;
|
|
using Microsoft.CodeAnalysis.Text;
|
|
using Xunit;
|
|
|
|
namespace Microsoft.CodeAnalysis.Razor.ProjectSystem
|
|
{
|
|
public class GeneratedCodeContainerTest
|
|
{
|
|
[Fact]
|
|
public void SetOutput_AcceptsInitialOutput()
|
|
{
|
|
// Arrange
|
|
var csharpDocument = RazorCSharpDocument.Create("...", RazorCodeGenerationOptions.CreateDefault(), Enumerable.Empty<RazorDiagnostic>());
|
|
var hostProject = new HostProject("C:/project.csproj", RazorConfiguration.Default);
|
|
var services = TestWorkspace.Create().Services;
|
|
var projectState = ProjectState.Create(services, hostProject);
|
|
var project = new DefaultProjectSnapshot(projectState);
|
|
var hostDocument = new HostDocument("C:/file.cshtml", "C:/file.cshtml");
|
|
var text = SourceText.From("...");
|
|
var textAndVersion = TextAndVersion.Create(text, VersionStamp.Default);
|
|
var documentState = new DocumentState(services, hostDocument, text, VersionStamp.Default, () => Task.FromResult(textAndVersion));
|
|
var document = new DefaultDocumentSnapshot(project, documentState);
|
|
var container = new GeneratedCodeContainer();
|
|
|
|
// Act
|
|
container.SetOutput(csharpDocument, document);
|
|
|
|
// Assert
|
|
Assert.NotNull(container.LatestDocument);
|
|
}
|
|
|
|
[Fact]
|
|
public void TryGetLinePositionSpan_SpanWithinSourceMapping_ReturnsTrue()
|
|
{
|
|
// Arrange
|
|
var content = @"
|
|
@{
|
|
var x = SomeClass.SomeProperty;
|
|
}
|
|
";
|
|
var sourceText = SourceText.From(content);
|
|
var codeDocument = GetCodeDocument(content);
|
|
var csharpDocument = codeDocument.GetCSharpDocument();
|
|
var generatedCode = csharpDocument.GeneratedCode;
|
|
|
|
// TODO: Make writing these tests a little less manual.
|
|
// Position of `SomeProperty` in the generated code.
|
|
var symbol = "SomeProperty";
|
|
var span = new TextSpan(generatedCode.IndexOf(symbol), symbol.Length);
|
|
|
|
// Position of `SomeProperty` in the source code.
|
|
var expectedLineSpan = new LinePositionSpan(new LinePosition(2, 22), new LinePosition(2, 34));
|
|
|
|
// Act
|
|
var result = GeneratedCodeContainer.TryGetLinePositionSpan(span, sourceText, csharpDocument, out var lineSpan);
|
|
|
|
// Assert
|
|
Assert.True(result);
|
|
Assert.Equal(expectedLineSpan, lineSpan);
|
|
}
|
|
|
|
[Fact]
|
|
public void TryGetLinePositionSpan_SpanOutsideSourceMapping_ReturnsFalse()
|
|
{
|
|
// Arrange
|
|
var content = @"
|
|
@{
|
|
var x = SomeClass.SomeProperty;
|
|
}
|
|
";
|
|
var sourceText = SourceText.From(content);
|
|
var codeDocument = GetCodeDocument(content);
|
|
var csharpDocument = codeDocument.GetCSharpDocument();
|
|
var generatedCode = csharpDocument.GeneratedCode;
|
|
|
|
// Position of `ExecuteAsync` in the generated code.
|
|
var symbol = "ExecuteAsync";
|
|
var span = new TextSpan(generatedCode.IndexOf(symbol), symbol.Length);
|
|
|
|
// Act
|
|
var result = GeneratedCodeContainer.TryGetLinePositionSpan(span, sourceText, csharpDocument, out var lineSpan);
|
|
|
|
// Assert
|
|
Assert.False(result);
|
|
}
|
|
|
|
private static RazorCodeDocument GetCodeDocument(string content)
|
|
{
|
|
var sourceProjectItem = new TestRazorProjectItem("test.cshtml")
|
|
{
|
|
Content = content,
|
|
};
|
|
|
|
var engine = RazorProjectEngine.Create();
|
|
var codeDocument = engine.ProcessDesignTime(sourceProjectItem);
|
|
return codeDocument;
|
|
}
|
|
}
|
|
}
|