如何在xUnit中实现测试前后保存并还原桌面壁纸状态？

要实现测试开始前保存当前壁纸、测试结束后恢复的需求，xUnit里最直接的方式是利用IDisposable接口——xUnit会自动调用测试类的Dispose方法，刚好可以用来做测试后的清理恢复工作。下面是具体的实现步骤和完整代码：

实现思路 #

• 在测试类的构造函数中，读取当前系统注册表中记录的桌面壁纸路径，保存到私有字段中
• 实现IDisposable接口，在Dispose方法里调用你的SetDesktopWallpaper方法，把壁纸恢复成初始状态
• 你的原有测试逻辑可以保持不变，无需额外修改

完整代码示例 #

using System;
using System.IO;
using Microsoft.Win32;
using Xunit;

public class WallChangerTests : IDisposable
{
    // 存储初始壁纸路径的私有字段
    private readonly string _originalWallpaperPath;
    private const string KeyName = @"HKEY_CURRENT_USER\Control Panel\Desktop";
    private const string KeyVal1 = "WallPaper";

    public WallChangerTests()
    {
        // 在测试开始前，读取并保存当前壁纸路径
        _originalWallpaperPath = Registry.GetValue(KeyName, KeyVal1, null)?.ToString();
    }

    [Theory]
    [InlineData(@".\Resources\JpgWallpaper.jpg")]
    [InlineData(@".\Resources\PngWallpaper.png")]
    public void Changing_Wallpaper_File_Should_Work(string value)
    {
        WallChanger.SetDesktopWallpaper(value);
        Assert.Equal(Path.GetFullPath(value), Registry.GetValue(KeyName, KeyVal1, null).ToString());
    }

    public void Dispose()
    {
        // 在测试结束后，恢复初始壁纸
        if (!string.IsNullOrEmpty(_originalWallpaperPath))
        {
            WallChanger.SetDesktopWallpaper(_originalWallpaperPath);
        }
        // 如果初始壁纸为空（用户未设置壁纸），可以根据需求添加对应的处理逻辑
    }
}

额外说明 #

• 这里我们在构造函数中完成前置操作（保存初始状态），在Dispose中完成后置操作（恢复状态），这是xUnit中处理测试前后清理的标准方式
• 针对初始壁纸为空的场景，你可以根据实际需求补充处理逻辑，比如调用SetDesktopWallpaper传入空字符串或者默认壁纸路径
• 确保SetDesktopWallpaper方法能够正确处理空路径或者原有壁纸不存在的情况，避免恢复时出现异常

内容的提问来源于stack exchange，提问作者0lan