基于AWS Lambda的JUnit 5测试枚举与执行技术问询

我们团队采用JUnit覆盖从单元测试到全阶段测试，为解决端到端（尤其是UI）测试的效率瓶颈，正在探索将测试类封装为AWS Lambda函数——通过弹性并行调用把测试时长从小时级压缩到分钟级。相比JUnit控制台启动器的多线程模式，Lambda能突破单机器CPU资源限制，避免闲置资源浪费。

针对需求：Lambda无参数调用时枚举所有测试用例并返回JSON详情；指定测试用例时执行该测试（含前后置钩子）并返回结果，以下是核心实现代码：

1. 枚举当前所有可见测试用例的Java代码 #

使用JUnit 5的Launcher API扫描并提取测试用例信息，可直接在Lambda入口函数中调用：

import org.junit.platform.engine.TestDescriptor;
import org.junit.platform.engine.TestTag;
import org.junit.platform.launcher.Launcher;
import org.junit.platform.launcher.LauncherDiscoveryRequest;
import org.junit.platform.launcher.core.LauncherDiscoveryRequestBuilder;
import org.junit.platform.launcher.core.LauncherFactory;
import com.fasterxml.jackson.databind.ObjectMapper;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

public class TestEnumerator {
    public String enumerateTests(Class<?> testClass) throws Exception {
        // 构建测试发现请求，指定目标测试类
        LauncherDiscoveryRequest request = LauncherDiscoveryRequestBuilder.request()
                .selectors(LauncherDiscoveryRequestBuilder.selectClass(testClass))
                .build();

        Launcher launcher = LauncherFactory.create();
        // 获取包含所有测试节点的测试计划
        org.junit.platform.launcher.TestPlan testPlan = launcher.discover(request);

        List<Map<String, Object>> testCases = new ArrayList<>();
        // 遍历所有测试方法节点
        testPlan.getChildren(testPlan.getRoot()).forEach(rootChild -> {
            testPlan.getDescendants(rootChild).stream()
                    .filter(desc -> desc.getType() == TestDescriptor.Type.TEST_METHOD)
                    .forEach(testMethod -> {
                        Map<String, Object> testDetails = Map.of(
                                "testClassName", testMethod.getParent().getDisplayName(),
                                "testMethodName", testMethod.getDisplayName(),
                                "tags", testMethod.getTags().stream().map(TestTag::getName).collect(Collectors.toList())
                        );
                        testCases.add(testDetails);
                    });
        });

        // 转换为JSON格式返回
        ObjectMapper mapper = new ObjectMapper();
        return mapper.writeValueAsString(testCases);
    }
}

2. 触发指定测试用例执行的Java代码 #

通过指定测试方法选择器执行目标测试，自动包含前后置钩子的执行，并收集结果：

import org.junit.platform.launcher.Launcher;
import org.junit.platform.launcher.LauncherDiscoveryRequest;
import org.junit.platform.launcher.core.LauncherDiscoveryRequestBuilder;
import org.junit.platform.launcher.core.LauncherFactory;
import org.junit.platform.launcher.listeners.SummaryGeneratingListener;
import org.junit.platform.launcher.listeners.TestExecutionSummary;
import com.fasterxml.jackson.databind.ObjectMapper;
import java.util.Map;

public class TestExecutor {
    public String executeTest(Class<?> testClass, String testMethodName) throws Exception {
        // 构建指定测试方法的发现请求
        LauncherDiscoveryRequest request = LauncherDiscoveryRequestBuilder.request()
                .selectors(LauncherDiscoveryRequestBuilder.selectMethod(testClass, testMethodName))
                .build();

        Launcher launcher = LauncherFactory.create();
        SummaryGeneratingListener listener = new SummaryGeneratingListener();

        // 执行测试
        launcher.execute(request, listener);

        // 整理执行结果
        TestExecutionSummary summary = listener.getSummary();
        TestExecutionSummary.Failure failure = summary.getFailures().stream().findFirst().orElse(null);

        Map<String, Object> result = Map.of(
                "testClassName", testClass.getName(),
                "testMethodName", testMethodName,
                "successful", summary.getTestsSucceededCount() == 1,
                "durationMs", summary.getTotalDuration().toMillis(),
                "failureReason", failure != null ? failure.getException().getMessage() : null,
                "failureStackTrace", failure != null ? failure.getTrace() : null
        );

        // 转换为JSON格式返回
        ObjectMapper mapper = new ObjectMapper();
        return mapper.writeValueAsString(result);
    }
}

Lambda入口整合示例 #

将上述逻辑整合到Lambda handler中，根据输入参数判断执行分支：

import com.amazonaws.services.lambda.runtime.Context;
import com.amazonaws.services.lambda.runtime.RequestHandler;
import java.util.Map;

public class TestLambdaHandler implements RequestHandler<Map<String, String>, String> {
    private final TestEnumerator enumerator = new TestEnumerator();
    private final TestExecutor executor = new TestExecutor();
    // 替换为你的目标测试类
    private static final Class<?> TEST_CLASS = YourTargetTestClass.class;

    @Override
    public String handleRequest(Map<String, String> input, Context context) {
        try {
            if (input == null || input.isEmpty()) {
                // 无参数调用，返回测试用例列表
                return enumerator.enumerateTests(TEST_CLASS);
            } else {
                String testMethod = input.get("testMethodName");
                if (testMethod == null || testMethod.isBlank()) {
                    return "{\"error\":\"请指定testMethodName参数\"}";
                }
                // 指定测试用例，执行并返回结果
                return executor.executeTest(TEST_CLASS, testMethod);
            }
        } catch (Exception e) {
            return "{\"error\":\"执行失败: " + e.getMessage() + "\"}";
        }
    }
}

依赖说明 #

需在Maven pom.xml中添加以下依赖：

<dependencies>
    <dependency>
        <groupId>org.junit.platform</groupId>
        <artifactId>junit-platform-launcher</artifactId>
        <version>1.10.0</version>
    </dependency>
    <dependency>
        <groupId>org.junit.jupiter</groupId>
        <artifactId>junit-jupiter-api</artifactId>
        <version>5.10.0</version>
    </dependency>
    <dependency>
        <groupId>com.fasterxml.jackson.core</groupId>
        <artifactId>jackson-databind</artifactId>
        <version>2.15.2</version>
    </dependency>
    <dependency>
        <groupId>com.amazonaws</groupId>
        <artifactId>aws-lambda-java-core</artifactId>
        <version>1.2.21</version>
    </dependency>
</dependencies>

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