1 /* 2 * Copyright (C) 2017 The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17 package androidx.contentpager.content; 18 19 import java.util.HashSet; 20 import java.util.Set; 21 22 /** 23 * Test friendly synchronous QueryRunner. Not suitable for use 24 * in production code. 25 */ 26 public final class TestQueryRunner implements ContentPager.QueryRunner { 27 28 // if false, we'll skip calling through to the mCallback when query is called 29 // this simulates async processing, and allows tests to check that cancel 30 // is handled correctly. 31 public boolean runQuery = true; 32 33 private final Set<Query> mRunning = new HashSet<>(); 34 35 @Override 36 public void query(Query query, Callback callback) { 37 if (runQuery) { 38 callback.onQueryFinished(query, callback.runQueryInBackground(query)); 39 } else { 40 mRunning.add(query); 41 } 42 } 43 44 @Override 45 public boolean isRunning(Query query) { 46 return mRunning.contains(query); 47 } 48 49 @Override 50 public void cancel(Query query) { 51 mRunning.remove(query); 52 } 53 } 54