managedprovisioning/provisioning/ProvisioningActivityRoboTest.java

/*
 * Copyright (C) 2019 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.managedprovisioning.provisioning;

import static android.app.admin.DevicePolicyManager.ACTION_PROVISION_MANAGED_DEVICE;
import static android.app.admin.DevicePolicyManager.ACTION_PROVISION_MANAGED_PROFILE;

import static com.android.managedprovisioning.common.LogoUtils.saveOrganisationLogo;
import static com.android.managedprovisioning.provisioning.AbstractProvisioningActivity.CANCEL_PROVISIONING_DIALOG_OK;
import static com.android.managedprovisioning.provisioning.AbstractProvisioningActivity.ERROR_DIALOG_OK;
import static com.android.managedprovisioning.provisioning.AbstractProvisioningActivity.ERROR_DIALOG_RESET;
import static com.android.managedprovisioning.provisioning.ProvisioningActivity.RESULT_CODE_DEVICE_OWNER_SET;
import static com.android.managedprovisioning.provisioning.ProvisioningActivity.RESULT_CODE_WORK_PROFILE_CREATED;

import static com.google.common.truth.Truth.assertThat;

import static org.mockito.Mockito.never;
import static org.mockito.Mockito.verify;
import static org.robolectric.Shadows.shadowOf;
import static org.robolectric.shadows.ShadowUserManager.FLAG_MANAGED_PROFILE;

import android.app.Activity;
import android.app.Application;
import android.app.DialogFragment;
import android.app.Fragment;
import android.app.admin.DevicePolicyManager;
import android.content.ComponentName;
import android.content.Context;
import android.content.Intent;
import android.graphics.Bitmap;
import android.graphics.Color;
import android.graphics.PorterDuffColorFilter;
import android.graphics.drawable.BitmapDrawable;
import android.graphics.drawable.Drawable;
import android.graphics.drawable.VectorDrawable;
import android.net.Uri;
import android.os.UserManager;
import android.widget.ImageView;

import androidx.test.core.content.pm.ApplicationInfoBuilder;
import androidx.test.core.content.pm.PackageInfoBuilder;

import com.android.managedprovisioning.R;
import com.android.managedprovisioning.model.ProvisioningParams;

import org.junit.Ignore;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.Mockito;
import org.robolectric.Robolectric;
import org.robolectric.RobolectricTestRunner;
import org.robolectric.RuntimeEnvironment;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.InputStream;
import java.util.List;

/**
 * Robolectric tests for {@link ProvisioningActivity}.
 */
@RunWith(RobolectricTestRunner.class)
public class ProvisioningActivityRoboTest {

    private static final String ADMIN_PACKAGE = "com.test.admin";
    private static final ComponentName ADMIN = new ComponentName(ADMIN_PACKAGE, ".Receiver");
    private static final ProvisioningParams DEVICE_OWNER_PARAMS = new ProvisioningParams.Builder()
            .setProvisioningAction(ACTION_PROVISION_MANAGED_DEVICE)
            .setDeviceAdminComponentName(ADMIN)
            .build();
    private static final ProvisioningParams PROFILE_OWNER_PARAMS = new ProvisioningParams.Builder()
            .setProvisioningAction(ACTION_PROVISION_MANAGED_PROFILE)
            .setDeviceAdminComponentName(ADMIN)
            .build();
    private static final ProvisioningParams QR_PROVISIONING_PARAMS_DO = new ProvisioningParams.Builder()
            .setProvisioningAction(ACTION_PROVISION_MANAGED_DEVICE)
            .setDeviceAdminComponentName(ADMIN)
            .setIsOrganizationOwnedProvisioning(true)
            .setFlowType(ProvisioningParams.FLOW_TYPE_ADMIN_INTEGRATED)
            .build();
    private static final ProvisioningParams QR_PROVISIONING_PARAMS_PO = new ProvisioningParams.Builder()
            .setProvisioningAction(ACTION_PROVISION_MANAGED_PROFILE)
            .setDeviceAdminComponentName(ADMIN)
            .setIsOrganizationOwnedProvisioning(true)
            .setFlowType(ProvisioningParams.FLOW_TYPE_ADMIN_INTEGRATED)
            .build();
    private static final Intent PROFILE_OWNER_INTENT = new Intent()
            .putExtra(ProvisioningParams.EXTRA_PROVISIONING_PARAMS, PROFILE_OWNER_PARAMS);
    private static final Intent DEVICE_OWNER_INTENT = new Intent()
            .putExtra(ProvisioningParams.EXTRA_PROVISIONING_PARAMS, DEVICE_OWNER_PARAMS);
    private static final Intent ADMIN_INTEGRATED_FLOW_INTENT_PO = new Intent()
            .putExtra(ProvisioningParams.EXTRA_PROVISIONING_PARAMS, QR_PROVISIONING_PARAMS_PO);
    private static final Intent ADMIN_INTEGRATED_FLOW_INTENT_DO = new Intent()
            .putExtra(ProvisioningParams.EXTRA_PROVISIONING_PARAMS, QR_PROVISIONING_PARAMS_DO);
    private static final int ERROR_MESSAGE_ID = R.string.managed_provisioning_error_text;
    private static final int DEFAULT_LOGO_COLOR = -15043608;
    private static final int CUSTOM_COLOR = Color.parseColor("#d40000");
    private static final Uri LOGO_URI = Uri.parse("http://logo");

    private ProvisioningManager mMockProvisioningManager = Mockito.mock(ProvisioningManager.class);
    private Application mContext = RuntimeEnvironment.application;

    @Test
    public void error_noFactoryReset_showsDialogue() {
        final ProvisioningActivity activity =
                Robolectric.buildActivity(ProvisioningActivity.class, PROFILE_OWNER_INTENT).get();

        activity.error(R.string.cant_set_up_device, ERROR_MESSAGE_ID, /* resetRequired= */ false);

        final Fragment dialog = activity.getFragmentManager().findFragmentByTag(ERROR_DIALOG_OK);
        assertThat(dialog).isNotNull();
    }

    @Test
    public void error_noFactoryReset_doesNotReset() throws Exception {
        final ProvisioningActivity activity =
                Robolectric.buildActivity(ProvisioningActivity.class, PROFILE_OWNER_INTENT).get();
        activity.error(R.string.cant_set_up_device, ERROR_MESSAGE_ID, /* resetRequired= */ false);

        final Fragment dialog = activity.getFragmentManager().findFragmentByTag(ERROR_DIALOG_OK);
        clickOnPositiveButton(activity, (DialogFragment) dialog);

        final List<Intent> intents = shadowOf(mContext).getBroadcastIntents();
        assertThat(intentsContainsAction(intents, Intent.ACTION_FACTORY_RESET)).isFalse();
    }

    @Test
    public void error_factoryReset_showsDialogue() {
        final ProvisioningActivity activity =
                Robolectric.buildActivity(ProvisioningActivity.class, PROFILE_OWNER_INTENT).get();

        activity.error(R.string.cant_set_up_device, ERROR_MESSAGE_ID, /* resetRequired= */ true);

        final Fragment dialog = activity.getFragmentManager().findFragmentByTag(ERROR_DIALOG_RESET);
        assertThat(dialog).isNotNull();
    }

    @Test
    public void error_factoryReset_resets() throws Exception {
        DevicePolicyManager devicePolicyManager =
                mContext.getSystemService(DevicePolicyManager.class);
        final ProvisioningActivity activity =
                Robolectric.buildActivity(ProvisioningActivity.class, PROFILE_OWNER_INTENT).get();
        activity.error(R.string.cant_set_up_device, ERROR_MESSAGE_ID, /* resetRequired= */ true);

        final Fragment dialog = activity.getFragmentManager().findFragmentByTag(ERROR_DIALOG_RESET);
        clickOnPositiveButton(activity, (DialogFragment) dialog);

        assertThat(shadowOf(devicePolicyManager).getWipeCalledTimes()).isEqualTo(1);
    }

    @Ignore("b/181326453")
    @Test
    public void profileOwnerIntent_usesDefaultLogo() throws Throwable {
        final ProvisioningActivity activity =
                Robolectric.buildActivity(ProvisioningActivity.class, PROFILE_OWNER_INTENT)
                        .setup().get();

        assertUsesDefaultLogo(activity);
    }

    @Ignore("b/181326453")
    @Test
    public void profileOwnerIntent_setCustomLogo_usesCustomLogo() throws Throwable {
        setupCustomLogo(mContext, LOGO_URI);

        final ProvisioningActivity activity =
                Robolectric.buildActivity(ProvisioningActivity.class, PROFILE_OWNER_INTENT)
                        .setup().get();

        assertUsesCustomLogo(activity);
    }

    @Ignore("b/181326453")
    @Test
    public void deviceOwnerIntent_usesDefaultLogo() throws Throwable {
        final ProvisioningActivity activity =
                Robolectric.buildActivity(ProvisioningActivity.class, DEVICE_OWNER_INTENT)
                        .setup().get();

        assertUsesDefaultLogo(activity);
    }

    @Ignore("b/181326453")
    @Test
    public void deviceOwnerIntent_setCustomLogo_usesCustomLogo() throws Throwable {
        setupCustomLogo(mContext, LOGO_URI);

        final ProvisioningActivity activity =
                Robolectric.buildActivity(ProvisioningActivity.class, DEVICE_OWNER_INTENT)
                        .setup().get();

        assertUsesCustomLogo(activity);
    }

    @Ignore("b/181326453")
    @Test
    public void managedProfileIntent_defaultColor_colorCorrect() {
        assertColorsCorrect(
                PROFILE_OWNER_INTENT,
                DEFAULT_LOGO_COLOR);
    }

    @Ignore("b/181326453")
    @Test
    public void deviceOwnerIntent_defaultColor_colorCorrect() {
        assertColorsCorrect(
                DEVICE_OWNER_INTENT,
                DEFAULT_LOGO_COLOR);
    }

    @Ignore("b/181326453")
    @Test
    public void activity_profileOwner_backPressed_showsCancelDialog() throws Throwable {
        final ProvisioningActivity activity =
                Robolectric.buildActivity(ProvisioningActivity.class, PROFILE_OWNER_INTENT)
                        .setup().get();

        activity.onBackPressed();

        final Fragment dialog =
                activity.getFragmentManager().findFragmentByTag(CANCEL_PROVISIONING_DIALOG_OK);
        assertThat(dialog).isNotNull();
    }

    @Ignore("b/181326453")
    @Test
    public void activity_profileOwner_backPressed_doNotCancel_doesNotFinishActivity() {
        final ProvisioningActivity activity =
                Robolectric.buildActivity(ProvisioningActivity.class, PROFILE_OWNER_INTENT)
                        .setup().get();

        activity.onBackPressed();
        final Fragment dialog =
                activity.getFragmentManager().findFragmentByTag(CANCEL_PROVISIONING_DIALOG_OK);
        clickOnNegativeButton(activity, (DialogFragment) dialog);

        assertThat(activity.isFinishing()).isFalse();
    }

    @Ignore("b/181326453")
    @Test
    public void activity_profileOwner_backPressed_doNotCancel_doesNotCancelProvisioning() {
        final ProvisioningActivity activity =
                Robolectric.buildActivity(ProvisioningActivity.class, PROFILE_OWNER_INTENT)
                        .setup().get();
        activity.setProvisioningManager(mMockProvisioningManager);

        activity.onBackPressed();
        final Fragment dialog =
                activity.getFragmentManager().findFragmentByTag(CANCEL_PROVISIONING_DIALOG_OK);
        clickOnNegativeButton(activity, (DialogFragment) dialog);

        verify(mMockProvisioningManager, never()).cancelProvisioning();
    }

    @Ignore("b/181326453")
    @Test
    public void activity_profileOwner_backPressed_cancel_doesFinishActivity() {
        final ProvisioningActivity activity =
                Robolectric.buildActivity(ProvisioningActivity.class, PROFILE_OWNER_INTENT)
                        .setup().get();

        activity.onBackPressed();
        final Fragment dialog =
                activity.getFragmentManager().findFragmentByTag(CANCEL_PROVISIONING_DIALOG_OK);
        clickOnPositiveButton(activity, (DialogFragment) dialog);

        assertThat(activity.isFinishing()).isTrue();
    }

    @Ignore("b/181326453")
    @Test
    public void activity_profileOwner_backPressed_cancel_doesCancelProvisioning() {
        final ProvisioningActivity activity =
                Robolectric.buildActivity(ProvisioningActivity.class, PROFILE_OWNER_INTENT)
                        .setup().get();
        activity.setProvisioningManager(mMockProvisioningManager);

        activity.onBackPressed();
        final Fragment dialog =
                activity.getFragmentManager().findFragmentByTag(CANCEL_PROVISIONING_DIALOG_OK);
        clickOnPositiveButton(activity, (DialogFragment) dialog);

        verify(mMockProvisioningManager).cancelProvisioning();
    }

    @Ignore("b/181326453")
    @Test
    public void activity_profileOwner_adminIntegrated_returnsIntermediateResult() {
        final ProvisioningActivity activity =
                Robolectric.buildActivity(ProvisioningActivity.class,
                        ADMIN_INTEGRATED_FLOW_INTENT_PO)
                        .setup().get();
        shadowOf(activity.getPackageManager())
                .installPackage(
                        PackageInfoBuilder.newBuilder()
                                .setPackageName(ADMIN_PACKAGE)
                                .setApplicationInfo(ApplicationInfoBuilder.newBuilder()
                                        .setPackageName(ADMIN_PACKAGE)
                                        .build())
                                .build());


        activity.preFinalizationCompleted();
        activity.onAllTransitionsShown();

        UserManager userManager = activity.getSystemService(UserManager.class);
        shadowOf(userManager).addProfile(0, 10, "profile name", FLAG_MANAGED_PROFILE);

        activity.onNextButtonClicked();

        assertThat(activity.isFinishing()).isTrue();
        assertThat(shadowOf(activity).getResultCode()).isEqualTo(RESULT_CODE_WORK_PROFILE_CREATED);
    }

    @Ignore("b/181326453")
    @Test
    public void activity_deviceOwner_adminIntegrated_returnsIntermediateResult() {
        final ProvisioningActivity activity =
                Robolectric.buildActivity(ProvisioningActivity.class,
                        ADMIN_INTEGRATED_FLOW_INTENT_DO)
                        .setup().get();
        shadowOf(activity.getPackageManager())
                .installPackage(
                        PackageInfoBuilder.newBuilder()
                                .setPackageName(ADMIN_PACKAGE)
                                .setApplicationInfo(ApplicationInfoBuilder.newBuilder()
                                        .setPackageName(ADMIN_PACKAGE)
                                        .build())
                                .build());


        activity.preFinalizationCompleted();
        activity.onAllTransitionsShown();

        activity.onNextButtonClicked();

        assertThat(activity.isFinishing()).isTrue();
        assertThat(shadowOf(activity).getResultCode()).isEqualTo(RESULT_CODE_DEVICE_OWNER_SET);
    }

    @Ignore("b/181326453")
    @Test
    public void activity_deviceOwner_notAdminIntegrated_returnsOk() {
        final ProvisioningActivity activity =
                Robolectric.buildActivity(ProvisioningActivity.class,
                        DEVICE_OWNER_INTENT)
                        .setup().get();
        shadowOf(activity.getPackageManager())
                .installPackage(
                        PackageInfoBuilder.newBuilder()
                                .setPackageName(ADMIN_PACKAGE)
                                .setApplicationInfo(ApplicationInfoBuilder.newBuilder()
                                        .setPackageName(ADMIN_PACKAGE)
                                        .build())
                                .build());


        activity.preFinalizationCompleted();
        activity.onAllTransitionsShown();

        activity.onNextButtonClicked();

        assertThat(activity.isFinishing()).isTrue();
        assertThat(shadowOf(activity).getResultCode()).isEqualTo(Activity.RESULT_OK);
    }

    private Intent createProvisioningIntent(String action) {
        final ProvisioningParams provisioningParams = new ProvisioningParams.Builder()
                .setProvisioningAction(action)
                .setDeviceAdminComponentName(ADMIN)
                .build();

        final Intent intent = new Intent();
        intent.putExtra(ProvisioningParams.EXTRA_PROVISIONING_PARAMS, provisioningParams);
        return intent;
    }

    private void assertColorsCorrect(Intent intent, int logoColor) {
        final ProvisioningActivity activity =
                Robolectric.buildActivity(ProvisioningActivity.class, intent)
                        .setup().get();

        assertDefaultLogoColorCorrect(activity, logoColor);
    }

    private void assertDefaultLogoColorCorrect(Activity activity, int targetColor) {
        Drawable actualLogo =
                ((ImageView) activity.findViewById(R.id.sud_layout_icon)).getDrawable();
        PorterDuffColorFilter colorFilter = (PorterDuffColorFilter) actualLogo.getColorFilter();

        assertThat(colorFilter.getColor()).isEqualTo(targetColor);
    }

    private static boolean intentsContainsAction(List<Intent> intents, String action) {
        return intents.stream().anyMatch(intent -> intent.getAction().equals(action));
    }

    private void clickOnOkButton(ProvisioningActivity activity, DialogFragment dialog) {
        // TODO(135181317): This should be replaced by
        //  activity.findViewById(android.R.id.button1).performClick();
        activity.onPositiveButtonClick(dialog);
    }

    private static void setupCustomLogo(Context context, Uri logoUri) {
        Bitmap bitmap = Bitmap.createBitmap(300, 300, Bitmap.Config.ARGB_8888);
        InputStream inputStream = bitmapToInputStream(bitmap);
        shadowOf(context.getContentResolver()).registerInputStream(logoUri, inputStream);
        saveOrganisationLogo(context, logoUri);
    }

    private static InputStream bitmapToInputStream(Bitmap bitmap) {
        ByteArrayOutputStream bos = new ByteArrayOutputStream();
        bitmap.compress(Bitmap.CompressFormat.PNG, 0 /* ignored for PNG */, bos);
        byte[] bitmapdata = bos.toByteArray();
        return new ByteArrayInputStream(bitmapdata);
    }

    private static void assertUsesDefaultLogo(Activity activity) {
        final ImageView imageView = activity.findViewById(R.id.sud_layout_icon);
        // We default to a vector logo
        assertThat(imageView.getDrawable()).isInstanceOf(VectorDrawable.class);
    }

    private static void assertUsesCustomLogo(Activity activity) {
        final ImageView imageView = activity.findViewById(R.id.sud_layout_icon);
        // The custom logo we have set is a bitmap
        assertThat(imageView.getDrawable()).isInstanceOf(BitmapDrawable.class);
    }

    private void clickOnPositiveButton(ProvisioningActivity activity, DialogFragment dialog) {
        // TODO(135181317): This should be replaced by
        //  activity.findViewById(android.R.id.button1).performClick();

        activity.onPositiveButtonClick(dialog);
    }

    private void clickOnNegativeButton(ProvisioningActivity activity, DialogFragment dialog) {
        // TODO(135181317): This should be replaced by
        //  activity.findViewById(android.R.id.button2).performClick();
        activity.onNegativeButtonClick(dialog);
    }
}