如何在C#+SpecFlow中实现Selenium操作后自动截图

一、基础准备：截图工具类与开关控制 #

先实现一个通用的截图工具，内置开关控制，避免不需要时生成截图。可以从配置文件读取开关状态，也可以用静态变量临时设置。

using OpenQA.Selenium;
using System.IO;

public class ScreenshotHelper
{
    // 截图开关，可从App.config读取（示例用静态变量，实际可替换为配置读取）
    public static bool EnableScreenshot { get; set; } = true;

    public static void TakeScreenshot(IWebDriver driver, string actionLabel)
    {
        if (!EnableScreenshot) return;

        try
        {
            var screenshot = ((ITakesScreenshot)driver).GetScreenshot();
            var saveDir = Path.Combine(Directory.GetCurrentDirectory(), "TestScreenshots");
            Directory.CreateDirectory(saveDir);
            var fileName = $"{actionLabel}_{DateTime.Now:yyyyMMddHHmmssfff}.png";
            screenshot.SaveAsFile(Path.Combine(saveDir, fileName), ScreenshotImageFormat.Png);
        }
        catch (Exception ex)
        {
            // 日志记录，避免截图失败中断测试流程
            Console.WriteLine($"截图失败：{ex.Message}");
        }
    }
}

在SpecFlow的全局钩子中初始化开关（比如从配置读取）：

using TechTalk.SpecFlow;

[Binding]
public class GlobalHooks
{
    [BeforeTestRun]
    public static void InitializeScreenshotSetting()
    {
        // 示例：从App.config读取开关状态
        // ScreenshotHelper.EnableScreenshot = bool.Parse(ConfigurationManager.AppSettings["EnableScreenshot"]);
        ScreenshotHelper.EnableScreenshot = true;
    }
}

二、方案1：扩展方法实现简洁调用（新手友好） #

如果不想完全自动，用扩展方法给IWebElement添加带截图的操作方法，调用时只需替换原有的Click()为ClickWithScreenshot()，简洁且可控。

using OpenQA.Selenium;

public static class WebElementExtensions
{
    // 带截图的点击操作
    public static void ClickWithScreenshot(this IWebElement element, IWebDriver driver, string actionDesc)
    {
        element.Click();
        ScreenshotHelper.TakeScreenshot(driver, actionDesc);
    }

    // 带截图的输入操作
    public static void SendKeysWithScreenshot(this IWebElement element, IWebDriver driver, string text, string actionDesc)
    {
        element.SendKeys(text);
        ScreenshotHelper.TakeScreenshot(driver, actionDesc);
    }
}

使用示例（SpecFlow步骤） #

[Binding]
public class LoginSteps
{
    private readonly IWebDriver _driver;

    public LoginSteps(IWebDriver driver)
    {
        _driver = driver;
    }

    [When(@"我点击登录按钮")]
    public void WhenClickLoginButton()
    {
        var loginBtn = _driver.FindElement(By.Id("login-btn"));
        loginBtn.ClickWithScreenshot(_driver, "点击登录按钮");
    }

    [When(@"我输入用户名""(.*)""")]
    public void WhenInputUsername(string username)
    {
        var usernameInput = _driver.FindElement(By.Id("username-input"));
        usernameInput.SendKeysWithScreenshot(_driver, username, $"输入用户名：{username}");
    }
}

三、方案2：代理模式实现完全自动截图（无需修改原有操作代码） #

如果希望完全自动触发截图，不用修改原有的Click()调用，可以用代理模式封装IWebDriver和IWebElement，拦截指定操作后自动执行截图。

1. 封装IWebElement代理 #

using OpenQA.Selenium;
using System.Collections.ObjectModel;

public class WebElementProxy : IWebElement
{
    private readonly IWebElement _originalElement;
    private readonly IWebDriver _driver;

    public WebElementProxy(IWebElement originalElement, IWebDriver driver)
    {
        _originalElement = originalElement;
        _driver = driver;
    }

    // 拦截Click操作，执行后自动截图
    public void Click()
    {
        _originalElement.Click();
        ScreenshotHelper.TakeScreenshot(_driver, "执行Click操作");
    }

    // 拦截SendKeys操作，执行后自动截图（可根据需求关闭）
    public void SendKeys(string text)
    {
        _originalElement.SendKeys(text);
        ScreenshotHelper.TakeScreenshot(_driver, $"执行SendKeys操作：{text}");
    }

    // 其他IWebElement接口方法直接委托给原元素
    public string TagName => _originalElement.TagName;
    public string Text => _originalElement.Text;
    public bool Enabled => _originalElement.Enabled;
    public bool Selected => _originalElement.Selected;
    public Point Location => _originalElement.Location;
    public Size Size => _originalElement.Size;
    public bool Displayed => _originalElement.Displayed;

    public void Clear() => _originalElement.Clear();
    public void Submit() => _originalElement.Submit();
    public IWebElement FindElement(By by) => new WebElementProxy(_originalElement.FindElement(by), _driver);
    public ReadOnlyCollection<IWebElement> FindElements(By by) => 
        _originalElement.FindElements(by).Select(e => new WebElementProxy(e, _driver)).ToList().AsReadOnly();
    public string GetAttribute(string attributeName) => _originalElement.GetAttribute(attributeName);
    public string GetCssValue(string propertyName) => _originalElement.GetCssValue(propertyName);
    public string GetDomAttribute(string attributeName) => _originalElement.GetDomAttribute(attributeName);
    public string GetDomProperty(string propertyName) => _originalElement.GetDomProperty(propertyName);
    public string GetProperty(string propertyName) => _originalElement.GetProperty(propertyName);
}

2. 封装IWebDriver代理 #

让FindElement返回代理后的元素，确保所有元素操作都被拦截：

using OpenQA.Selenium;
using System.Collections.ObjectModel;

public class WebDriverProxy : IWebDriver
{
    private readonly IWebDriver _originalDriver;

    public WebDriverProxy(IWebDriver originalDriver)
    {
        _originalDriver = originalDriver;
    }

    public IWebElement FindElement(By by)
    {
        var originalElement = _originalDriver.FindElement(by);
        return new WebElementProxy(originalElement, _originalDriver);
    }

    public ReadOnlyCollection<IWebElement> FindElements(By by)
    {
        return _originalDriver.FindElements(by).Select(e => new WebElementProxy(e, _originalDriver)).ToList().AsReadOnly();
    }

    // 其他IWebDriver接口方法直接委托给原驱动
    public string Url
    {
        get => _originalDriver.Url;
        set => _originalDriver.Url = value;
    }

    public string Title => _originalDriver.Title;
    public string PageSource => _originalDriver.PageSource;
    public string CurrentWindowHandle => _originalDriver.CurrentWindowHandle;
    public ReadOnlyCollection<string> WindowHandles => _originalDriver.WindowHandles;

    public void Close() => _originalDriver.Close();
    public void Quit() => _originalDriver.Quit();
    public INavigation Navigate() => _originalDriver.Navigate();
    public IOptions Manage() => _originalDriver.Manage();
    public ITargetLocator SwitchTo() => _originalDriver.SwitchTo();

    public void Dispose() => _originalDriver.Dispose();
}

3. 注册代理驱动到SpecFlow依赖注入 #

using OpenQA.Selenium.Chrome;
using TechTalk.SpecFlow;

[Binding]
public class DriverHooks
{
    [BeforeScenario]
    public void InitializeDriver(IObjectContainer container)
    {
        var originalDriver = new ChromeDriver();
        var proxiedDriver = new WebDriverProxy(originalDriver);
        container.RegisterInstanceAs<IWebDriver>(proxiedDriver);
    }

    [AfterScenario]
    public void CleanupDriver(IWebDriver driver)
    {
        driver.Quit();
        driver.Dispose();
    }
}

使用示例（无需修改原有操作代码） #

[When(@"我点击登录按钮")]
public void WhenClickLoginButton()
{
    var loginBtn = _driver.FindElement(By.Id("login-btn"));
    loginBtn.Click(); // 自动触发截图，无需额外调用
}

四、操作类型过滤（按需触发截图） #

如果只需要在特定操作（比如仅Click）后截图，只需在WebElementProxy中对应方法保留截图逻辑，其他方法注释掉即可。例如：

public void Click()
{
    _originalElement.Click();
    ScreenshotHelper.TakeScreenshot(_driver, "Click操作");
}

public void SendKeys(string text)
{
    _originalElement.SendKeys(text);
    // 注释掉此行，SendKeys操作后不截图
    // ScreenshotHelper.TakeScreenshot(_driver, $"SendKeys操作：{text}");
}

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