Fix broken unit test
Change-Id: I65f2784dfc6643ee22e7e6679b36f08e50d44766
diff --git a/tests/src/com/android/quicksearchbox/MockCorpus.java b/tests/src/com/android/quicksearchbox/MockCorpus.java
index f151b41..d2975cd 100644
--- a/tests/src/com/android/quicksearchbox/MockCorpus.java
+++ b/tests/src/com/android/quicksearchbox/MockCorpus.java
@@ -38,9 +38,16 @@
private final Source mSource;
+ private final boolean mDefaultEnabled;
+
public MockCorpus(Source source) {
+ this(source, true);
+ }
+
+ public MockCorpus(Source source, boolean defaultEnabled) {
mName = "corpus_" + source.getName();
mSource = source;
+ mDefaultEnabled = defaultEnabled;
}
public Intent createSearchIntent(String query, Bundle appData) {
@@ -137,7 +144,7 @@
}
public boolean isCorpusDefaultEnabled() {
- return true;
+ return mDefaultEnabled;
}
public boolean isCorpusEnabled() {
diff --git a/tests/src/com/android/quicksearchbox/RankAwarePromoterTest.java b/tests/src/com/android/quicksearchbox/RankAwarePromoterTest.java
index 8765179..c701867 100644
--- a/tests/src/com/android/quicksearchbox/RankAwarePromoterTest.java
+++ b/tests/src/com/android/quicksearchbox/RankAwarePromoterTest.java
@@ -80,11 +80,8 @@
MockCorpora corpora = new MockCorpora();
for (int i = 0; i < count; i++) {
Source mockSource = new MockSource("Source" + i);
- Corpus mockCorpus = new MockCorpus(mockSource);
+ Corpus mockCorpus = new MockCorpus(mockSource, i < defaultCount);
corpora.addCorpus(mockCorpus);
- if (i < defaultCount) {
- corpora.addDefaultCorpus(mockCorpus);
- }
}
return corpora;
}
