在Selenium中如何在使用Dataprovider时获取测试结果并将其保存到excel中

<dependencies>
    <!-- TestNG -->
    <dependency>
        <groupId>org.testng</groupId>
        <artifactId>testng</artifactId>
        <version>7.4.0</version>
    </dependency>
    <!-- Apache POI -->
    <dependency>
        <groupId>org.apache.poi</groupId>
        <artifactId>poi</artifactId>
        <version>5.0.0</version>
    </dependency>
    <dependency>
        <groupId>org.apache.poi</groupId>
        <artifactId>poi-ooxml</artifactId>
        <version>5.0.0</version>
    </dependency>
</dependencies>

import org.testng.annotations.DataProvider;
import org.testng.annotations.Listeners;
import org.testng.annotations.Test;
import org.testng.ITestListener;
import org.testng.ITestResult;
import org.apache.poi.ss.usermodel.*;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import java.io.FileOutputStream;
import java.io.IOException;

@Listeners(TestListener.class)
public class SeleniumTest {
    
    @Test(dataProvider = "testData")
    public void testMethod(String data) {
        // 测试逻辑
    }
    
    @DataProvider(name = "testData")
    public Object[][] testData() {
        // 提供测试数据
        return new Object[][] {
            {"data1"},
            {"data2"},
            {"data3"}
        };
    }
}

import org.testng.ITestListener;
import org.testng.ITestResult;
import org.apache.poi.ss.usermodel.*;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import java.io.FileOutputStream;
import java.io.IOException;

public class TestListener implements ITestListener {
    
    @Override
    public void onTestFailure(ITestResult result) {
        saveResultToExcel(result, "Failure");
    }
    
    @Override
    public void onTestSuccess(ITestResult result) {
        saveResultToExcel(result, "Success");
    }
    
    private void saveResultToExcel(ITestResult result, String status) {
        String methodName = result.getMethod().getMethodName();
        Workbook workbook = new XSSFWorkbook();
        Sheet sheet = workbook.createSheet("Test Results");
        Row row = sheet.createRow(0);
        Cell cell = row.createCell(0);
        cell.setCellValue("Test Method");
        cell = row.createCell(1);
        cell.setCellValue("Status");
        row = sheet.createRow(1);
        cell = row.createCell(0);
        cell.setCellValue(methodName);
        cell = row.createCell(1);
        cell.setCellValue(status);
        try {
            FileOutputStream outputStream = new FileOutputStream("test_results.xlsx");
            workbook.write(outputStream);
            workbook.close();
            outputStream.close();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}