Android Espresso -等待对话框显示

// 创建一个自定义的IdlingResource
public class DialogIdlingResource implements IdlingResource {
    private ResourceCallback resourceCallback;
    private boolean isIdle;

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

    @Override
    public boolean isIdleNow() {
        return isIdle;
    }

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

    // 在对话框显示完成后调用该方法，通知Espresso测试框架应用程序处于空闲状态
    public void setIdle(boolean isIdle) {
        this.isIdle = isIdle;
        if (isIdle && resourceCallback != null) {
            resourceCallback.onTransitionToIdle();
        }
    }
}

// 在测试代码中使用自定义的IdlingResource
@Test
public void testDialog() {
    // 创建对话框
    showDialog();

    // 创建IdlingResource并注册
    DialogIdlingResource idlingResource = new DialogIdlingResource();
    Espresso.registerIdlingResources(idlingResource);

    // 等待对话框显示完成
    idlingResource.setIdle(true);

    // 执行后续操作
    performAction();

    // 取消注册IdlingResource
    Espresso.unregisterIdlingResources(idlingResource);
}