47 lines
1.7 KiB
C#
47 lines
1.7 KiB
C#
using System.Collections.Immutable;
|
|
using System.IO;
|
|
using System.Linq;
|
|
using Microsoft.CodeAnalysis;
|
|
using PacketMediator.Generator.Tests.Utils;
|
|
using Microsoft.CodeAnalysis.CSharp;
|
|
using Xunit;
|
|
|
|
namespace PacketMediator.Generator.Tests;
|
|
|
|
public class SourceGeneratorWithAdditionalFilesTests
|
|
{
|
|
private const string DddRegistryText = @"User
|
|
Document
|
|
Customer";
|
|
|
|
[Fact]
|
|
public void GenerateClassesBasedOnDDDRegistry()
|
|
{
|
|
// Create an instance of the source generator.
|
|
var generator = new SourceGeneratorWithAdditionalFiles();
|
|
|
|
// Source generators should be tested using 'GeneratorDriver'.
|
|
GeneratorDriver driver = CSharpGeneratorDriver.Create(generator);
|
|
|
|
// Add the additional file separately from the compilation.
|
|
driver = driver.AddAdditionalTexts(
|
|
ImmutableArray.Create<AdditionalText>(
|
|
new TestAdditionalFile("./DDD.UbiquitousLanguageRegistry.txt", DddRegistryText)
|
|
)
|
|
);
|
|
|
|
// To run generators, we can use an empty compilation.
|
|
var compilation = CSharpCompilation.Create(nameof(SourceGeneratorWithAdditionalFilesTests));
|
|
|
|
// Run generators. Don't forget to use the new compilation rather than the previous one.
|
|
driver.RunGeneratorsAndUpdateCompilation(compilation, out var newCompilation, out _);
|
|
|
|
// Retrieve all files in the compilation.
|
|
var generatedFiles = newCompilation.SyntaxTrees
|
|
.Select(t => Path.GetFileName(t.FilePath))
|
|
.ToArray();
|
|
|
|
// In this case, it is enough to check the file name.
|
|
Assert.Equivalent(new[] { "User.g.cs", "Document.g.cs", "Customer.g.cs" }, generatedFiles);
|
|
}
|
|
}
|