mirror of
https://github.com/mozilla/gecko-dev.git
synced 2024-10-31 14:15:30 +00:00
136 lines
6.5 KiB
Java
136 lines
6.5 KiB
Java
#filter substitution
|
|
package @ANDROID_PACKAGE_NAME@.tests;
|
|
|
|
import @ANDROID_PACKAGE_NAME@.*;
|
|
import android.app.Activity;
|
|
import android.content.Context;
|
|
import android.view.View;
|
|
import android.view.ViewGroup;
|
|
import android.widget.TextView;
|
|
import android.util.Log;
|
|
|
|
import java.lang.reflect.Constructor;
|
|
import java.lang.reflect.Field;
|
|
import java.lang.reflect.Method;
|
|
import java.lang.RuntimeException;
|
|
import java.util.ArrayList;
|
|
import java.util.HashMap;
|
|
|
|
/**
|
|
* Test for search suggestions.
|
|
* Sends queries from AwesomeBar input and verifies that suggestions match
|
|
* expected values.
|
|
*/
|
|
public class testSearchSuggestions extends BaseTest {
|
|
private static final int SUGGESTION_MAX = 3;
|
|
private static final int SUGGESTION_TIMEOUT = 5000;
|
|
private static final String TEST_QUERY = "foo barz";
|
|
private static final String SUGGESTION_TEMPLATE = "/robocop/robocop_suggestions.sjs?query=__searchTerms__";
|
|
|
|
@Override
|
|
protected int getTestType() {
|
|
return TEST_MOCHITEST;
|
|
}
|
|
|
|
public void testSearchSuggestions() {
|
|
mActions.expectGeckoEvent("Gecko:Ready").blockForEvent();
|
|
|
|
// Map of expected values. See robocop_suggestions.sjs.
|
|
final HashMap<String, ArrayList<String>> suggestMap = new HashMap<String, ArrayList<String>>();
|
|
buildSuggestMap(suggestMap);
|
|
|
|
final int suggestionLayoutId = mDriver.findElement(getActivity(), "suggestion_layout").getId();
|
|
final int suggestionTextId = mDriver.findElement(getActivity(), "suggestion_text").getId();
|
|
|
|
// workaround for bug 769524
|
|
mSolo.sleep(5000);
|
|
|
|
Actions.EventExpecter enginesEventExpecter = mActions.expectGeckoEvent("SearchEngines:Data");
|
|
final Activity awesomeBarActivity = clickOnAwesomeBar();
|
|
enginesEventExpecter.blockForEvent();
|
|
connectSuggestClient(awesomeBarActivity);
|
|
|
|
for (int i = 0; i < TEST_QUERY.length(); i++) {
|
|
mActions.sendKeys(TEST_QUERY.substring(i, i+1));
|
|
|
|
final String query = TEST_QUERY.substring(0, i+1);
|
|
boolean success = waitForTest(new BooleanTest() {
|
|
public boolean test() {
|
|
// get the first suggestion row
|
|
ViewGroup suggestionGroup = (ViewGroup) awesomeBarActivity.findViewById(suggestionLayoutId);
|
|
if (suggestionGroup == null)
|
|
return false;
|
|
|
|
ArrayList<String> expected = suggestMap.get(query);
|
|
for (int i = 0; i < expected.size(); i++) {
|
|
View queryChild = suggestionGroup.getChildAt(i);
|
|
if (queryChild == null || queryChild.getVisibility() == View.GONE)
|
|
return false;
|
|
|
|
String suggestion = ((TextView) queryChild.findViewById(suggestionTextId)).getText().toString();
|
|
if (!suggestion.equals(expected.get(i)))
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
}, SUGGESTION_TIMEOUT);
|
|
mAsserter.is(success, true, "Results for query '" + query + "' matched expected suggestions");
|
|
}
|
|
}
|
|
|
|
private void buildSuggestMap(HashMap<String, ArrayList<String>> suggestMap) {
|
|
// these values assume SUGGESTION_MAX = 3
|
|
suggestMap.put("f", new ArrayList<String>() {{ add("f"); add("facebook"); add("fandango"); add("frys"); }});
|
|
suggestMap.put("fo", new ArrayList<String>() {{ add("fo"); add("forever 21"); add("food network"); add("fox news"); }});
|
|
suggestMap.put("foo", new ArrayList<String>() {{ add("foo"); add("food network"); add("foothill college"); add("foot locker"); }});
|
|
suggestMap.put("foo ", new ArrayList<String>() {{ add("foo "); add("foo fighters"); add("foo bar"); add("foo bat"); }});
|
|
suggestMap.put("foo b", new ArrayList<String>() {{ add("foo b"); add("foo bar"); add("foo bat"); add("foo bay"); }});
|
|
suggestMap.put("foo ba", new ArrayList<String>() {{ add("foo ba"); add("foo bar"); add("foo bat"); add("foo bay"); }});
|
|
suggestMap.put("foo bar", new ArrayList<String>() {{ add("foo bar"); }});
|
|
suggestMap.put("foo barz", new ArrayList<String>() {{ add("foo barz"); }});
|
|
}
|
|
|
|
private void connectSuggestClient(final Activity awesomeBarActivity) {
|
|
final int awesomeBarTabsId = mDriver.findElement(getActivity(), "awesomebar_tabs").getId();
|
|
|
|
try {
|
|
// create a SuggestClient that uses robocop_suggestions.sjs
|
|
ClassLoader classLoader = getActivity().getApplicationContext().getClassLoader();
|
|
Class suggestClass = classLoader.loadClass("org.mozilla.gecko.SuggestClient");
|
|
Constructor suggestConstructor = suggestClass.getConstructor(
|
|
new Class[] { Context.class, String.class, int.class, int.class });
|
|
String suggestTemplate = getAbsoluteRawUrl(SUGGESTION_TEMPLATE);
|
|
Object client = suggestConstructor.newInstance(awesomeBarActivity, suggestTemplate, SUGGESTION_TIMEOUT, SUGGESTION_MAX);
|
|
|
|
// enable offline HTTP requests for testing
|
|
final Field checkNetworkField = suggestClass.getDeclaredField("mCheckNetwork");
|
|
checkNetworkField.setAccessible(true);
|
|
checkNetworkField.setBoolean(client, false);
|
|
|
|
// replace mSuggestClient with test client
|
|
final Class awesomeBarTabsClass = classLoader.loadClass("org.mozilla.gecko.AwesomeBarTabs");
|
|
final Method getAllPagesTabMethod = awesomeBarTabsClass.getMethod("getAllPagesTab");
|
|
final View awesomeBarTabs = (View) awesomeBarActivity.findViewById(awesomeBarTabsId);
|
|
final Class allPagesTabClass = classLoader.loadClass("org.mozilla.gecko.AllPagesTab");
|
|
final Field suggestClientField = allPagesTabClass.getDeclaredField("mSuggestClient");
|
|
final Object allPagesTab = getAllPagesTabMethod.invoke(awesomeBarTabs);
|
|
suggestClientField.setAccessible(true);
|
|
waitForTest(new BooleanTest() {
|
|
public boolean test() {
|
|
// wait for mSuggestClient to be set before we replace it
|
|
try {
|
|
return suggestClientField.get(allPagesTab) != null;
|
|
} catch (IllegalAccessException e) {
|
|
return false;
|
|
}
|
|
}
|
|
}, SUGGESTION_TIMEOUT);
|
|
suggestClientField.set(allPagesTab, client);
|
|
} catch (Exception e) {
|
|
throw new RuntimeException("Error setting SuggestClient", e);
|
|
}
|
|
}
|
|
}
|
|
|