如何在kotest中使用ThreadLocal Webdriver实现并行浏览器执行

基础实现步骤 #

Kotest默认的JVM级并行执行基于线程调度，基础的ThreadLocal封装可以满足无协程切换场景的需求：

1. 封装ThreadLocal WebDriver容器

object WebDriverHolder {
    private val driverStorage = ThreadLocal<WebDriver>()
    
    fun getDriver(): WebDriver = driverStorage.get() ?: error("WebDriver未在当前线程初始化")
    fun setDriver(driver: WebDriver) = driverStorage.set(driver)
    fun clear() {
        driverStorage.get()?.quit()
        driverStorage.remove()
    }
}

2. 实现Kotest生命周期监听器
   在测试启动前初始化WebDriver，测试结束后销毁实例：

class WebDriverLifecycleListener : BeforeTestListener, AfterTestListener {
    override suspend fun beforeTest(testCase: TestCase) {
        val options = ChromeOptions().apply {
            addArguments("--headless=new", "--disable-gpu", "--no-sandbox")
        }
        WebDriverHolder.setDriver(ChromeDriver(options))
    }
    override suspend fun afterTest(testCase: TestCase, result: TestResult) {
        WebDriverHolder.clear()
    }
}

3. 开启并行配置
   在项目的kotest.properties中配置并行参数：

kotest.parallelism=4 # 并行执行的线程数，对应同时启动的浏览器实例数
kotest.framework.test.launch.parallel.mode=concurrent

也可以通过项目配置类实现：

class TestProjectConfig : AbstractProjectConfig() {
    override val parallelism = 4
    override fun listeners() = listOf(WebDriverLifecycleListener())
}

协程干扰问题解决 #

协程调度导致线程切换是ThreadLocal方案失效的核心原因：Kotest的测试函数和生命周期钩子都是suspend方法，默认调度器可能会将协程切换到其他线程执行，导致ThreadLocal读取不到对应WebDriver实例。
根据使用场景可以选择两种解决方案：

方案1：绑定测试协程到固定单线程调度器（推荐，适合绝大多数UI自动化场景） #

UI自动化操作本身是阻塞IO型任务，不需要多协程调度，直接为每个测试绑定独立的单线程调度器，避免协程切换：
修改项目配置类，重写协程调度器配置：

class TestProjectConfig : AbstractProjectConfig() {
    override val parallelism = 4
    override val coroutineDispatcher: CoroutineDispatcher
        get() = Executors.newSingleThreadExecutor().asCoroutineDispatcher()
    override fun listeners() = listOf(WebDriverLifecycleListener())
}

该配置下每个测试用例的协程全程在同一个线程执行，ThreadLocal可以正常工作。

方案2：用协程上下文携带WebDriver（适合需要多协程切换的复杂测试场景） #

如果测试逻辑需要用到协程切换，可以放弃ThreadLocal，改为通过协程上下文传递WebDriver实例：

1. 定义WebDriver协程上下文元素

class WebDriverContext(val driver: WebDriver) : AbstractCoroutineContextElement(Key) {
    companion object Key : CoroutineContext.Key<WebDriverContext>
}

2. 修改监听器，在测试启动时将WebDriver注入协程上下文

class WebDriverCoroutineListener : BeforeTestListener, AfterTestListener {
    override suspend fun beforeTest(testCase: TestCase): TestCase {
        val driver = ChromeDriver(ChromeOptions().apply { 
            addArguments("--headless=new", "--disable-gpu", "--no-sandbox") 
        })
        return testCase.copy(
            test = { params ->
                withContext(WebDriverContext(driver)) {
                    testCase.test(params)
                }
            }
        )
    }
    override suspend fun afterTest(testCase: TestCase, result: TestResult) {
        coroutineContext[WebDriverContext.Key]?.driver?.quit()
    }
}

3. 在测试用例中获取实例

class SampleWebTest : FunSpec({
    test("打开百度首页") {
        val driver = coroutineContext[WebDriverContext.Key]?.driver ?: error("WebDriver未初始化")
        driver.get("https://www.baidu.com")
        assert(driver.title.contains("百度"))
    }
})

注意事项 #

• 并行数建议不超过设备CPU核心数，避免过多浏览器实例占用过高资源导致测试不稳定
• 无论用哪种方案，测试结束后必须调用WebDriver的quit()方法，避免残留浏览器进程占用内存
• 使用协程上下文方案时，如果手动切换调度器，需要将WebDriverContext一起传递到新的协程上下文

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