1/* 2 * Copyright (C) 2010 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 17package com.android.quicksearchbox; 18 19import android.content.Context; 20import android.content.Intent; 21import android.graphics.drawable.Drawable; 22import android.net.Uri; 23import android.os.Bundle; 24import android.test.AndroidTestCase; 25import android.test.MoreAsserts; 26import android.test.suitebuilder.annotation.LargeTest; 27 28import java.util.concurrent.Executor; 29import java.util.concurrent.Executors; 30 31/** 32 * Tests for {@link MultiSourceCorpus}. 33 */ 34@LargeTest 35public class MultiSourceCorpusTest extends AndroidTestCase { 36 37 protected MultiSourceCorpus mCorpus; 38 39 @Override 40 protected void setUp() throws Exception { 41 super.setUp(); 42 43 Config config = new Config(getContext()); 44 // Using a single thread to make the test deterministic 45 Executor executor = Executors.newSingleThreadExecutor(); 46 mCorpus = new SkeletonMultiSourceCorpus(getContext(), config, executor, 47 MockSource.SOURCE_1, MockSource.SOURCE_2); 48 } 49 50 public void testGetSources() { 51 MoreAsserts.assertContentsInOrder(mCorpus.getSources(), 52 MockSource.SOURCE_1, MockSource.SOURCE_2); 53 } 54 55 public void testGetSuggestions() { 56 ListSuggestionCursor expected = concatSuggestionCursors("foo", 57 MockSource.SOURCE_1.getSuggestions("foo", 50, false), 58 MockSource.SOURCE_2.getSuggestions("foo", 50, false)); 59 CorpusResult observed = mCorpus.getSuggestions("foo", 50, false); 60 SuggestionCursorUtil.assertSameSuggestions(expected, observed); 61 } 62 63 public void testIncludeInAll() { 64 assertTrue(mCorpus.includeInAll()); 65 } 66 67 private static ListSuggestionCursor concatSuggestionCursors(String query, 68 SuggestionCursor... cursors) { 69 ListSuggestionCursor out = new ListSuggestionCursor("foo"); 70 for (SuggestionCursor cursor : cursors) { 71 int count = cursor.getCount(); 72 for (int i = 0; i < count; i++) { 73 out.add(new SuggestionPosition(cursor, i)); 74 } 75 } 76 return out; 77 } 78 79 private static class SkeletonMultiSourceCorpus extends MultiSourceCorpus { 80 81 public SkeletonMultiSourceCorpus(Context context, Config config, Executor executor, 82 Source... sources) { 83 super(context, config, executor, sources); 84 } 85 86 public Intent createSearchIntent(String query, Bundle appData) { 87 return null; 88 } 89 90 public Intent createVoiceSearchIntent(Bundle appData) { 91 return null; 92 } 93 94 public Drawable getCorpusIcon() { 95 return null; 96 } 97 98 public Uri getCorpusIconUri() { 99 return null; 100 } 101 102 public CharSequence getHint() { 103 return null; 104 } 105 106 public CharSequence getLabel() { 107 return null; 108 } 109 110 @Override 111 public int getQueryThreshold() { 112 return 0; 113 } 114 115 public CharSequence getSettingsDescription() { 116 return null; 117 } 118 119 public boolean isWebCorpus() { 120 return false; 121 } 122 123 @Override 124 public boolean queryAfterZeroResults() { 125 return false; 126 } 127 128 @Override 129 public boolean voiceSearchEnabled() { 130 return false; 131 } 132 133 public String getName() { 134 return "SkeletonMultiSourceCorpus"; 135 } 136 137 } 138} 139