1 files changed, 132 insertions, 0 deletions
diff --git a/tests/unit/src/com/android/tv/search/LocalSearchProviderTest.java b/tests/unit/src/com/android/tv/search/LocalSearchProviderTest.java
new file mode 100644
index 00000000..b0d342c6
--- /dev/null
+++ b/tests/unit/src/com/android/tv/search/LocalSearchProviderTest.java
@@ -0,0 +1,132 @@
+/*
+ * Copyright (C) 2017 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 com.android.tv.search;
+
+import static android.support.test.InstrumentationRegistry.getTargetContext;
+import static org.mockito.Mockito.clearInvocations;
+import static org.mockito.Mockito.verify;
+
+import android.app.SearchManager;
+import android.database.Cursor;
+import android.net.Uri;
+import android.support.test.filters.SmallTest;
+import android.test.ProviderTestCase2;
+
+import com.android.tv.ApplicationSingletons;
+import com.android.tv.TvApplication;
+import com.android.tv.perf.PerformanceMonitor;
+import com.android.tv.util.MockApplicationSingletons;
+
+import org.junit.After;
+import org.junit.Before;
+import org.junit.Test;
+import org.mockito.Mock;
+import org.mockito.MockitoAnnotations;
+
+/** Unit test for {@link LocalSearchProvider}. */
+@SmallTest
+public class LocalSearchProviderTest extends ProviderTestCase2<LocalSearchProvider> {
+    private static final String AUTHORITY = "com.android.tv.search";
+    private static final String KEYWORD = "keyword";
+    private static final Uri BASE_SEARCH_URI = Uri.parse("content://" + AUTHORITY + "/"
+            + SearchManager.SUGGEST_URI_PATH_QUERY + "/" + KEYWORD);
+    private static final Uri WRONG_SERACH_URI = Uri.parse("content://" + AUTHORITY + "/wrong_path/"
+            + KEYWORD);
+
+    private ApplicationSingletons mOldAppSingletons;
+    MockApplicationSingletons mMockAppSingletons;
+    @Mock PerformanceMonitor mMockPerformanceMointor;
+    @Mock SearchInterface mMockSearchInterface;
+
+    public LocalSearchProviderTest() {
+        super(LocalSearchProvider.class, AUTHORITY);
+    }
+
+    @Before
+    @Override
+    public void setUp() throws Exception {
+        MockitoAnnotations.initMocks(this);
+        setContext(getTargetContext());
+        mOldAppSingletons = TvApplication.sAppSingletons;
+        mMockAppSingletons = new MockApplicationSingletons(getTargetContext());
+        mMockAppSingletons.setPerformanceMonitor(mMockPerformanceMointor);
+        TvApplication.sAppSingletons = mMockAppSingletons;
+        super.setUp();
+        getProvider().setSearchInterface(mMockSearchInterface);
+    }
+
+    @After
+    @Override
+    public void tearDown() throws Exception {
+        TvApplication.sAppSingletons = mOldAppSingletons;
+        super.tearDown();
+    }
+
+    @Test
+    public void testQuery_normalUri() {
+        verifyQueryWithArguments(null, null);
+        verifyQueryWithArguments(1, null);
+        verifyQueryWithArguments(null, 1);
+        verifyQueryWithArguments(1, 1);
+    }
+
+    @Test
+    public void testQuery_invalidUri() {
+        try (Cursor c = getProvider().query(WRONG_SERACH_URI, null, null, null, null)) {
+            fail("Query with invalid URI should fail.");
+        } catch (IllegalArgumentException e) {
+            // Success.
+        }
+    }
+
+    @Test
+    public void testQuery_invalidLimit() {
+        verifyQueryWithArguments(-1, null);
+    }
+
+    @Test
+    public void testQuery_invalidAction() {
+        verifyQueryWithArguments(null, SearchInterface.ACTION_TYPE_START - 1);
+        verifyQueryWithArguments(null, SearchInterface.ACTION_TYPE_END + 1);
+    }
+
+    private void verifyQueryWithArguments(Integer limit, Integer action) {
+        Uri uri = BASE_SEARCH_URI;
+        if (limit != null || action != null) {
+            Uri.Builder builder = uri.buildUpon();
+            if (limit != null) {
+                builder.appendQueryParameter(SearchManager.SUGGEST_PARAMETER_LIMIT,
+                        limit.toString());
+            }
+            if (action != null) {
+                builder.appendQueryParameter(LocalSearchProvider.SUGGEST_PARAMETER_ACTION,
+                        action.toString());
+            }
+            uri = builder.build();
+        }
+        try (Cursor c = getProvider().query(uri, null, null, null, null)) {
+            // Do nothing.
+        }
+        int expectedLimit = limit == null || limit <= 0 ?
+                LocalSearchProvider.DEFAULT_SEARCH_LIMIT : limit;
+        int expectedAction = (action == null || action < SearchInterface.ACTION_TYPE_START
+                || action > SearchInterface.ACTION_TYPE_END) ?
+                LocalSearchProvider.DEFAULT_SEARCH_ACTION : action;
+        verify(mMockSearchInterface).search(KEYWORD, expectedLimit, expectedAction);
+        clearInvocations(mMockSearchInterface);
+    }
+}