gewuyou/GFramework
1
0
Fork 0
mirror of https://github.com/GeWuYou/GFramework.git synced 2026-05-13 22:25:37 +08:00
Code Issues Packages Projects Releases Wiki Activity

test(EnumExtensions): 添加枚举扩展生成器快照测试

Browse Source 
- 实现基本枚举的 IsMethods 快照测试
- 实现基本枚举的 IsInMethod 快照测试
- 实现带标志值枚举的快照测试
- 实现禁用 IsMethods 选项的快照测试
- 实现禁用 IsInMethod 选项的快照测试
- 创建统一的源代码构建方法用于测试场景
This commit is contained in:
GeWuYou 2026-04-16 21:49:59 +08:00
parent 534818f5f6
commit 08b12ae852
1 changed files with 73 additions and 129 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

140
GFramework.SourceGenerators.Tests/Enums/EnumExtensionsGeneratorSnapshotTests.cs
View File

@ -7,35 +7,20 @@ namespace GFramework.SourceGenerators.Tests.Enums;
[TestFixture] [TestFixture]
public class EnumExtensionsGeneratorSnapshotTests public class EnumExtensionsGeneratorSnapshotTests
{ {
private const string EnumAttributeNamespace = "GFramework.Core.SourceGenerators.Abstractions.Enums";
[Test] [Test]
public async Task Snapshot_BasicEnum_IsMethods() public async Task Snapshot_BasicEnum_IsMethods()
{ {
const string source = """ var source = BuildSource(
using System; """
namespace GFramework.Core.SourceGenerators.Abstractions.Enums
{
[AttributeUsage(AttributeTargets.Enum)]
public sealed class GenerateEnumExtensionsAttribute : Attribute
{
public bool GenerateIsMethods { get; set; } = true;
public bool GenerateIsInMethod { get; set; } = true;
}
}
namespace TestApp
{
using GFramework.Core.SourceGenerators.Abstractions.Enums;
[GenerateEnumExtensions]
public enum Status public enum Status
{ {
Active, Active,
Inactive, Inactive,
Pending Pending
} }
} """);
""";
await GeneratorSnapshotTest<EnumExtensionsGenerator>.RunAsync( await GeneratorSnapshotTest<EnumExtensionsGenerator>.RunAsync(
source, source,
@ -50,31 +35,14 @@ public class EnumExtensionsGeneratorSnapshotTests
[Test] [Test]
public async Task Snapshot_BasicEnum_IsInMethod() public async Task Snapshot_BasicEnum_IsInMethod()
{ {
const string source = """ var source = BuildSource(
using System; """
namespace GFramework.Core.SourceGenerators.Abstractions.Enums
{
[AttributeUsage(AttributeTargets.Enum)]
public sealed class GenerateEnumExtensionsAttribute : Attribute
{
public bool GenerateIsMethods { get; set; } = true;
public bool GenerateIsInMethod { get; set; } = true;
}
}
namespace TestApp
{
using GFramework.Core.SourceGenerators.Abstractions.Enums;
[GenerateEnumExtensions]
public enum Status public enum Status
{ {
Active, Active,
Inactive Inactive
} }
} """);
""";
await GeneratorSnapshotTest<EnumExtensionsGenerator>.RunAsync( await GeneratorSnapshotTest<EnumExtensionsGenerator>.RunAsync(
source, source,
@ -89,24 +57,8 @@ public class EnumExtensionsGeneratorSnapshotTests
[Test] [Test]
public async Task Snapshot_EnumWithFlagValues() public async Task Snapshot_EnumWithFlagValues()
{ {
const string source = """ var source = BuildSource(
using System; """
namespace GFramework.Core.SourceGenerators.Abstractions.Enums
{
[AttributeUsage(AttributeTargets.Enum)]
public sealed class GenerateEnumExtensionsAttribute : Attribute
{
public bool GenerateIsMethods { get; set; } = true;
public bool GenerateIsInMethod { get; set; } = true;
}
}
namespace TestApp
{
using GFramework.Core.SourceGenerators.Abstractions.Enums;
[GenerateEnumExtensions]
[Flags] [Flags]
public enum Permissions public enum Permissions
{ {
@ -115,8 +67,7 @@ public class EnumExtensionsGeneratorSnapshotTests
Write = 2, Write = 2,
Execute = 4 Execute = 4
} }
} """);
""";
await GeneratorSnapshotTest<EnumExtensionsGenerator>.RunAsync( await GeneratorSnapshotTest<EnumExtensionsGenerator>.RunAsync(
source, source,
@ -131,31 +82,15 @@ public class EnumExtensionsGeneratorSnapshotTests
[Test] [Test]
public async Task Snapshot_DisableIsMethods() public async Task Snapshot_DisableIsMethods()
{ {
const string source = """ var source = BuildSource(
using System; """
namespace GFramework.Core.SourceGenerators.Abstractions.Enums
{
[AttributeUsage(AttributeTargets.Enum)]
public sealed class GenerateEnumExtensionsAttribute : Attribute
{
public bool GenerateIsMethods { get; set; } = true;
public bool GenerateIsInMethod { get; set; } = true;
}
}
namespace TestApp
{
using GFramework.Core.SourceGenerators.Abstractions.Enums;
[GenerateEnumExtensions(GenerateIsMethods = false)]
public enum Status public enum Status
{ {
Active, Active,
Inactive Inactive
} }
} """,
"""; "[GenerateEnumExtensions(GenerateIsMethods = false)]");
await GeneratorSnapshotTest<EnumExtensionsGenerator>.RunAsync( await GeneratorSnapshotTest<EnumExtensionsGenerator>.RunAsync(
source, source,
@ -170,10 +105,32 @@ public class EnumExtensionsGeneratorSnapshotTests
[Test] [Test]
public async Task Snapshot_DisableIsInMethod() public async Task Snapshot_DisableIsInMethod()
{ {
const string source = """ var source = BuildSource(
"""
public enum Status
{
Active,
Inactive
}
""",
"[GenerateEnumExtensions(GenerateIsInMethod = false)]");
await GeneratorSnapshotTest<EnumExtensionsGenerator>.RunAsync(
source,
Path.Combine(
TestContext.CurrentContext.TestDirectory,
"enums",
"snapshots",
"EnumExtensionsGenerator",
"DisableIsInMethod"));
}
private static string BuildSource(string enumBody, string attributeUsage = "[GenerateEnumExtensions]")
{
return $$"""
using System; using System;
namespace GFramework.Core.SourceGenerators.Abstractions.Enums namespace {{EnumAttributeNamespace}}
{ {
[AttributeUsage(AttributeTargets.Enum)] [AttributeUsage(AttributeTargets.Enum)]
public sealed class GenerateEnumExtensionsAttribute : Attribute public sealed class GenerateEnumExtensionsAttribute : Attribute
@ -185,24 +142,11 @@ public class EnumExtensionsGeneratorSnapshotTests
namespace TestApp namespace TestApp
{ {
using GFramework.Core.SourceGenerators.Abstractions.Enums; using {{EnumAttributeNamespace}};
[GenerateEnumExtensions(GenerateIsInMethod = false)] {{attributeUsage}}
public enum Status {{enumBody}}
{
Active,
Inactive
}
} }
"""; """;
await GeneratorSnapshotTest<EnumExtensionsGenerator>.RunAsync(
source,
Path.Combine(
TestContext.CurrentContext.TestDirectory,
"enums",
"snapshots",
"EnumExtensionsGenerator",
"DisableIsInMethod"));
} }
} }