在MSTest中能否为测试类级别配置[DataRow]实现Selenium数据驱动测试？

需求说明 #

• 需要支持4种不同类型用户登录被测应用，按用户权限划分测试类：
  • TestClass1：仅UserType1、UserType2可执行的测试
  • TestClass2：仅UserType1、UserType4可执行的测试

当前现状 #

• 目前为每个测试方法添加[DataRow]实现数据驱动，可行但存在重复操作：同一用户每次执行测试都要重复登录，效率低下。

期望目标 #

• 测试类级别复用用户登录：比如TestClass1中，UserType1登录一次后执行该类所有测试方法，再切换UserType2登录一次执行所有测试方法，其他类同理。

疑问 #

是否可以在测试类级别而非单个测试方法级别使用数据源？

现有代码示例（修正语法错误后） #

[TestClass]
public class TestClass1
{
   [DataTestMethod]
   [DataRow("UserType1", "Selenium_pssw1")]
   [DataRow("UserType2", "Selenium_pssw2")]
   public void Test1 (string username, string password)
   {
       // 测试逻辑与断言
   }

   [DataTestMethod]
   [DataRow("UserType1", "Selenium_pssw1")]
   [DataRow("UserType2", "Selenium_pssw2")]
   public void Test2 (string username, string password)
   {
       // 测试逻辑与断言
   }
}

MSTest本身没有直接的类级DataRow属性，但可以通过自定义测试初始化逻辑+类级数据源实现需求，核心是让测试类针对每个用户实例化一次，复用该用户的登录状态。

步骤1：封装用户数据类 #

先把用户信息封装成实体，方便后续复用：

public class TestUser
{
    public string Username { get; set; }
    public string Password { get; set; }
}

步骤2：自定义类级数据源 #

实现ITestDataSource接口，提供测试类需要的用户数据：

// 自定义数据源提供器，返回需要的用户列表
public class UserDataSource : Attribute, ITestDataSource
{
    public IEnumerable<object[]> GetData(MethodInfo methodInfo)
    {
        // 可替换为从配置文件/Excel读取，这里硬编码示例数据
        yield return new object[] { new TestUser { Username = "UserType1", Password = "Selenium_pssw1" } };
        yield return new object[] { new TestUser { Username = "UserType2", Password = "Selenium_pssw2" } };
    }

    public string GetDisplayName(MethodInfo methodInfo, object[] data)
    {
        var user = data[0] as TestUser;
        return $"[{user.Username}] {methodInfo.Name}";
    }
}

步骤3：改造测试类，实现登录复用 #

让测试类通过构造函数接收用户数据，在TestInitialize中完成一次登录，所有测试方法复用该登录状态：

[TestClass]
public class TestClass1
{
    private IWebDriver _driver;
    private TestUser _currentUser;

    // 每个用户对应一个测试类实例，构造函数注入用户数据
    public TestClass1(TestUser user)
    {
        _currentUser = user;
    }

    // 实例初始化时执行登录，每个用户仅登录一次
    [TestInitialize]
    public void Setup()
    {
        _driver = new ChromeDriver();
        // 执行登录流程
        _driver.Navigate().GoToUrl("你的登录页地址");
        _driver.FindElement(By.Id("username")).SendKeys(_currentUser.Username);
        _driver.FindElement(By.Id("password")).SendKeys(_currentUser.Password);
        _driver.FindElement(By.Id("login-btn")).Click();
        // 验证登录成功
        Assert.IsTrue(_driver.FindElement(By.CssSelector(".user-avatar")).Displayed);
    }

    // 测试方法直接复用类级的_driver和_currentUser，无需重复传参
    [TestMethod]
    public void Test1()
    {
        _driver.Navigate().GoToUrl("测试页面1");
        // 针对UserType1/UserType2的权限断言逻辑
    }

    [TestMethod]
    public void Test2()
    {
        _driver.Navigate().GoToUrl("测试页面2");
        // 针对UserType1/UserType2的权限断言逻辑
    }

    // 实例销毁时关闭浏览器，每个用户对应一次销毁
    [TestCleanup]
    public void Teardown()
    {
        _driver.Quit();
    }
}

核心逻辑说明 #

• 每个用户会生成一个独立的测试类实例，[TestInitialize]为每个实例执行一次登录，所有测试方法复用该实例的浏览器会话
• 自定义数据源可灵活扩展，比如从JSON/Excel批量读取用户数据，避免硬编码
• 测试结果会通过GetDisplayName显示用户标识，清晰区分不同用户的执行结果

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