ResettingStubberImpl.java
/*
* Copyright (C) 2015 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.intent;
import static androidx.test.espresso.intent.Checks.checkNotNull;
import static androidx.test.espresso.intent.Checks.checkState;
import android.app.Instrumentation.ActivityResult;
import android.content.Intent;
import android.content.pm.PackageManager;
import android.content.pm.ResolveInfo;
import android.os.Looper;
import android.util.Pair;
import androidx.test.platform.app.InstrumentationRegistry;
import java.util.ArrayList;
import java.util.List;
import java.util.ListIterator;
import org.hamcrest.Matcher;
/** Implementation of {@link ResettingStubber} */
public final class ResettingStubberImpl implements ResettingStubber {
// Should be accessed only from main thread
private List<Pair<Matcher<Intent>, ActivityResultFunction>> intentResponsePairs =
new ArrayList<Pair<Matcher<Intent>, ActivityResultFunction>>();
private PackageManager packageManager;
private boolean isInitialized;
@Override
public void initialize() {
packageManager =
InstrumentationRegistry.getInstrumentation().getTargetContext().getPackageManager();
isInitialized = true;
}
@Override
public boolean isInitialized() {
return isInitialized;
}
@Override
public void reset() {
checkMain();
intentResponsePairs.clear();
isInitialized = false;
}
@Override
public void setActivityResultForIntent(Matcher<Intent> matcher, ActivityResult result) {
setActivityResultFunctionForIntent(matcher, intent -> result);
}
@Override
public void setActivityResultFunctionForIntent(
Matcher<Intent> matcher, ActivityResultFunction result) {
checkState(isInitialized, "ResettingStubber must be initialized before calling this method");
checkNotNull(matcher);
checkMain();
intentResponsePairs.add(new Pair<Matcher<Intent>, ActivityResultFunction>(matcher, result));
}
@Override
public ActivityResult getActivityResultForIntent(Intent intent) {
checkState(isInitialized, "ResettingStubber must be initialized before calling this method");
checkNotNull(intent);
checkMain();
ListIterator<Pair<Matcher<Intent>, ActivityResultFunction>> reverseIterator =
intentResponsePairs.listIterator(intentResponsePairs.size());
while (reverseIterator.hasPrevious()) {
Pair<Matcher<Intent>, ActivityResultFunction> pair = reverseIterator.previous();
// We resolve the intent so that the toPackage matcher has the necessary information to match
// the intent.
if (pair.first.matches(resolveIntent(intent))) {
return pair.second.apply(intent);
}
}
return null;
}
// package private, so that Intents can use this to resolve intents as it records them.
ResolvedIntent resolveIntent(Intent intent) {
// Android API is setting flags param to zero per discussion here:
// http://stackoverflow.com/questions/9623079/
// why-does-the-flag-specified-in-queryintentactivities-method-is-set-to-zero and
// http://developer.android.com/training/basics/intents/sending.html
List<ResolveInfo> resolveInfos = packageManager.queryIntentActivities(intent, 0);
if (null == resolveInfos) {
// Gingerbread returns null here if nothing resolves, other APIs return an empty list.
resolveInfos = new ArrayList<ResolveInfo>();
}
return new ResolvedIntentImpl(intent, resolveInfos);
}
private static void checkMain() {
checkState(Looper.myLooper() == Looper.getMainLooper(), "Must be called on main thread.");
}
}