TUnit项目中共享测试资源的生命周期管理实践

引言

在单元测试和集成测试中，测试资源的生命周期管理是一个常见但容易被忽视的问题。TUnit作为一个现代化的测试框架，提供了强大的依赖注入和资源共享机制。本文将深入探讨如何在TUnit中有效管理具有不同生命周期的测试资源。

测试资源共享的基本概念

TUnit框架支持三种主要的资源共享级别：

1. PerTest：每个测试方法创建新实例
2. PerClass：同一测试类中的所有测试方法共享实例
3. PerAssembly：整个测试程序集中的所有测试共享实例

这种机制特别适合管理测试基础设施，如数据库连接、Web服务器模拟器等昂贵资源。

实际应用场景分析

考虑一个典型场景：我们有一个测试容器池(PerAssembly级别)，需要在多个测试类(PerClass级别)中使用。测试工厂类(PerClass级别)在初始化时需要访问这个池。

初始实现方案

[ClassDataSource<MyFactory>(Shared = SharedType.PerClass)]
public class MyTests1(MyFactory factory)
{
    // 测试代码
}

public class MyFactory : IAsyncInitializer
{
    [ClassDataSource<MyServer>(Shared = SharedType.PerAssembly)]
    public required MyServer Server { get; init; }
    
    public async Task InitializeAsync()
    {
        // 使用Server进行初始化
    }
}

生命周期问题

在上述实现中，当测试运行时：

1. MyServer实例按预期只初始化一次(PerAssembly)
2. 每个测试类创建自己的MyFactory实例(PerClass)
3. 但当单个测试类完成时，MyServer会被意外释放

解决方案演进

方案一：通过TestContext访问

在InitializeAsync方法中，可以通过TestContext.Current访问所有注入的属性：

public async Task InitializeAsync()
{
    var server = TestContext.Current!.TestDetails.TestClassInjectedPropertyArguments
        .OfType<MyServer>()
        .FirstOrDefault();
    // 使用server
}

这种方法虽然可行，但不够优雅，属于临时解决方案。

方案二：属性注入的正确方式

TUnit 0.24版本提供了更优雅的解决方案：

public class MyFactory : IAsyncInitializer
{
    [ClassDataSource<MyServer>(Shared = SharedType.PerAssembly)]
    public required MyServer Server { get; init; }
    
    public async Task InitializeAsync()
    {
        // 直接使用this.Server
    }
}

生命周期管理最佳实践

对于需要精确控制生命周期的场景，可以实现IAsyncDisposable接口：

public class MyServer : IAsyncInitializer, IAsyncDisposable
{
    public async Task InitializeAsync()
    {
        // 初始化代码
    }
    
    public async ValueTask DisposeAsync()
    {
        // 清理代码
    }
}

高级场景：嵌套资源共享

TUnit支持嵌套的资源共享，这是其强大之处：

[ClassDataSource<MyFactory>(Shared = SharedType.PerClass)]
public class MyTests1(MyFactory factory)
{
    // 测试代码
}

[ClassDataSource<MyFactory>(Shared = SharedType.PerClass)]
public class MyTests2(MyFactory factory)
{
    // 测试代码
}

public class MyFactory : IAsyncInitializer, IAsyncDisposable
{
    [ClassDataSource<MyServer>(Shared = SharedType.PerAssembly)]
    public required MyServer Server { get; init; }
    // 其他成员
}

在这种结构中：

• MyServer实例在整个程序集生命周期内保持
• 每个测试类有独立的MyFactory实例
• 当所有测试完成后，MyServer才会被释放

结论

TUnit框架提供了灵活而强大的测试资源生命周期管理机制。通过合理使用ClassDataSource和共享级别，可以构建高效、可靠的测试套件。关键点在于：

1. 明确各类资源的生命周期需求
2. 正确使用属性注入而非手动解析
3. 实现适当的初始化和清理接口
4. 理解嵌套资源共享的行为

掌握这些技巧后，开发者可以构建出既高效又可靠的自动化测试基础设施。