Push full slice index to device index

Test: make RunSettingsRoboTests
Bug: 74555610
Change-Id: I3f0aa1218e1d7e736dc918d83e76423fa81ac6ab

tests/robotests/src/com/android/settings/search/DeviceIndexUpdateJobServiceTest.java

@@ -0,0 +1,146 @@
+/*
+ * Copyright (C) 2018 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.settings.search;
+
+import static org.mockito.ArgumentMatchers.any;
+import static org.mockito.ArgumentMatchers.eq;
+import static org.mockito.Mockito.atLeast;
+import static org.mockito.Mockito.doAnswer;
+import static org.mockito.Mockito.doNothing;
+import static org.mockito.Mockito.doReturn;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.spy;
+import static org.mockito.Mockito.times;
+import static org.mockito.Mockito.verify;
+import static org.mockito.Mockito.when;
+
+import android.app.Activity;
+import android.content.ContentResolver;
+import android.net.Uri;
+
+import com.android.settings.slices.SettingsSliceProvider;
+import com.android.settings.testutils.FakeFeatureFactory;
+import com.android.settings.testutils.SettingsRobolectricTestRunner;
+
+import org.junit.Before;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.mockito.invocation.InvocationOnMock;
+import org.mockito.stubbing.Answer;
+import org.robolectric.Robolectric;
+
+import java.util.ArrayList;
+import java.util.List;
+
+import androidx.slice.Slice;
+import androidx.slice.SliceManager;
+import androidx.slice.SliceMetadata;
+
+@RunWith(SettingsRobolectricTestRunner.class)
+public class DeviceIndexUpdateJobServiceTest {
+        private static final Uri BASE_URI = new Uri.Builder()
+                .scheme(ContentResolver.SCHEME_CONTENT)
+                .authority(SettingsSliceProvider.SLICE_AUTHORITY)
+                .build();
+
+    private Activity mActivity;
+    private DeviceIndexUpdateJobService mJob;
+    private SliceManager mSliceManager;
+
+    @Before
+    public void setup() {
+        FakeFeatureFactory.setupForTest();
+        mActivity = spy(Robolectric.buildActivity(Activity.class).create().visible().get());
+        mJob = spy(new DeviceIndexUpdateJobService());
+        mSliceManager = mock(SliceManager.class);
+
+        doReturn(mActivity.getPackageName()).when(mJob).getPackageName();
+        doReturn(mSliceManager).when(mJob).getSliceManager();
+        doNothing().when(mJob).jobFinished(null, false);
+    }
+
+    @Test
+    public void testGetsSlices() {
+        setSlices();
+
+        mJob.updateIndex(null);
+        verify(mSliceManager).getSliceDescendants(eq(BASE_URI));
+    }
+
+    @Test
+    public void testIndexesSlices() {
+        setSlices(genSlice("path1"), genSlice("path2"));
+
+        mJob.mRunningJob = true;
+        mJob.updateIndex(null);
+        verify(mSliceManager).getSliceDescendants(eq(BASE_URI));
+
+        DeviceIndexFeatureProvider indexFeatureProvider = FakeFeatureFactory.getFactory(mActivity)
+                .getDeviceIndexFeatureProvider();
+        verify(indexFeatureProvider, times(2)).index(any(), any(), any(), any(), any());
+    }
+
+    @Test
+    public void testDoNotIndexWithoutTitle() {
+        Slice testSlice = genSlice("path2");
+        setSlices(genSlice("path1"), testSlice);
+        doReturn(null).when(mJob).findTitle(testSlice, mJob.getMetadata(testSlice));
+
+        mJob.mRunningJob = true;
+        mJob.updateIndex(null);
+        verify(mSliceManager).getSliceDescendants(eq(BASE_URI));
+
+        DeviceIndexFeatureProvider indexFeatureProvider = FakeFeatureFactory.getFactory(mActivity)
+                .getDeviceIndexFeatureProvider();
+        verify(indexFeatureProvider, times(1)).index(any(), any(), any(), any(), any());
+    }
+
+    @Test
+    public void testStopIndexing() {
+        Slice testSlice = genSlice("path1");
+        setSlices(testSlice, genSlice("path2"));
+        mJob.mRunningJob = true;
+
+        doAnswer(invocation -> {
+            // Stop running after the first iteration
+            mJob.mRunningJob = false;
+            return testSlice;
+        }).when(mJob).bindSliceSynchronous(mSliceManager, testSlice.getUri());
+
+        mJob.updateIndex(null);
+        verify(mSliceManager).getSliceDescendants(eq(BASE_URI));
+
+        DeviceIndexFeatureProvider indexFeatureProvider = FakeFeatureFactory.getFactory(mActivity)
+                .getDeviceIndexFeatureProvider();
+        verify(indexFeatureProvider, times(1)).index(any(), any(), any(), any(), any());
+    }
+
+    private Slice genSlice(String path) {
+        return new Slice.Builder(BASE_URI.buildUpon().path(path).build()).build();
+    }
+
+    private void setSlices(Slice... slice) {
+        List<Uri> mUris = new ArrayList<>();
+        for (Slice slouse : slice) {
+            SliceMetadata m = mock(SliceMetadata.class);
+            mUris.add(slouse.getUri());
+            doReturn(slouse).when(mJob).bindSliceSynchronous(mSliceManager, slouse.getUri());
+            doReturn(m).when(mJob).getMetadata(slouse);
+            doReturn(slouse.getUri().getPath()).when(mJob).findTitle(slouse, m);
+        }
+        when(mSliceManager.getSliceDescendants(BASE_URI)).thenReturn(mUris);
+    }
+
+}