tests/autofillservice/src/android/autofillservice/cts/inline/InlineFillEventHistoryTest.java

/*
 * Copyright (C) 2020 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.autofillservice.cts.inline;

import static android.autofillservice.cts.Helper.ID_PASSWORD;
import static android.autofillservice.cts.Helper.ID_USERNAME;
import static android.autofillservice.cts.Helper.NULL_DATASET_ID;
import static android.autofillservice.cts.Helper.assertFillEventForDatasetSelected;
import static android.autofillservice.cts.Helper.assertFillEventForDatasetShown;
import static android.autofillservice.cts.Helper.assertFillEventForSaveShown;
import static android.autofillservice.cts.Helper.assertNoDeprecatedClientState;
import static android.autofillservice.cts.Helper.getContext;
import static android.autofillservice.cts.inline.InstrumentedAutoFillServiceInlineEnabled.SERVICE_NAME;
import static android.service.autofill.SaveInfo.SAVE_DATA_TYPE_GENERIC;

import android.autofillservice.cts.CannedFillResponse;
import android.autofillservice.cts.CannedFillResponse.CannedDataset;
import android.autofillservice.cts.FillEventHistoryCommonTestCase;
import android.autofillservice.cts.Helper;
import android.autofillservice.cts.InstrumentedAutoFillService;
import android.autofillservice.cts.LoginActivity;
import android.platform.test.annotations.AppModeFull;
import android.service.autofill.FillEventHistory;
import android.service.autofill.FillEventHistory.Event;
import android.support.test.uiautomator.UiObject2;

import org.junit.Test;

import java.util.List;

/**
 * Test that uses {@link LoginActivity} to test {@link FillEventHistory}.
 */
@AppModeFull(reason = "Service-specific test")
public class InlineFillEventHistoryTest extends FillEventHistoryCommonTestCase {

    public InlineFillEventHistoryTest() {
        super(getInlineUiBot());
    }

    @Override
    protected boolean isInlineMode() {
        return true;
    }

    @Override
    protected void enableService() {
        Helper.enableAutofillService(getContext(), SERVICE_NAME);
    }

    @Test
    public void testOneDatasetAndSave() throws Exception {
        enableService();

        // Set expectations.
        sReplier.addResponse(new CannedFillResponse.Builder()
                .setRequiredSavableIds(SAVE_DATA_TYPE_GENERIC, ID_USERNAME)
                .addDataset(new CannedDataset.Builder()
                        .setField(ID_USERNAME, "id")
                        .setField(ID_PASSWORD, "pass")
                        .setPresentation(createPresentation("Dataset"))
                        .setInlinePresentation(createInlinePresentation("Dataset"))
                        .build())
                .build());

        // Trigger auto-fill and IME.
        mUiBot.selectByRelativeId(ID_USERNAME);
        mUiBot.waitForIdle();
        sReplier.getNextFillRequest();

        // Suggestion strip was shown.
        mUiBot.assertDatasets("Dataset");
        mUiBot.selectDataset("Dataset");
        mUiBot.waitForIdle();

        // Change username and password
        mActivity.syncRunOnUiThread(() ->  mActivity.onUsername((v) -> v.setText("ID")));
        mActivity.syncRunOnUiThread(() ->  mActivity.onPassword((v) -> v.setText("PASS")));
        mUiBot.waitForIdle();

        // Trigger save UI.
        mActivity.tapSave();
        mUiBot.waitForIdle();

        // Confirm the save UI shown
        final UiObject2 saveUi = mUiBot.assertUpdateShowing(SAVE_DATA_TYPE_GENERIC);

        // Save it...
        mUiBot.saveForAutofill(saveUi, true);
        mUiBot.waitForIdle();
        sReplier.getNextSaveRequest();

        // Verify events history
        final FillEventHistory selection = InstrumentedAutoFillService.getFillEventHistory(4);
        assertNoDeprecatedClientState(selection);
        final List<Event> events = selection.getEvents();
        assertFillEventForDatasetShown(events.get(0));
        assertFillEventForDatasetSelected(events.get(1), NULL_DATASET_ID);
        assertFillEventForDatasetShown(events.get(0));
        assertFillEventForSaveShown(events.get(3), NULL_DATASET_ID);
    }
}