如何配置TestNG仅在使用testng-console.xml执行时才运行@AfterSuite方法

实现思路 #

通过判断当前执行上下文是否来自testng-console.xml套件配置，决定是否执行邮件发送逻辑，以下为两种可直接落地的实现方案： #

方案1：通过套件自定义参数判断（推荐，兼容性最高） #

步骤1：修改testng-console.xml配置 #

在套件根标签下新增自定义参数，标记当前为套件执行模式：

<suite name="全量测试套件" parallel="none">
  <!-- 新增专属标识参数 -->
  <parameter name="runEmailAfterSuite" value="true"/>
  <!-- 原有测试类、用例配置保持不变 -->
  <test name="登录测试">
    <classes>
      <class name="com.yourpackage.LoginTest"/>
    </classes>
  </test>
</suite>

步骤2：修改BaseTest中@AfterSuite方法 #

注入TestNG上下文对象，读取参数判断是否执行邮件逻辑：

import org.testng.ITestContext;
import org.testng.annotations.AfterSuite;

public class BaseTest extends Base {
    // 原有其他方法保持不变

    @AfterSuite
    public void sendEmailWithExtentReport(ITestContext testContext) {
        // 读取套件级别的自定义参数
        String enableEmail = testContext.getSuite().getParameter("runEmailAfterSuite");
        // 仅参数存在且为true时执行邮件发送
        if ("true".equals(enableEmail)) {
            // 原有邮件发送的业务代码放在此处
        }
        // 无匹配参数时直接跳过，不执行任何操作
    }
}

方案2：通过套件文件名自动判断（无需修改XML配置） #

如果不想修改现有testng-console.xml，可以直接判断当前加载的套件文件名是否匹配：

@AfterSuite
public void sendEmailWithExtentReport(ITestContext testContext) {
    String suiteFilePath = testContext.getSuite().getXmlSuite().getFileName();
    // 判断文件名是否包含testng-console.xml，可根据实际文件路径调整匹配规则
    if (suiteFilePath != null && suiteFilePath.endsWith("testng-console.xml")) {
        // 执行邮件发送逻辑
    }
}

扩展：适配Maven命令执行场景 #

如果需要通过Maven命令灵活控制是否发邮件，可以配合JVM系统参数判断：
执行套件时传入参数：
mvn clean test -Dbrowsername=chrome -DrunEmailReport=true
@AfterSuite方法调整为：

@AfterSuite
public void sendEmailWithExtentReport() {
    String enableEmail = System.getProperty("runEmailReport");
    if ("true".equals(enableEmail)) {
        // 执行邮件发送
    }
}

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