能否向Espresso测试发送数据？自定义IdlingResource相关疑问

好问题！这两个需求在Espresso测试场景下都是完全可以实现的，我结合实际开发经验给你详细说明：

1. 能否向Espresso测试发送数据？ #

当然可以。Espresso本身没有直接提供跨UI和测试的通信API，但我们可以通过全局状态容器或者自定义通信类来实现数据传递。最常用的方式是创建一个静态单例类，作为UI和测试之间的桥梁：

// 全局数据持有类，UI和测试都可以访问
object TestDataHolder {
    private val dataMap = mutableMapOf<String, Any?>()

    fun putData(key: String, data: Any?) {
        dataMap[key] = data
    }

    fun getData(key: String): Any? {
        return dataMap[key]
    }
}

在UI层的业务逻辑中，你可以在合适的时机把数据存入这个类：

// UI代码中发送数据到测试
TestDataHolder.putData("async_operation_id", "operation_123")

然后在Espresso测试中，就可以读取这些数据：

// Espresso测试代码中接收数据
val operationId = TestDataHolder.getData("async_operation_id") as String
// 基于数据执行后续测试逻辑

2. 自定义IdlingResource的两个细分问题 #

2.1 UI向Espresso测试发送数据，让IdlingResource计数增加 #

完全可行。你只需要让自定义的IdlingResource暴露一个可以被UI调用的方法，用来手动增加计数。这里是一个典型的实现示例：

class CustomIdlingResource(private val name: String) : IdlingResource {
    private val count = AtomicInteger(0)
    private var callback: IdlingResource.ResourceCallback? = null

    // UI层可以调用这个方法，增加空闲等待计数
    fun incrementCount() {
        count.incrementAndGet()
        // 如果之前处于空闲状态，现在需要通知测试开始等待
        if (count.get() == 1) {
            callback?.onTransitionToBusy()
        }
    }

    // UI层操作完成后调用，减少计数
    fun decrementCount() {
        val newCount = count.decrementAndGet()
        if (newCount == 0) {
            // 计数归0，通知测试当前已空闲
            callback?.onTransitionToIdle()
        }
    }

    override fun getName(): String = name

    override fun isIdleNow(): Boolean = count.get() == 0

    override fun registerIdleTransitionCallback(callback: IdlingResource.ResourceCallback?) {
        this.callback = callback
    }
}

在UI层，当触发需要Espresso等待的操作时，直接调用incrementCount()：

// UI中启动异步任务时，通知IdlingResource开始计数
customIdlingResource.incrementCount()
// 执行异步操作...
// 操作完成后调用decrementCount()
customIdlingResource.decrementCount()

测试代码中，你只需要提前注册这个IdlingResource：

val idlingResource = CustomIdlingResource("MyCustomIdlingResource")
@Before
fun setup() {
    IdlingRegistry.getInstance().register(idlingResource)
}

@After
fun teardown() {
    IdlingRegistry.getInstance().unregister(idlingResource)
}

2.2 创建回调，在IdlingResource空闲时触发并向测试发送数据 #

同样可以实现。你可以在自定义IdlingResource中添加自定义回调接口，当状态从忙碌切换为空闲时，触发回调并传递数据：

class CustomIdlingResource(private val name: String) : IdlingResource {
    private val count = AtomicInteger(0)
    private var callback: IdlingResource.ResourceCallback? = null
    // 自定义空闲回调接口
    var onIdleCallback: ((data: Any?) -> Unit)? = null
    // 存储需要传递给测试的数据
    private var idleData: Any? = null

    // UI层可以调用这个方法，设置空闲时要传递的数据
    fun setIdleData(data: Any?) {
        this.idleData = data
    }

    fun incrementCount() {
        count.incrementAndGet()
        if (count.get() == 1) {
            callback?.onTransitionToBusy()
        }
    }

    fun decrementCount() {
        val newCount = count.decrementAndGet()
        if (newCount == 0) {
            callback?.onTransitionToIdle()
            // 触发自定义回调，传递数据给测试
            onIdleCallback?.invoke(idleData)
        }
    }

    // 其余方法和之前一致...
}

然后在测试代码中设置这个回调，接收数据：

@Before
fun setup() {
    val idlingResource = CustomIdlingResource("MyCustomIdlingResource")
    IdlingRegistry.getInstance().register(idlingResource)
    
    // 设置空闲回调，接收UI传递的数据
    idlingResource.onIdleCallback = { data ->
        if (data != null) {
            // 处理UI传递过来的数据，比如验证操作结果
            println("Received data from UI when idle: $data")
        }
    }
}

在UI层，你可以在操作完成前设置要传递的数据：

// 异步操作完成前，设置要传递给测试的数据
customIdlingResource.setIdleData("Operation completed successfully with id: 123")
customIdlingResource.decrementCount()

这样就能实现当IdlingResource切换到空闲状态时，自动向测试传递数据的需求了。

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