1 /* 2 * Copyright (C) 2009 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 com.android.quicksearchbox; 18 19 import android.test.AndroidTestCase; 20 import android.test.suitebuilder.annotation.SmallTest; 21 22 import java.util.Set; 23 24 /** 25 * Tests for {@link Suggestions}. 26 * 27 */ 28 @SmallTest 29 public class SuggestionsTest extends AndroidTestCase { 30 31 private Suggestions mSuggestions; 32 33 @Override 34 protected void setUp() throws Exception { 35 mSuggestions = new Suggestions(null, 0, "foo", 2); 36 } 37 38 @Override 39 protected void tearDown() throws Exception { 40 mSuggestions.close(); 41 mSuggestions = null; 42 } 43 44 public void testGetExpectedSourceCount() { 45 assertEquals(2, mSuggestions.getExpectedSourceCount()); 46 } 47 48 public void testGetUserQuery() { 49 assertEquals("foo", mSuggestions.getQuery()); 50 } 51 52 public void testGetIncludedCorpora() { 53 Corpus corpus = MockCorpus.CORPUS_1; 54 mSuggestions.addCorpusResult(corpus.getSuggestions("foo", 50)); 55 Set<Corpus> includedCorpora = mSuggestions.getIncludedCorpora(); 56 assertEquals(includedCorpora.size(), 1); 57 assertTrue(includedCorpora.contains(corpus)); 58 } 59 60 } 61
