Espresso：自定义Idling Resource

Espresso的一个关键功能是测试程序和被测应用是同步执行的。这是基于idling的概念：Espresso等待app处于idle状态，才会执行下个动作和检查下个断言。

Idle

app处于idle状态是什么意思？Espresso检查下面几个场景：

• 在当前消息队列中没有UI事件；
• 在默认的AsyncTask线程池没有任务；

但是，如果app以其他方式执行长时间运行操作，Espresso不知道如何判断这些操作已经完成。如果是这样的话，可以通过编写自定义的IdelingResource来通知Espresso的等待时间。

IntentServiceIdlingResource

假设你使用IntentService来做一些长时间运算，然后通过broadcast将结果返回给activity。我们希望Espresso一直等到结果返回，才来验证界面显示正确。

为了实现IdlingResource，需要重写3个函数：getName()，registerIdleTransitionCallback()，isIdleNow()。

@Override
public String getName() {
  return IntentServiceIdlingResource.class.getName();
}

@Override
public void registerIdleTransitionCallback(ResourceCallback resourceCallback) {
    this.resourceCallback = resourceCallback;
}

@Override
public boolean isIdleNow() {
    boolean idle = !isIntentServiceRunning();
    if (idle && resourceCallback != null) {
        resourceCallback.onTransitionToIdle();
    }
    return idle;
}

private boolean isIntentServiceRunning() {
    ActivityManager manager = (ActivityManager) context.getSystemService(Context.ACTIVITY_SERVICE);
    for (ActivityManager.RunningServiceInfo info : manager.getRunningServices(Integer.MAX_VALUE)) {
        if (RepeatService.class.getName().equals(info.service.getClassName())) {
            return true;
        }
    }
    return false;
}

idle逻辑是在isIdleNow()实现的。在这个例子中，我们通过查询ActivityManager来检查IntentService是否正在运行。如果IntentService停止运行，我们调用resourceCallback.onTransitionToIdle()来通知Espresso。

注册idling resource

为了让Espresso等待自定义的idling resource，你需要注册它。在测试代码的@Before方法中执行注册，在@After中执行注销。

@Before
public void registerIntentServiceIdlingResource() {
    Instrumentation instrumentation = InstrumentationRegistry.getInstrumentation();
    idlingResource = new IntentServiceIdlingResource(instrumentation.getTargetContext());
    Espresso.registerIdlingResources(idlingResource);
}

@After
public void unregisterIntentServiceIdlingResource() {
    Espresso.unregisterIdlingResources(idlingResource);
}

完整示例

Check out完整示例的源码。尝试注释掉IdlingResource注册，测试会失败。