ActivityResultMatchers.java

/*
 * Copyright (C) 2016 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package androidx.test.espresso.contrib;

import android.app.Instrumentation.ActivityResult;
import android.content.Intent;
import org.hamcrest.Description;
import org.hamcrest.Matcher;
import org.hamcrest.TypeSafeMatcher;

/** Hamcrest matchers for a {@link ActivityResult}. */
public final class ActivityResultMatchers {

  private ActivityResultMatchers() {
    // no instance
  }

  /**
   * Returns a matcher that verifies that the resultData of a given {@link ActivityResult} matches
   * the given specification by the {@code intentMatcher}
   */
  public static Matcher<? super ActivityResult> hasResultData(final Matcher<Intent> intentMatcher) {
    return new TypeSafeMatcher<ActivityResult>(ActivityResult.class) {

      @Override
      public void describeTo(Description description) {
        description.appendDescriptionOf(intentMatcher);
      }

      @Override
      protected boolean matchesSafely(ActivityResult item) {
        return intentMatcher.matches(item.getResultData());
      }

      @Override
      protected void describeMismatchSafely(ActivityResult item, Description mismatchDescription) {
        intentMatcher.describeMismatch(item.getResultData(), mismatchDescription);
      }
    };
  }

  /**
   * Returns a matcher that verifies that the {@code resultCode} of a given {@link ActivityResult}
   * matches the given code
   */
  public static Matcher<? super ActivityResult> hasResultCode(final int resultCode) {
    return new TypeSafeMatcher<ActivityResult>(ActivityResult.class) {
      @Override
      public void describeTo(Description description) {
        description.appendText("has result code " + resultCode);
      }

      @Override
      protected boolean matchesSafely(ActivityResult activityResult) {
        return activityResult.getResultCode() == resultCode;
      }
    };
  }
}