55bb0c258d
Bug: 62072727 Test: RunSettingsRoboTests Change-Id: I2140c41d01f18a1e4b9e3f505e0afa4a8757d049
587 lines
27 KiB
Java
587 lines
27 KiB
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 com.android.settings.search;
|
|
|
|
import android.app.Activity;
|
|
import android.content.Context;
|
|
import android.content.Intent;
|
|
import android.graphics.drawable.Drawable;
|
|
import android.util.Pair;
|
|
import android.view.ViewGroup;
|
|
import android.widget.FrameLayout;
|
|
|
|
import com.android.settings.R;
|
|
import com.android.settings.testutils.SettingsRobolectricTestRunner;
|
|
import com.android.settings.TestConfig;
|
|
import com.android.settings.search.SearchResult.Builder;
|
|
import com.android.settings.search.ranking.SearchResultsRankerCallback;
|
|
|
|
import org.junit.Before;
|
|
import org.junit.Test;
|
|
import org.junit.runner.RunWith;
|
|
import org.mockito.ArgumentCaptor;
|
|
import org.mockito.Captor;
|
|
import org.mockito.Mock;
|
|
import org.mockito.MockitoAnnotations;
|
|
import org.robolectric.Robolectric;
|
|
import org.robolectric.annotation.Config;
|
|
import org.robolectric.shadows.ShadowLooper;
|
|
|
|
import java.util.ArrayList;
|
|
import java.util.Arrays;
|
|
import java.util.HashSet;
|
|
import java.util.List;
|
|
import java.util.Objects;
|
|
import java.util.Set;
|
|
|
|
import static com.google.common.truth.Truth.assertThat;
|
|
import static org.mockito.Matchers.any;
|
|
import static org.mockito.Matchers.anyString;
|
|
import static org.mockito.Mockito.never;
|
|
import static org.mockito.Mockito.times;
|
|
import static org.mockito.Mockito.verify;
|
|
import static org.mockito.Mockito.when;
|
|
|
|
@RunWith(SettingsRobolectricTestRunner.class)
|
|
@Config(manifest = TestConfig.MANIFEST_PATH, sdk = TestConfig.SDK_VERSION)
|
|
public class SearchResultsAdapterTest {
|
|
|
|
@Mock
|
|
private SearchFragment mFragment;
|
|
@Mock
|
|
private SearchFeatureProvider mSearchFeatureProvider;
|
|
@Mock
|
|
private Context mMockContext;
|
|
@Captor
|
|
private ArgumentCaptor<Integer> mSearchResultsCountCaptor =
|
|
ArgumentCaptor.forClass(Integer.class);
|
|
private SearchResultsAdapter mAdapter;
|
|
private Context mContext;
|
|
private String mLoaderClassName;
|
|
|
|
private String[] TITLES = {"alpha", "bravo", "charlie", "appAlpha", "appBravo", "appCharlie"};
|
|
|
|
@Before
|
|
public void setUp() {
|
|
MockitoAnnotations.initMocks(this);
|
|
mContext = Robolectric.buildActivity(Activity.class).get();
|
|
mLoaderClassName = DatabaseResultLoader.class.getName();
|
|
when(mFragment.getContext()).thenReturn(mMockContext);
|
|
when(mMockContext.getApplicationContext()).thenReturn(mContext);
|
|
mAdapter = new SearchResultsAdapter(mFragment, mSearchFeatureProvider);
|
|
}
|
|
|
|
@Test
|
|
public void testNoResultsAdded_emptyListReturned() {
|
|
List<SearchResult> updatedResults = mAdapter.getSearchResults();
|
|
assertThat(updatedResults).isEmpty();
|
|
}
|
|
|
|
@Test
|
|
public void testSingleSourceMerge_exactCopyReturned() {
|
|
Set<SearchResult> intentResults = getIntentSampleResults();
|
|
mAdapter.initializeSearch("");
|
|
mAdapter.addSearchResults(intentResults, mLoaderClassName);
|
|
mAdapter.notifyResultsLoaded();
|
|
|
|
List<SearchResult> updatedResults = mAdapter.getSearchResults();
|
|
assertThat(updatedResults).containsAllIn(intentResults);
|
|
}
|
|
|
|
@Test
|
|
public void testCreateViewHolder_returnsIntentResult() {
|
|
ViewGroup group = new FrameLayout(mContext);
|
|
SearchViewHolder view = mAdapter.onCreateViewHolder(group,
|
|
ResultPayload.PayloadType.INTENT);
|
|
assertThat(view).isInstanceOf(IntentSearchViewHolder.class);
|
|
}
|
|
|
|
@Test
|
|
public void testCreateViewHolder_returnsIntentSwitchResult() {
|
|
// TODO (b/62807132) test for InlineResult
|
|
ViewGroup group = new FrameLayout(mContext);
|
|
SearchViewHolder view = mAdapter.onCreateViewHolder(group,
|
|
ResultPayload.PayloadType.INLINE_SWITCH);
|
|
assertThat(view).isInstanceOf(IntentSearchViewHolder.class);
|
|
}
|
|
|
|
@Test
|
|
public void testEndToEndSearch_properResultsMerged_correctOrder() {
|
|
mAdapter.initializeSearch("");
|
|
mAdapter.addSearchResults(new HashSet<>(getDummyAppResults()),
|
|
InstalledAppResultLoader.class.getName());
|
|
mAdapter.addSearchResults(new HashSet<>(getDummyDbResults()),
|
|
DatabaseResultLoader.class.getName());
|
|
mAdapter.notifyResultsLoaded();
|
|
|
|
List<SearchResult> results = mAdapter.getSearchResults();
|
|
assertThat(results.get(0).title).isEqualTo(TITLES[0]); // alpha
|
|
assertThat(results.get(1).title).isEqualTo(TITLES[3]); // appAlpha
|
|
assertThat(results.get(2).title).isEqualTo(TITLES[4]); // appBravo
|
|
assertThat(results.get(3).title).isEqualTo(TITLES[1]); // bravo
|
|
assertThat(results.get(4).title).isEqualTo(TITLES[5]); // appCharlie
|
|
assertThat(results.get(5).title).isEqualTo(TITLES[2]); // charlie
|
|
verify(mFragment).onSearchResultsDisplayed(mSearchResultsCountCaptor.capture());
|
|
assertThat(mSearchResultsCountCaptor.getValue()).isEqualTo(6);
|
|
}
|
|
|
|
@Test
|
|
public void testEndToEndSearch_addResults_resultsAddedInOrder() {
|
|
List<SearchResult> appResults = getDummyAppResults();
|
|
List<SearchResult> dbResults = getDummyDbResults();
|
|
mAdapter.initializeSearch("");
|
|
// Add two individual items
|
|
mAdapter.addSearchResults(new HashSet<>(appResults.subList(0, 1)),
|
|
InstalledAppResultLoader.class.getName());
|
|
mAdapter.addSearchResults(new HashSet<>(dbResults.subList(0, 1)),
|
|
DatabaseResultLoader.class.getName());
|
|
mAdapter.notifyResultsLoaded();
|
|
// Add super-set of items
|
|
mAdapter.initializeSearch("");
|
|
mAdapter.addSearchResults(
|
|
new HashSet<>(appResults), InstalledAppResultLoader.class.getName());
|
|
mAdapter.addSearchResults(
|
|
new HashSet<>(dbResults), DatabaseResultLoader.class.getName());
|
|
mAdapter.notifyResultsLoaded();
|
|
|
|
List<SearchResult> results = mAdapter.getSearchResults();
|
|
assertThat(results.get(0).title).isEqualTo(TITLES[0]); // alpha
|
|
assertThat(results.get(1).title).isEqualTo(TITLES[3]); // appAlpha
|
|
assertThat(results.get(2).title).isEqualTo(TITLES[4]); // appBravo
|
|
assertThat(results.get(3).title).isEqualTo(TITLES[1]); // bravo
|
|
assertThat(results.get(4).title).isEqualTo(TITLES[5]); // appCharlie
|
|
assertThat(results.get(5).title).isEqualTo(TITLES[2]); // charlie
|
|
verify(mFragment, times(2)).onSearchResultsDisplayed(mSearchResultsCountCaptor.capture());
|
|
assertThat(mSearchResultsCountCaptor.getAllValues().toArray())
|
|
.isEqualTo(new Integer[] {2, 6});
|
|
}
|
|
|
|
@Test
|
|
public void testEndToEndSearch_removeResults_resultsAdded() {
|
|
List<SearchResult> appResults = getDummyAppResults();
|
|
List<SearchResult> dbResults = getDummyDbResults();
|
|
// Add list of items
|
|
mAdapter.initializeSearch("");
|
|
mAdapter.addSearchResults(new HashSet<>(appResults),
|
|
InstalledAppResultLoader.class.getName());
|
|
mAdapter.addSearchResults(new HashSet<>(dbResults),
|
|
DatabaseResultLoader.class.getName());
|
|
mAdapter.notifyResultsLoaded();
|
|
// Add subset of items
|
|
mAdapter.initializeSearch("");
|
|
mAdapter.addSearchResults(new HashSet<>(appResults.subList(0, 1)),
|
|
InstalledAppResultLoader.class.getName());
|
|
mAdapter.addSearchResults(new HashSet<>(dbResults.subList(0, 1)),
|
|
DatabaseResultLoader.class.getName());
|
|
mAdapter.notifyResultsLoaded();
|
|
|
|
List<SearchResult> results = mAdapter.getSearchResults();
|
|
assertThat(results.get(0).title).isEqualTo(TITLES[0]);
|
|
assertThat(results.get(1).title).isEqualTo(TITLES[3]);
|
|
verify(mFragment, times(2)).onSearchResultsDisplayed(mSearchResultsCountCaptor.capture());
|
|
assertThat(mSearchResultsCountCaptor.getAllValues().toArray())
|
|
.isEqualTo(new Integer[] {6, 2});
|
|
}
|
|
@Test
|
|
public void testEndToEndSearch_smartSearchRankingEnabledAndSucceededAfterResultsLoaded() {
|
|
when(mSearchFeatureProvider.isSmartSearchRankingEnabled(any())).thenReturn(true);
|
|
|
|
List<SearchResult> appResults = getDummyAppResults();
|
|
List<SearchResult> dbResults = getDummyDbResults();
|
|
mAdapter.initializeSearch("");
|
|
mAdapter.addSearchResults(
|
|
new HashSet<>(appResults), InstalledAppResultLoader.class.getName());
|
|
mAdapter.addSearchResults(
|
|
new HashSet<>(dbResults), DatabaseResultLoader.class.getName());
|
|
mAdapter.notifyResultsLoaded();
|
|
mAdapter.onRankingScoresAvailable(getDummyRankingScores());
|
|
|
|
List<SearchResult> results = mAdapter.getSearchResults();
|
|
assertThat(results.get(0).title).isEqualTo(TITLES[2]); // charlie
|
|
assertThat(results.get(1).title).isEqualTo(TITLES[0]); // alpha
|
|
assertThat(results.get(2).title).isEqualTo(TITLES[1]); // bravo
|
|
assertThat(results.get(3).title).isEqualTo(TITLES[3]); // appAlpha
|
|
assertThat(results.get(4).title).isEqualTo(TITLES[4]); // appBravo
|
|
assertThat(results.get(5).title).isEqualTo(TITLES[5]); // appCharlie
|
|
verify(mFragment).onSearchResultsDisplayed(mSearchResultsCountCaptor.capture());
|
|
assertThat(mSearchResultsCountCaptor.getValue()).isEqualTo(6);
|
|
assertThat(mAdapter.getAsyncRankingState()).isEqualTo(SearchResultsAdapter.SUCCEEDED);
|
|
}
|
|
|
|
@Test
|
|
public void testEndToEndSearch_smartSearchRankingEnabledAndSucceededBeforeResultsLoaded() {
|
|
when(mSearchFeatureProvider.isSmartSearchRankingEnabled(any())).thenReturn(true);
|
|
|
|
List<SearchResult> appResults = getDummyAppResults();
|
|
List<SearchResult> dbResults = getDummyDbResults();
|
|
mAdapter.initializeSearch("");
|
|
mAdapter.onRankingScoresAvailable(getDummyRankingScores());
|
|
mAdapter.addSearchResults(
|
|
new HashSet<>(appResults), InstalledAppResultLoader.class.getName());
|
|
mAdapter.addSearchResults(
|
|
new HashSet<>(dbResults), DatabaseResultLoader.class.getName());
|
|
mAdapter.notifyResultsLoaded();
|
|
|
|
List<SearchResult> results = mAdapter.getSearchResults();
|
|
assertThat(results.get(0).title).isEqualTo(TITLES[2]); // charlie
|
|
assertThat(results.get(1).title).isEqualTo(TITLES[0]); // alpha
|
|
assertThat(results.get(2).title).isEqualTo(TITLES[1]); // bravo
|
|
assertThat(results.get(3).title).isEqualTo(TITLES[3]); // appAlpha
|
|
assertThat(results.get(4).title).isEqualTo(TITLES[4]); // appBravo
|
|
assertThat(results.get(5).title).isEqualTo(TITLES[5]); // appCharlie
|
|
verify(mFragment).onSearchResultsDisplayed(mSearchResultsCountCaptor.capture());
|
|
assertThat(mSearchResultsCountCaptor.getValue()).isEqualTo(6);
|
|
assertThat(mAdapter.getAsyncRankingState()).isEqualTo(SearchResultsAdapter.SUCCEEDED);
|
|
}
|
|
|
|
@Test
|
|
public void testEndToEndSearch_smartSearchRankingEnabledAndFailedAfterResultsLoaded() {
|
|
when(mSearchFeatureProvider.isSmartSearchRankingEnabled(any())).thenReturn(true);
|
|
|
|
List<SearchResult> appResults = getDummyAppResults();
|
|
List<SearchResult> dbResults = getDummyDbResults();
|
|
mAdapter.initializeSearch("");
|
|
mAdapter.addSearchResults(
|
|
new HashSet<>(appResults), InstalledAppResultLoader.class.getName());
|
|
mAdapter.addSearchResults(
|
|
new HashSet<>(dbResults), DatabaseResultLoader.class.getName());
|
|
mAdapter.notifyResultsLoaded();
|
|
mAdapter.onRankingFailed();
|
|
|
|
List<SearchResult> results = mAdapter.getSearchResults();
|
|
assertThat(results.get(0).title).isEqualTo(TITLES[0]); // alpha
|
|
assertThat(results.get(1).title).isEqualTo(TITLES[3]); // appAlpha
|
|
assertThat(results.get(2).title).isEqualTo(TITLES[4]); // appBravo
|
|
assertThat(results.get(3).title).isEqualTo(TITLES[1]); // bravo
|
|
assertThat(results.get(4).title).isEqualTo(TITLES[5]); // appCharlie
|
|
assertThat(results.get(5).title).isEqualTo(TITLES[2]); // charlie
|
|
verify(mFragment).onSearchResultsDisplayed(mSearchResultsCountCaptor.capture());
|
|
assertThat(mSearchResultsCountCaptor.getValue()).isEqualTo(6);
|
|
assertThat(mAdapter.getAsyncRankingState()).isEqualTo(SearchResultsAdapter.FAILED);
|
|
}
|
|
|
|
@Test
|
|
public void testEndToEndSearch_smartSearchRankingEnabledAndFailedBeforeResultsLoaded() {
|
|
when(mSearchFeatureProvider.isSmartSearchRankingEnabled(any())).thenReturn(true);
|
|
|
|
List<SearchResult> appResults = getDummyAppResults();
|
|
List<SearchResult> dbResults = getDummyDbResults();
|
|
mAdapter.initializeSearch("");
|
|
mAdapter.onRankingFailed();
|
|
mAdapter.addSearchResults(
|
|
new HashSet<>(appResults), InstalledAppResultLoader.class.getName());
|
|
mAdapter.addSearchResults(
|
|
new HashSet<>(dbResults), DatabaseResultLoader.class.getName());
|
|
mAdapter.notifyResultsLoaded();
|
|
|
|
List<SearchResult> results = mAdapter.getSearchResults();
|
|
assertThat(results.get(0).title).isEqualTo(TITLES[0]); // alpha
|
|
assertThat(results.get(1).title).isEqualTo(TITLES[3]); // appAlpha
|
|
assertThat(results.get(2).title).isEqualTo(TITLES[4]); // appBravo
|
|
assertThat(results.get(3).title).isEqualTo(TITLES[1]); // bravo
|
|
assertThat(results.get(4).title).isEqualTo(TITLES[5]); // appCharlie
|
|
assertThat(results.get(5).title).isEqualTo(TITLES[2]); // charlie
|
|
verify(mFragment).onSearchResultsDisplayed(mSearchResultsCountCaptor.capture());
|
|
assertThat(mSearchResultsCountCaptor.getValue()).isEqualTo(6);
|
|
assertThat(mAdapter.getAsyncRankingState()).isEqualTo(SearchResultsAdapter.FAILED);
|
|
}
|
|
|
|
@Test
|
|
public void testEndToEndSearch_smartSearchRankingEnabledAndTimedoutAfterResultsLoaded() {
|
|
when(mSearchFeatureProvider.isSmartSearchRankingEnabled(any())).thenReturn(true);
|
|
|
|
List<SearchResult> appResults = getDummyAppResults();
|
|
List<SearchResult> dbResults = getDummyDbResults();
|
|
mAdapter.initializeSearch("");
|
|
mAdapter.addSearchResults(
|
|
new HashSet<>(appResults), InstalledAppResultLoader.class.getName());
|
|
mAdapter.addSearchResults(
|
|
new HashSet<>(dbResults), DatabaseResultLoader.class.getName());
|
|
mAdapter.notifyResultsLoaded();
|
|
|
|
waitUntilRankingTimesOut();
|
|
|
|
List<SearchResult> results = mAdapter.getSearchResults();
|
|
assertThat(results.get(0).title).isEqualTo(TITLES[0]); // alpha
|
|
assertThat(results.get(1).title).isEqualTo(TITLES[3]); // appAlpha
|
|
assertThat(results.get(2).title).isEqualTo(TITLES[4]); // appBravo
|
|
assertThat(results.get(3).title).isEqualTo(TITLES[1]); // bravo
|
|
assertThat(results.get(4).title).isEqualTo(TITLES[5]); // appCharlie
|
|
assertThat(results.get(5).title).isEqualTo(TITLES[2]); // charlie
|
|
verify(mFragment).onSearchResultsDisplayed(mSearchResultsCountCaptor.capture());
|
|
assertThat(mSearchResultsCountCaptor.getValue()).isEqualTo(6);
|
|
assertThat(mAdapter.getAsyncRankingState()).isEqualTo(SearchResultsAdapter.TIMED_OUT);
|
|
}
|
|
|
|
@Test
|
|
public void testEndToEndSearch_smartSearchRankingEnabledAndTimedoutBeforeResultsLoaded() {
|
|
when(mSearchFeatureProvider.isSmartSearchRankingEnabled(any())).thenReturn(true);
|
|
|
|
List<SearchResult> appResults = getDummyAppResults();
|
|
List<SearchResult> dbResults = getDummyDbResults();
|
|
mAdapter.initializeSearch("");
|
|
|
|
waitUntilRankingTimesOut();
|
|
|
|
mAdapter.addSearchResults(
|
|
new HashSet<>(appResults), InstalledAppResultLoader.class.getName());
|
|
mAdapter.addSearchResults(
|
|
new HashSet<>(dbResults), DatabaseResultLoader.class.getName());
|
|
mAdapter.notifyResultsLoaded();
|
|
|
|
List<SearchResult> results = mAdapter.getSearchResults();
|
|
assertThat(results.get(0).title).isEqualTo(TITLES[0]); // alpha
|
|
assertThat(results.get(1).title).isEqualTo(TITLES[3]); // appAlpha
|
|
assertThat(results.get(2).title).isEqualTo(TITLES[4]); // appBravo
|
|
assertThat(results.get(3).title).isEqualTo(TITLES[1]); // bravo
|
|
assertThat(results.get(4).title).isEqualTo(TITLES[5]); // appCharlie
|
|
assertThat(results.get(5).title).isEqualTo(TITLES[2]); // charlie
|
|
verify(mFragment).onSearchResultsDisplayed(mSearchResultsCountCaptor.capture());
|
|
assertThat(mSearchResultsCountCaptor.getValue()).isEqualTo(6);
|
|
assertThat(mAdapter.getAsyncRankingState()).isEqualTo(SearchResultsAdapter.TIMED_OUT);
|
|
}
|
|
|
|
@Test
|
|
public void testDoSmartRanking_shouldRankAppResultsAfterDbResults() {
|
|
when(mSearchFeatureProvider.isSmartSearchRankingEnabled(any())).thenReturn(true);
|
|
|
|
List<SearchResult> appResults = getDummyAppResults();
|
|
List<SearchResult> dbResults = getDummyDbResults();
|
|
mAdapter.initializeSearch("");
|
|
mAdapter.addSearchResults(
|
|
new HashSet<>(appResults), InstalledAppResultLoader.class.getName());
|
|
mAdapter.addSearchResults(
|
|
new HashSet<>(dbResults), DatabaseResultLoader.class.getName());
|
|
mAdapter.notifyResultsLoaded();
|
|
mAdapter.onRankingScoresAvailable(getDummyRankingScores());
|
|
List<SearchResult> results = mAdapter.doAsyncRanking();
|
|
assertThat(results.get(0).title).isEqualTo(TITLES[2]); // charlie
|
|
assertThat(results.get(1).title).isEqualTo(TITLES[0]); // alpha
|
|
assertThat(results.get(2).title).isEqualTo(TITLES[1]); // bravo
|
|
assertThat(results.get(3).title).isEqualTo(TITLES[3]); // appAlpha
|
|
assertThat(results.get(4).title).isEqualTo(TITLES[4]); // appBravo
|
|
assertThat(results.get(5).title).isEqualTo(TITLES[5]); // appCharlie
|
|
}
|
|
|
|
@Test
|
|
public void testDoSmartRanking_shouldRankResultsWithMissingScoresAfterScoredResults() {
|
|
when(mSearchFeatureProvider.isSmartSearchRankingEnabled(any())).thenReturn(true);
|
|
|
|
List<SearchResult> appResults = getDummyAppResults();
|
|
List<SearchResult> dbResults = getDummyDbResults();
|
|
mAdapter.initializeSearch("");
|
|
mAdapter.addSearchResults(
|
|
new HashSet<>(appResults), InstalledAppResultLoader.class.getName());
|
|
mAdapter.addSearchResults(
|
|
new HashSet<>(dbResults), DatabaseResultLoader.class.getName());
|
|
mAdapter.notifyResultsLoaded();
|
|
List<Pair<String, Float>> rankingScores = getDummyRankingScores();
|
|
rankingScores.remove(1); // no ranking score for alpha
|
|
mAdapter.onRankingScoresAvailable(rankingScores);
|
|
List<SearchResult> results = mAdapter.doAsyncRanking();
|
|
assertThat(results.get(0).title).isEqualTo(TITLES[2]); // charlie
|
|
assertThat(results.get(1).title).isEqualTo(TITLES[1]); // bravo
|
|
assertThat(results.get(2).title).isEqualTo(TITLES[0]); // alpha
|
|
assertThat(results.get(3).title).isEqualTo(TITLES[3]); // appAlpha
|
|
assertThat(results.get(4).title).isEqualTo(TITLES[4]); // appBravo
|
|
assertThat(results.get(5).title).isEqualTo(TITLES[5]); // appCharlie
|
|
}
|
|
|
|
@Test
|
|
public void testGetUnsortedLoadedResults () {
|
|
List<SearchResult> appResults = getDummyAppResults();
|
|
List<SearchResult> dbResults = getDummyDbResults();
|
|
mAdapter.initializeSearch("");
|
|
mAdapter.addSearchResults(
|
|
new HashSet<>(appResults), InstalledAppResultLoader.class.getName());
|
|
mAdapter.addSearchResults(
|
|
new HashSet<>(dbResults), DatabaseResultLoader.class.getName());
|
|
Set<CharSequence> expectedDbTitles = new HashSet<>(
|
|
Arrays.asList("alpha", "bravo", "charlie"));
|
|
Set<CharSequence> expectedAppTitles = new HashSet<>(
|
|
Arrays.asList("appAlpha", "appBravo", "appCharlie"));
|
|
Set<CharSequence> actualDbTitles = new HashSet<>();
|
|
Set<CharSequence> actualAppTitles = new HashSet<>();
|
|
for (SearchResult result : mAdapter.getUnsortedLoadedResults(SearchResultsAdapter
|
|
.DB_RESULTS_LOADER_KEY)) {
|
|
actualDbTitles.add(result.title);
|
|
}
|
|
for (SearchResult result : mAdapter.getUnsortedLoadedResults(SearchResultsAdapter
|
|
.APP_RESULTS_LOADER_KEY)) {
|
|
actualAppTitles.add(result.title);
|
|
}
|
|
assertThat(actualDbTitles).isEqualTo(expectedDbTitles);
|
|
assertThat(actualAppTitles).isEqualTo(expectedAppTitles);
|
|
}
|
|
|
|
@Test
|
|
public void testGetSortedLoadedResults() {
|
|
List<SearchResult> appResults = getDummyAppResults();
|
|
List<SearchResult> dbResults = getDummyDbResults();
|
|
mAdapter.initializeSearch("");
|
|
mAdapter.addSearchResults(
|
|
new HashSet<>(appResults), InstalledAppResultLoader.class.getName());
|
|
mAdapter.addSearchResults(
|
|
new HashSet<>(dbResults), DatabaseResultLoader.class.getName());
|
|
List<? extends SearchResult> actualDbResults =
|
|
mAdapter.getSortedLoadedResults(SearchResultsAdapter.DB_RESULTS_LOADER_KEY);
|
|
List<? extends SearchResult> actualAppResults =
|
|
mAdapter.getSortedLoadedResults(SearchResultsAdapter.APP_RESULTS_LOADER_KEY);
|
|
assertThat(actualDbResults.get(0).title).isEqualTo(TITLES[0]); // charlie
|
|
assertThat(actualDbResults.get(1).title).isEqualTo(TITLES[1]); // bravo
|
|
assertThat(actualDbResults.get(2).title).isEqualTo(TITLES[2]); // alpha
|
|
assertThat(actualAppResults.get(0).title).isEqualTo(TITLES[3]); // appAlpha
|
|
assertThat(actualAppResults.get(1).title).isEqualTo(TITLES[4]); // appBravo
|
|
assertThat(actualAppResults.get(2).title).isEqualTo(TITLES[5]); // appCharlie
|
|
}
|
|
|
|
@Test
|
|
public void testInitializeSearch_shouldNotRunSmartRankingIfDisabled() {
|
|
when(mSearchFeatureProvider.isSmartSearchRankingEnabled(any())).thenReturn(false);
|
|
mAdapter.initializeSearch("");
|
|
mAdapter.notifyResultsLoaded();
|
|
verify(mSearchFeatureProvider, never()).querySearchResults(
|
|
any(Context.class), anyString(), any(SearchResultsRankerCallback.class));
|
|
assertThat(mAdapter.getAsyncRankingState()).isEqualTo(SearchResultsAdapter.DISABLED);
|
|
}
|
|
|
|
@Test
|
|
public void testInitialSearch_shouldRunSmartRankingIfEnabled() {
|
|
when(mSearchFeatureProvider.isSmartSearchRankingEnabled(any())).thenReturn(true);
|
|
mAdapter.initializeSearch("");
|
|
mAdapter.notifyResultsLoaded();
|
|
verify(mSearchFeatureProvider, times(1)).querySearchResults(
|
|
any(Context.class), anyString(), any(SearchResultsRankerCallback.class));
|
|
assertThat(mAdapter.getAsyncRankingState())
|
|
.isEqualTo(SearchResultsAdapter.PENDING_RESULTS);
|
|
}
|
|
|
|
@Test
|
|
public void testGetRankingScoreByStableId() {
|
|
when(mSearchFeatureProvider.isSmartSearchRankingEnabled(any())).thenReturn(true);
|
|
|
|
List<SearchResult> appResults = getDummyAppResults();
|
|
List<SearchResult> dbResults = getDummyDbResults();
|
|
mAdapter.initializeSearch("");
|
|
mAdapter.onRankingScoresAvailable(getDummyRankingScores());
|
|
assertThat(mAdapter.getRankingScoreByStableId(dbResults.get(0).stableId))
|
|
.isWithin(1e-10f).of(0.8f);
|
|
assertThat(mAdapter.getRankingScoreByStableId(dbResults.get(1).stableId))
|
|
.isWithin(1e-10f).of(0.2f);
|
|
assertThat(mAdapter.getRankingScoreByStableId(dbResults.get(2).stableId))
|
|
.isWithin(1e-10f).of(0.9f);
|
|
assertThat(mAdapter.getRankingScoreByStableId(appResults.get(0).stableId))
|
|
.isEqualTo(-Float.MAX_VALUE);
|
|
assertThat(mAdapter.getRankingScoreByStableId(appResults.get(1).stableId))
|
|
.isEqualTo(-Float.MAX_VALUE);
|
|
assertThat(mAdapter.getRankingScoreByStableId(appResults.get(2).stableId))
|
|
.isEqualTo(-Float.MAX_VALUE);
|
|
}
|
|
|
|
private void waitUntilRankingTimesOut() {
|
|
while (mAdapter.getHandler().hasMessages(mAdapter.MSG_RANKING_TIMED_OUT)) {
|
|
try {
|
|
ShadowLooper.runUiThreadTasksIncludingDelayedTasks();
|
|
Thread.sleep(100);
|
|
} catch (InterruptedException e) {
|
|
// Do nothing
|
|
}
|
|
}
|
|
}
|
|
|
|
private List<SearchResult> getDummyDbResults() {
|
|
List<SearchResult> results = new ArrayList<>();
|
|
ResultPayload payload = new ResultPayload(new Intent());
|
|
SearchResult.Builder builder = new SearchResult.Builder();
|
|
builder.setPayload(payload)
|
|
.setTitle(TITLES[0])
|
|
.setRank(1)
|
|
.setStableId(Objects.hash(TITLES[0], "db"));
|
|
results.add(builder.build());
|
|
|
|
builder.setTitle(TITLES[1])
|
|
.setRank(3)
|
|
.setStableId(Objects.hash(TITLES[1], "db"));
|
|
results.add(builder.build());
|
|
|
|
builder.setTitle(TITLES[2])
|
|
.setRank(6)
|
|
.setStableId(Objects.hash(TITLES[2], "db"));
|
|
results.add(builder.build());
|
|
|
|
return results;
|
|
}
|
|
|
|
private List<SearchResult> getDummyAppResults() {
|
|
List<SearchResult> results = new ArrayList<>();
|
|
ResultPayload payload = new ResultPayload(new Intent());
|
|
AppSearchResult.Builder builder = new AppSearchResult.Builder();
|
|
builder.setPayload(payload)
|
|
.setTitle(TITLES[3])
|
|
.setRank(1)
|
|
.setStableId(Objects.hash(TITLES[3], "app"));
|
|
results.add(builder.build());
|
|
|
|
builder.setTitle(TITLES[4])
|
|
.setRank(2)
|
|
.setStableId(Objects.hash(TITLES[4], "app"));
|
|
results.add(builder.build());
|
|
|
|
builder.setTitle(TITLES[5])
|
|
.setRank(4)
|
|
.setStableId(Objects.hash(TITLES[5], "app"));
|
|
results.add(builder.build());
|
|
|
|
return results;
|
|
}
|
|
|
|
private Set<SearchResult> getIntentSampleResults() {
|
|
Set<SearchResult> sampleResults = new HashSet<>();
|
|
ArrayList<String> breadcrumbs = new ArrayList<>();
|
|
final Drawable icon = mContext.getDrawable(R.drawable.ic_search_history);
|
|
final ResultPayload payload = new ResultPayload(null);
|
|
final SearchResult.Builder builder = new Builder();
|
|
builder.setTitle("title")
|
|
.setSummary("summary")
|
|
.setRank(1)
|
|
.addBreadcrumbs(breadcrumbs)
|
|
.setIcon(icon)
|
|
.setPayload(payload)
|
|
.setStableId(Objects.hash("title", "summary", 1));
|
|
sampleResults.add(builder.build());
|
|
|
|
builder.setRank(2)
|
|
.setStableId(Objects.hash("title", "summary", 2));
|
|
sampleResults.add(builder.build());
|
|
|
|
builder.setRank(3)
|
|
.setStableId(Objects.hash("title", "summary", 3));
|
|
sampleResults.add(builder.build());
|
|
return sampleResults;
|
|
}
|
|
|
|
private List<Pair<String, Float>> getDummyRankingScores() {
|
|
List<SearchResult> results = getDummyDbResults();
|
|
List<Pair<String, Float>> scores = new ArrayList<>();
|
|
scores.add(new Pair<>(Long.toString(results.get(2).stableId), 0.9f)); // charlie
|
|
scores.add(new Pair<>(Long.toString(results.get(0).stableId), 0.8f)); // alpha
|
|
scores.add(new Pair<>(Long.toString(results.get(1).stableId), 0.2f)); // bravo
|
|
return scores;
|
|
}
|
|
}
|