TestNG @BeforeSuite中初始化的lateinit变量调用时提示未初始化怎么解决

问题根因 #

TestNG 中 @BeforeSuite 注解的方法全局仅执行 1 次，但执行该方法时，TestNG 只会随机选取一个继承了 BaseTest 的测试类实例调用 init 方法，仅该实例的 driver 实例变量会被赋值。
而每个测试类运行时都会创建全新的实例，其余测试类实例的 driver 变量从未被赋值，因此会抛出未初始化异常。@BeforeClass 能正常运行是因为它会在每个测试类实例初始化后执行，每个实例的 driver 都能被单独赋值。

解决方案 #

方案1：静态变量存储（适合单 suite 串行测试） #

将 driver 改为类级别的静态变量，所有测试类实例共享同一份driver，仅初始化一次即可全局使用：

abstract class BaseTest {
    // 放在companion object中实现静态变量效果
    companion object {
        lateinit var driver: AppiumDriver<MobileElement>
    }

    @Parameters("platform")
    @BeforeSuite(alwaysRun = true)
    fun init(platform: String) {
        Configuration.isIOS = platform.lowercase() == "ios"
        // 增加初始化判断，避免异常场景下重复初始化
        if (!::driver.isInitialized) {
            driver = <driver initialization>
            println(driver)
        }
    }
}

测试类中直接使用driver即可，无需额外修改。

方案2：ISuite 上下文存储（适合多 suite 并行测试） #

如果需要多suite并行、避免静态变量互相干扰，可以将driver存入TestNG的Suite上下文，每个测试类初始化时从上下文取driver赋值给实例变量：

abstract class BaseTest {
    lateinit var driver: AppiumDriver<MobileElement>

    @Parameters("platform")
    @BeforeSuite(alwaysRun = true)
    fun init(platform: String, suite: ISuite) {
        Configuration.isIOS = platform.lowercase() == "ios"
        val driver = <driver initialization>
        // 将driver存入suite全局上下文
        suite.setAttribute("appiumDriver", driver)
        println(driver)
    }

    @BeforeClass(alwaysRun = true)
    fun fetchDriver(context: ITestContext) {
        // 每个测试类初始化时从suite上下文取出driver赋值给当前实例
        driver = context.suite.getAttribute("appiumDriver") as AppiumDriver<MobileElement>
    }
}

补充注意事项 #

• 使用静态变量方案时，需要在@AfterSuite中主动关闭driver并重置变量，避免内存泄漏
• 如果需要多线程并行执行测试用例，建议使用ThreadLocal包装静态driver，保证每个线程的driver实例隔离
• 多设备并行测试场景不适合全局单driver初始化，建议将初始化逻辑下沉到@BeforeTest或@BeforeMethod层级

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