tests/tests/widget/src/android/widget/cts/SearchView_CursorTest.java - platform/cts - Git at Google

 /*
  * Copyright (C) 2016 The Android Open Source Project
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
  * You may obtain a copy of the License at
  *
  *      http://www.apache.org/licenses/LICENSE-2.0
  *
  * Unless required by applicable law or agreed to in writing, software
  * distributed under the License is distributed on an "AS IS" BASIS,
  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */

 package android.widget.cts;

 import static org.junit.Assert.assertEquals;
 import static org.junit.Assert.assertTrue;
 import static org.mockito.Matchers.anyInt;
 import static org.mockito.Matchers.anyString;
 import static org.mockito.Mockito.spy;
 import static org.mockito.Mockito.times;
 import static org.mockito.Mockito.verify;
 import static org.mockito.Mockito.verifyNoMoreInteractions;
 import static org.mockito.Mockito.when;

 import android.app.Activity;
 import android.app.Instrumentation;
 import android.database.Cursor;
 import android.database.MatrixCursor;
 import android.provider.BaseColumns;
 import android.support.test.InstrumentationRegistry;
 import android.support.test.annotation.UiThreadTest;
 import android.support.test.filters.MediumTest;
 import android.support.test.rule.ActivityTestRule;
 import android.support.test.runner.AndroidJUnit4;
 import android.text.TextUtils;
 import android.view.View;
 import android.view.ViewGroup;
 import android.widget.AutoCompleteTextView;
 import android.widget.CursorAdapter;
 import android.widget.SearchView;
 import android.widget.SimpleCursorAdapter;

 import com.android.compatibility.common.util.CtsTouchUtils;
 import com.android.compatibility.common.util.PollingCheck;

 import org.junit.Before;
 import org.junit.Rule;
 import org.junit.Test;
 import org.junit.runner.RunWith;

 /**
  * Test {@link SearchView} with {@link Cursor}-backed suggestions adapter.
  */
 @MediumTest
 @RunWith(AndroidJUnit4.class)
 public class SearchView_CursorTest {
     private Instrumentation mInstrumentation;
     private Activity mActivity;
     private SearchView mSearchView;

     private static final String TEXT_COLUMN_NAME = "text";
     private String[] mTextContent;

     private CursorAdapter mSuggestionsAdapter;

     // This should be protected to spy an object of this class.
     protected class MyQueryTextListener implements SearchView.OnQueryTextListener {
         @Override
         public boolean onQueryTextSubmit(String s) {
             return false;
         }

         @Override
         public boolean onQueryTextChange(String s) {
             if (mSuggestionsAdapter == null) {
                 return false;
             }
             if (!enoughToFilter()) {
                 return false;
             }
             final MatrixCursor c = new MatrixCursor(
                     new String[] { BaseColumns._ID, TEXT_COLUMN_NAME} );
             for (int i = 0; i < mTextContent.length; i++) {
                 if (mTextContent[i].toLowerCase().startsWith(s.toLowerCase())) {
                     c.addRow(new Object[]{i, mTextContent[i]});
                 }
             }
             mSuggestionsAdapter.swapCursor(c);
             return false;
         }

         private boolean enoughToFilter() {
             final AutoCompleteTextView searchSrcText = findAutoCompleteTextView(mSearchView);
             return searchSrcText != null && searchSrcText.enoughToFilter();
         }

         private AutoCompleteTextView findAutoCompleteTextView(final ViewGroup viewGroup) {
             final int count = viewGroup.getChildCount();
             for (int index = 0; index < count; index++) {
                 final View view = viewGroup.getChildAt(index);
                 if (view instanceof AutoCompleteTextView) {
                     return (AutoCompleteTextView) view;
                 }
                 if (view instanceof ViewGroup) {
                     final AutoCompleteTextView findView =
                             findAutoCompleteTextView((ViewGroup) view);
                     if (findView != null) {
                         return findView;
                     }
                 }
             }
             return null;
         }
     }

     // This should be protected to spy an object of this class.
     protected class MySuggestionListener implements SearchView.OnSuggestionListener {
         @Override
         public boolean onSuggestionSelect(int position) {
             return false;
         }

         @Override
         public boolean onSuggestionClick(int position) {
             if (mSuggestionsAdapter != null) {
                 final Cursor cursor = mSuggestionsAdapter.getCursor();
                 if (cursor != null) {
                     cursor.moveToPosition(position);
                     mSearchView.setQuery(cursor.getString(1), false);
                 }
             }
             return true;
         }
     }

     @Rule
     public ActivityTestRule<SearchViewCtsActivity> mActivityRule =
             new ActivityTestRule<>(SearchViewCtsActivity.class);

     @UiThreadTest
     @Before
     public void setup() throws Throwable {
         mInstrumentation = InstrumentationRegistry.getInstrumentation();
         mActivity = mActivityRule.getActivity();
         mSearchView = (SearchView) mActivity.findViewById(R.id.search_view);

         // Local test data for the tests
         mTextContent = new String[] { "Akon", "Bono", "Ciara", "Dido", "Diplo" };

         // Use an adapter with our custom layout for each entry. The adapter "maps"
         // the content of the text column of our cursor to the @id/text1 view in the
         // layout.
         mActivityRule.runOnUiThread(() -> {
             mSuggestionsAdapter = new SimpleCursorAdapter(
                     mActivity,
                     R.layout.searchview_suggestion_item,
                     null,
                     new String[] { TEXT_COLUMN_NAME },
                     new int[] { android.R.id.text1 },
                     CursorAdapter.FLAG_REGISTER_CONTENT_OBSERVER);
             mSearchView.setSuggestionsAdapter(mSuggestionsAdapter);
         });
     }

     @UiThreadTest
     @Test
     public void testSuggestionFiltering() {
         final SearchView.OnQueryTextListener mockQueryTextListener =
                 spy(new MyQueryTextListener());
         when(mockQueryTextListener.onQueryTextChange(anyString())).thenCallRealMethod();

         mSearchView.setIconifiedByDefault(false);
         mSearchView.setOnQueryTextListener(mockQueryTextListener);
         mSearchView.requestFocus();

         assertTrue(mSearchView.hasFocus());
         assertEquals(mSuggestionsAdapter, mSearchView.getSuggestionsAdapter());

         mSearchView.setQuery("Bon", false);
         verify(mockQueryTextListener, times(1)).onQueryTextChange("Bon");

         mSearchView.setQuery("Di", false);
         verify(mockQueryTextListener, times(1)).onQueryTextChange("Di");
     }

     @Test
     public void testSuggestionSelection() throws Throwable {
         final SearchView.OnSuggestionListener mockSuggestionListener =
                 spy(new MySuggestionListener());
         when(mockSuggestionListener.onSuggestionClick(anyInt())).thenCallRealMethod();

         final SearchView.OnQueryTextListener mockQueryTextListener =
                 spy(new MyQueryTextListener());
         when(mockQueryTextListener.onQueryTextChange(anyString())).thenCallRealMethod();

         mActivityRule.runOnUiThread(() -> {
                     mSearchView.setIconifiedByDefault(false);
                     mSearchView.setOnQueryTextListener(mockQueryTextListener);
                     mSearchView.setOnSuggestionListener(mockSuggestionListener);
                     mSearchView.requestFocus();
                 });

         assertTrue(mSearchView.hasFocus());
         assertEquals(mSuggestionsAdapter, mSearchView.getSuggestionsAdapter());

         mActivityRule.runOnUiThread(() -> mSearchView.setQuery("Di", false));
         mInstrumentation.waitForIdleSync();
         verify(mockQueryTextListener, times(1)).onQueryTextChange("Di");

         // Emulate click on the first suggestion - which should be Dido
         final int suggestionRowHeight = mActivity.getResources().getDimensionPixelSize(
                 R.dimen.search_view_suggestion_row_height);
         CtsTouchUtils.emulateTapOnView(mInstrumentation, mSearchView, mSearchView.getWidth() / 2,
                 mSearchView.getHeight() + suggestionRowHeight / 2);

         // At this point we expect the click on the first suggestion to have activated a sequence
         // of events that ends up in our suggestion listener that sets the full suggestion text
         // as the current query. Some parts of this sequence of events are asynchronous, and those
         // are not "caught" by Instrumentation.waitForIdleSync - which is in general not a very
         // reliable way to wait for everything to be completed. As such, we are using our own
         // polling check mechanism to wait until the search view's query is the fully completed
         // suggestion for Dido. This check will time out and fail after a few seconds if anything
         // goes wrong during the processing of the emulated tap and the code never gets to our
         // suggestion listener
         PollingCheck.waitFor(() -> TextUtils.equals("Dido", mSearchView.getQuery()));

         // Just to be sure, verify that our spy suggestion listener was called
         verify(mockSuggestionListener, times(1)).onSuggestionClick(0);
         verifyNoMoreInteractions(mockSuggestionListener);
     }
 }
	/*
	* Copyright (C) 2016 The Android Open Source Project
	*
	* Licensed under the Apache License, Version 2.0 (the "License");
	* you may not use this file except in compliance with the License.
	* You may obtain a copy of the License at
	*
	* http://www.apache.org/licenses/LICENSE-2.0
	*
	* Unless required by applicable law or agreed to in writing, software
	* distributed under the License is distributed on an "AS IS" BASIS,
	* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
	* See the License for the specific language governing permissions and
	* limitations under the License.
	*/

	package android.widget.cts;

	import static org.junit.Assert.assertEquals;
	import static org.junit.Assert.assertTrue;
	import static org.mockito.Matchers.anyInt;
	import static org.mockito.Matchers.anyString;
	import static org.mockito.Mockito.spy;
	import static org.mockito.Mockito.times;
	import static org.mockito.Mockito.verify;
	import static org.mockito.Mockito.verifyNoMoreInteractions;
	import static org.mockito.Mockito.when;

	import android.app.Activity;
	import android.app.Instrumentation;
	import android.database.Cursor;
	import android.database.MatrixCursor;
	import android.provider.BaseColumns;
	import android.support.test.InstrumentationRegistry;
	import android.support.test.annotation.UiThreadTest;
	import android.support.test.filters.MediumTest;
	import android.support.test.rule.ActivityTestRule;
	import android.support.test.runner.AndroidJUnit4;
	import android.text.TextUtils;
	import android.view.View;
	import android.view.ViewGroup;
	import android.widget.AutoCompleteTextView;
	import android.widget.CursorAdapter;
	import android.widget.SearchView;
	import android.widget.SimpleCursorAdapter;

	import com.android.compatibility.common.util.CtsTouchUtils;
	import com.android.compatibility.common.util.PollingCheck;

	import org.junit.Before;
	import org.junit.Rule;
	import org.junit.Test;
	import org.junit.runner.RunWith;

	/**
	* Test {@link SearchView} with {@link Cursor}-backed suggestions adapter.
	*/
	@MediumTest
	@RunWith(AndroidJUnit4.class)
	public class SearchView_CursorTest {
	private Instrumentation mInstrumentation;
	private Activity mActivity;
	private SearchView mSearchView;

	private static final String TEXT_COLUMN_NAME = "text";
	private String[] mTextContent;

	private CursorAdapter mSuggestionsAdapter;

	// This should be protected to spy an object of this class.
	protected class MyQueryTextListener implements SearchView.OnQueryTextListener {
	@Override
	public boolean onQueryTextSubmit(String s) {
	return false;
	}

	@Override
	public boolean onQueryTextChange(String s) {
	if (mSuggestionsAdapter == null) {
	return false;
	}
	if (!enoughToFilter()) {
	return false;
	}
	final MatrixCursor c = new MatrixCursor(
	new String[] { BaseColumns._ID, TEXT_COLUMN_NAME} );
	for (int i = 0; i < mTextContent.length; i++) {
	if (mTextContent[i].toLowerCase().startsWith(s.toLowerCase())) {
	c.addRow(new Object[]{i, mTextContent[i]});
	}
	}
	mSuggestionsAdapter.swapCursor(c);
	return false;
	}

	private boolean enoughToFilter() {
	final AutoCompleteTextView searchSrcText = findAutoCompleteTextView(mSearchView);
	return searchSrcText != null && searchSrcText.enoughToFilter();
	}

	private AutoCompleteTextView findAutoCompleteTextView(final ViewGroup viewGroup) {
	final int count = viewGroup.getChildCount();
	for (int index = 0; index < count; index++) {
	final View view = viewGroup.getChildAt(index);
	if (view instanceof AutoCompleteTextView) {
	return (AutoCompleteTextView) view;
	}
	if (view instanceof ViewGroup) {
	final AutoCompleteTextView findView =
	findAutoCompleteTextView((ViewGroup) view);
	if (findView != null) {
	return findView;
	}
	}
	}
	return null;
	}
	}

	// This should be protected to spy an object of this class.
	protected class MySuggestionListener implements SearchView.OnSuggestionListener {
	@Override
	public boolean onSuggestionSelect(int position) {
	return false;
	}

	@Override
	public boolean onSuggestionClick(int position) {
	if (mSuggestionsAdapter != null) {
	final Cursor cursor = mSuggestionsAdapter.getCursor();
	if (cursor != null) {
	cursor.moveToPosition(position);
	mSearchView.setQuery(cursor.getString(1), false);
	}
	}
	return true;
	}
	}

	@Rule
	public ActivityTestRule<SearchViewCtsActivity> mActivityRule =
	new ActivityTestRule<>(SearchViewCtsActivity.class);

	@UiThreadTest
	@Before
	public void setup() throws Throwable {
	mInstrumentation = InstrumentationRegistry.getInstrumentation();
	mActivity = mActivityRule.getActivity();
	mSearchView = (SearchView) mActivity.findViewById(R.id.search_view);

	// Local test data for the tests
	mTextContent = new String[] { "Akon", "Bono", "Ciara", "Dido", "Diplo" };

	// Use an adapter with our custom layout for each entry. The adapter "maps"
	// the content of the text column of our cursor to the @id/text1 view in the
	// layout.
	mActivityRule.runOnUiThread(() -> {
	mSuggestionsAdapter = new SimpleCursorAdapter(
	mActivity,
	R.layout.searchview_suggestion_item,
	null,
	new String[] { TEXT_COLUMN_NAME },
	new int[] { android.R.id.text1 },
	CursorAdapter.FLAG_REGISTER_CONTENT_OBSERVER);
	mSearchView.setSuggestionsAdapter(mSuggestionsAdapter);
	});
	}

	@UiThreadTest
	@Test
	public void testSuggestionFiltering() {
	final SearchView.OnQueryTextListener mockQueryTextListener =
	spy(new MyQueryTextListener());
	when(mockQueryTextListener.onQueryTextChange(anyString())).thenCallRealMethod();

	mSearchView.setIconifiedByDefault(false);
	mSearchView.setOnQueryTextListener(mockQueryTextListener);
	mSearchView.requestFocus();

	assertTrue(mSearchView.hasFocus());
	assertEquals(mSuggestionsAdapter, mSearchView.getSuggestionsAdapter());

	mSearchView.setQuery("Bon", false);
	verify(mockQueryTextListener, times(1)).onQueryTextChange("Bon");

	mSearchView.setQuery("Di", false);
	verify(mockQueryTextListener, times(1)).onQueryTextChange("Di");
	}

	@Test
	public void testSuggestionSelection() throws Throwable {
	final SearchView.OnSuggestionListener mockSuggestionListener =
	spy(new MySuggestionListener());
	when(mockSuggestionListener.onSuggestionClick(anyInt())).thenCallRealMethod();

	final SearchView.OnQueryTextListener mockQueryTextListener =
	spy(new MyQueryTextListener());
	when(mockQueryTextListener.onQueryTextChange(anyString())).thenCallRealMethod();

	mActivityRule.runOnUiThread(() -> {
	mSearchView.setIconifiedByDefault(false);
	mSearchView.setOnQueryTextListener(mockQueryTextListener);
	mSearchView.setOnSuggestionListener(mockSuggestionListener);
	mSearchView.requestFocus();
	});

	assertTrue(mSearchView.hasFocus());
	assertEquals(mSuggestionsAdapter, mSearchView.getSuggestionsAdapter());

	mActivityRule.runOnUiThread(() -> mSearchView.setQuery("Di", false));
	mInstrumentation.waitForIdleSync();
	verify(mockQueryTextListener, times(1)).onQueryTextChange("Di");

	// Emulate click on the first suggestion - which should be Dido
	final int suggestionRowHeight = mActivity.getResources().getDimensionPixelSize(
	R.dimen.search_view_suggestion_row_height);
	CtsTouchUtils.emulateTapOnView(mInstrumentation, mSearchView, mSearchView.getWidth() / 2,
	mSearchView.getHeight() + suggestionRowHeight / 2);

	// At this point we expect the click on the first suggestion to have activated a sequence
	// of events that ends up in our suggestion listener that sets the full suggestion text
	// as the current query. Some parts of this sequence of events are asynchronous, and those
	// are not "caught" by Instrumentation.waitForIdleSync - which is in general not a very
	// reliable way to wait for everything to be completed. As such, we are using our own
	// polling check mechanism to wait until the search view's query is the fully completed
	// suggestion for Dido. This check will time out and fail after a few seconds if anything
	// goes wrong during the processing of the emulated tap and the code never gets to our
	// suggestion listener
	PollingCheck.waitFor(() -> TextUtils.equals("Dido", mSearchView.getQuery()));

	// Just to be sure, verify that our spy suggestion listener was called
	verify(mockSuggestionListener, times(1)).onSuggestionClick(0);
	verifyNoMoreInteractions(mockSuggestionListener);
	}
	}