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 17 package com.android.quicksearchbox; 18 19 import android.database.DataSetObservable; 20 import android.database.DataSetObserver; 21 import android.util.Log; 22 23 import java.util.ArrayList; 24 import java.util.Collections; 25 import java.util.HashMap; 26 import java.util.HashSet; 27 import java.util.List; 28 29 30 /** 31 * Mock implementation of {@link Corpora}. 32 */ 33 public class MockCorpora implements Corpora { 34 35 private static final String TAG = "QSB.MockCorpora"; 36 37 private final DataSetObservable mDataSetObservable = new DataSetObservable(); 38 39 private HashMap<String,Corpus> mCorporaByName = new HashMap<String,Corpus>(); 40 private HashSet<Corpus> mDefaultCorpora = new HashSet<Corpus>(); 41 42 private Corpus mWebCorpus; 43 44 public void addCorpus(Corpus corpus) { 45 Corpus oldCorpus = mCorporaByName.put(corpus.getName(), corpus); 46 if (oldCorpus != null) { 47 Log.d(TAG, "Replaced " + oldCorpus + " with " + corpus); 48 } 49 notifyDataSetChanged(); 50 } 51 52 public void setWebCorpus(Corpus webCorpus) { 53 mWebCorpus = webCorpus; 54 } 55 56 public void addDefaultCorpus(Corpus corpus) { 57 mDefaultCorpora.add(corpus); 58 } 59 60 public List<Corpus> getAllCorpora() { 61 return Collections.unmodifiableList(new ArrayList<Corpus>(mCorporaByName.values())); 62 } 63 64 public Corpus getCorpus(String name) { 65 return mCorporaByName.get(name); 66 } 67 68 public Corpus getWebCorpus() { 69 return mWebCorpus; 70 } 71 72 public Corpus getCorpusForSource(Source source) { 73 for (Corpus corpus : mCorporaByName.values()) { 74 for (Source corpusSource : corpus.getSources()) { 75 if (corpusSource.equals(source)) { 76 return corpus; 77 } 78 } 79 } 80 return null; 81 } 82 83 public List<Corpus> getEnabledCorpora() { 84 return getAllCorpora(); 85 } 86 87 public List<Corpus> getCorporaInAll() { 88 return getAllCorpora(); 89 } 90 91 public Source getSource(String name) { 92 for (Corpus corpus : mCorporaByName.values()) { 93 for (Source source : corpus.getSources()) { 94 if (source.getName().equals(name)) { 95 return source; 96 } 97 } 98 } 99 return null; 100 } 101 102 public boolean isCorpusDefaultEnabled(Corpus corpus) { 103 return mDefaultCorpora.contains(corpus); 104 } 105 106 public boolean isCorpusEnabled(Corpus corpus) { 107 return true; 108 } 109 110 public void update() { 111 } 112 113 public void registerDataSetObserver(DataSetObserver observer) { 114 mDataSetObservable.registerObserver(observer); 115 } 116 117 public void unregisterDataSetObserver(DataSetObserver observer) { 118 mDataSetObservable.unregisterObserver(observer); 119 } 120 121 protected void notifyDataSetChanged() { 122 mDataSetObservable.notifyChanged(); 123 } 124 125 } 126