Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,8 @@ namespace TUnit.Core.SourceGenerator.CodeGenerators;
[Generator]
public class DynamicTestsGenerator : IIncrementalGenerator
{
public const string ParseDynamicTests = "ParseDynamicTests";

public void Initialize(IncrementalGeneratorInitializationContext context)
{
var enabledProvider = context.AnalyzerConfigOptionsProvider
Expand All @@ -23,7 +25,8 @@ public void Initialize(IncrementalGeneratorInitializationContext context)
predicate: static (_, _) => true,
transform: static (ctx, _) => ExtractDynamicTestModel(ctx))
.Where(static m => m is not null)
.Combine(enabledProvider);
.Combine(enabledProvider)
.WithTrackingName(ParseDynamicTests);

context.RegisterSourceOutput(standardTests, (sourceContext, data) =>
{
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,97 @@
using Microsoft.CodeAnalysis;
using Microsoft.CodeAnalysis.CSharp;
using TUnit.Core.SourceGenerator.CodeGenerators;

namespace TUnit.Assertions.SourceGenerator.IncrementalTests;

public class DynamicTestsGeneratorIncrementalTests
{
private const string DefaultConverter =
"""
using TUnit.TestProject.Attributes;
using TUnit.Core;

#nullable enabled
public class DynamicTestArgumentsTests
{
[DynamicTestBuilder]
public void BuildDynamicTests(DynamicTestBuilderContext context)
{
}

public void SecondBuildDynamicTests(DynamicTestBuilderContext context)
{
}
}
""";

[Fact]
public void AddUnrelatedType_MethodShouldNotRegenerate()
{
var syntaxTree = CSharpSyntaxTree.ParseText(DefaultConverter, CSharpParseOptions.Default);
var compilation1 = Fixture.CreateLibrary(syntaxTree);

var driver1 = TestHelper.GenerateTracked<DynamicTestsGenerator>(compilation1);
AssertRunReasons(driver1, IncrementalGeneratorRunReasons.New);

var compilation2 = compilation1.AddSyntaxTrees(CSharpSyntaxTree.ParseText("struct MyValue {}"));
var driver2 = driver1.RunGenerators(compilation2);
AssertRunReasons(driver2, IncrementalGeneratorRunReasons.Cached);
}

[Fact]
public void ModifyDynamicMethodShouldRegenerate()
{
var syntaxTree = CSharpSyntaxTree.ParseText(DefaultConverter, CSharpParseOptions.Default);
var compilation1 = Fixture.CreateLibrary(syntaxTree);

var driver1 = TestHelper.GenerateTracked<DynamicTestsGenerator>(compilation1);
AssertRunReasons(driver1, IncrementalGeneratorRunReasons.New, 0);

var compilation2 = TestHelper.ReplaceMethodDeclaration(compilation1, "BuildDynamicTests",
"""
[DynamicTestBuilder]
public static void BuildDynamicTests(DynamicTestBuilderContext context)
{
}
"""
);

var driver2 = driver1.RunGenerators(compilation2);
AssertRunReasons(driver2, IncrementalGeneratorRunReasons.Modified);
}

[Fact]
public void AddDynamicMethodShouldRegenerate()
{
var syntaxTree = CSharpSyntaxTree.ParseText(DefaultConverter, CSharpParseOptions.Default);
var compilation1 = Fixture.CreateLibrary(syntaxTree);

var driver1 = TestHelper.GenerateTracked<DynamicTestsGenerator>(compilation1);
AssertRunReasons(driver1, IncrementalGeneratorRunReasons.New);

var compilation2 = TestHelper.ReplaceMethodDeclaration(compilation1, "SecondBuildDynamicTests",
"""
[DynamicTestBuilder]
public void SecondBuildDynamicTests(DynamicTestBuilderContext context)
{
}
"""
);

var driver2 = driver1.RunGenerators(compilation2);
AssertRunReasons(driver2, IncrementalGeneratorRunReasons.Cached, 0);
AssertRunReasons(driver2, IncrementalGeneratorRunReasons.New, 1);
}

private static void AssertRunReasons(
GeneratorDriver driver,
IncrementalGeneratorRunReasons reasons,
int outputIndex = 0
)
{
var runResult = driver.GetRunResult().Results[0];

TestHelper.AssertRunReason(runResult, DynamicTestsGenerator.ParseDynamicTests, reasons.BuildStep, outputIndex);
}
}
Loading