Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit 8d6c2f80 authored by ronish's avatar ronish
Browse files

Add tests for QRCodeScannerController 

Bug: 200536509
Test: atest QRCodeScannerControllerTest

Change-Id: Ib4c6455c68784d390a7da511c54bda90c0bac912
parent 0f68d24b

packages/SystemUI/tests/src/com/android/systemui/qrcodescanner/controller/QRCodeScannerControllerTest.java

0 → 100644
+324 −0
Original line number Diff line number Diff line 
/*

 * Copyright (C) 2021 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.systemui.qrcodescanner.controller;



import static android.provider.Settings.Secure.LOCK_SCREEN_SHOW_QR_CODE_SCANNER;



import static com.android.systemui.qrcodescanner.controller.QRCodeScannerController.DEFAULT_QR_CODE_SCANNER_CHANGE;

import static com.android.systemui.qrcodescanner.controller.QRCodeScannerController.QR_CODE_SCANNER_PREFERENCE_CHANGE;



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



import static org.mockito.ArgumentMatchers.any;

import static org.mockito.Mockito.times;

import static org.mockito.Mockito.verify;

import static org.mockito.Mockito.when;



import android.content.Intent;

import android.content.pm.PackageManager;

import android.content.pm.ResolveInfo;

import android.os.UserHandle;

import android.provider.DeviceConfig;

import android.provider.Settings;

import android.testing.AndroidTestingRunner;

import android.testing.TestableLooper;



import androidx.test.filters.SmallTest;



import com.android.internal.config.sysui.SystemUiDeviceConfigFlags;

import com.android.systemui.R;

import com.android.systemui.SysuiTestCase;

import com.android.systemui.settings.UserTracker;

import com.android.systemui.util.DeviceConfigProxyFake;

import com.android.systemui.util.settings.FakeSettings;



import com.google.common.util.concurrent.MoreExecutors;



import org.junit.Test;

import org.junit.runner.RunWith;

import org.mockito.Mock;

import org.mockito.MockitoAnnotations;



import java.util.ArrayList;

import java.util.Collections;

import java.util.List;



@RunWith(AndroidTestingRunner.class)

@TestableLooper.RunWithLooper

@SmallTest

public class QRCodeScannerControllerTest extends SysuiTestCase {

    @Mock

    private PackageManager mPackageManager;

    @Mock

    private QRCodeScannerController.Callback mCallback;

    @Mock

    private UserTracker mUserTracker;



    private FakeSettings mSecureSettings;

    private QRCodeScannerController mController;

    private DeviceConfigProxyFake mProxyFake;



    private void setUpLocal(String deviceConfigActivity, String defaultActivity,

            boolean validateActivity, boolean enableSetting) {

        MockitoAnnotations.initMocks(this);

        int enableSettingInt = enableSetting ? 1 : 0;



        mSecureSettings = new FakeSettings();

        mSecureSettings.putInt(LOCK_SCREEN_SHOW_QR_CODE_SCANNER, enableSettingInt);



        mContext.setMockPackageManager(mPackageManager);



        List<ResolveInfo> resolveInfoList = new ArrayList();

        if (validateActivity) {

            resolveInfoList = new ArrayList(Collections.singleton(new ResolveInfo()));

        }

        when(mPackageManager.queryIntentActivities(any(Intent.class),

                any(Integer.class))).thenReturn(resolveInfoList);

        when(mPackageManager.hasSystemFeature(PackageManager.FEATURE_CAMERA)).thenReturn(true);

        mContext.getOrCreateTestableResources().addOverride(R.string.def_qr_code_component,

                defaultActivity);



        mProxyFake = new DeviceConfigProxyFake();

        mProxyFake.setProperty(DeviceConfig.NAMESPACE_SYSTEMUI,

                SystemUiDeviceConfigFlags.DEFAULT_QR_CODE_SCANNER,

                deviceConfigActivity, false);



        when(mUserTracker.getUserId()).thenReturn(UserHandle.USER_CURRENT);



        mController = new QRCodeScannerController(

                mContext,

                MoreExecutors.directExecutor(),

                mSecureSettings, mProxyFake, mUserTracker);

        mController.addCallback(mCallback);

        mController.registerQRCodeScannerChangeObservers(DEFAULT_QR_CODE_SCANNER_CHANGE,

                QR_CODE_SCANNER_PREFERENCE_CHANGE);

    }



    private String getSettingsQRCodeDefaultComponent() {

        return mSecureSettings.getStringForUser(Settings.Secure.SHOW_QR_CODE_SCANNER_SETTING,

                UserHandle.USER_CURRENT);

    }



    private void verifyActivityDetails(String componentName) {

        if (componentName != null) {

            assertThat(mController.getIntent()).isNotNull();

            assertThat(componentName).isEqualTo(getSettingsQRCodeDefaultComponent());

        } else {

            assertThat(mController.getIntent()).isNull();

            assertThat(getSettingsQRCodeDefaultComponent()).isNull();

        }

    }



    @Test

    public void qrCodeScannerInit_withoutDefaultValue() {

        setUpLocal(/* deviceConfigActivity */ null, /* defaultActivity */

                "", /* validateActivity */ true, /* enableSetting */true);

        verifyActivityDetails(null);

        assertThat(mController.isEnabledForLockScreenButton()).isFalse();

        assertThat(mController.isEnabledForQuickSettings()).isFalse();

    }



    @Test

    public void qrCodeScannerInit_withIncorrectDefaultValue() {

        setUpLocal(/* deviceConfigActivity */ null, /* defaultActivity */

                "abc/.def", /* validateActivity */ false, /* enableSetting */ true);

        verifyActivityDetails(null);

        assertThat(mController.isEnabledForLockScreenButton()).isFalse();

    }



    @Test

    public void qrCodeScannerInit_withCorrectDefaultValue() {

        setUpLocal(/* deviceConfigActivity */ null, /* defaultActivity */

                "abc/.def", /* validateActivity */ true, /* enableSetting */true);

        verifyActivityDetails("abc/.def");

        assertThat(mController.isEnabledForLockScreenButton()).isTrue();

        assertThat(mController.isEnabledForQuickSettings()).isTrue();

    }



    @Test

    public void qrCodeScannerInit_withCorrectDeviceConfig() {

        setUpLocal(/* deviceConfigActivity */ "abc/.def", /* defaultActivity */

                "", /* validateActivity */ true, /* enableSetting */true);

        verifyActivityDetails("abc/.def");

        assertThat(mController.isEnabledForLockScreenButton()).isTrue();

        assertThat(mController.isEnabledForQuickSettings()).isTrue();

    }



    @Test

    public void qrCodeScannerInit_withCorrectDeviceConfig_withCorrectDefaultValue() {

        setUpLocal(/* deviceConfigActivity */ "abc/.def", /* defaultActivity */

                "xyz/.qrs", /* validateActivity */true, /* enableSetting */ true);

        verifyActivityDetails("abc/.def");

        assertThat(mController.isEnabledForLockScreenButton()).isTrue();

        assertThat(mController.isEnabledForQuickSettings()).isTrue();

    }



    @Test

    public void qrCodeScannerInit_withCorrectDeviceConfig_fullActivity() {

        setUpLocal(/* deviceConfigActivity */ "abc/abc.def", /* defaultActivity */

                "", /* validateActivity */  true, /* enableSetting */ true);

        verifyActivityDetails("abc/abc.def");

        assertThat(mController.isEnabledForLockScreenButton()).isTrue();

        assertThat(mController.isEnabledForQuickSettings()).isTrue();

    }



    @Test

    public void qrCodeScannerInit_withIncorrectDeviceConfig() {

        setUpLocal(/* deviceConfigActivity */ "def/.efg", /* defaultActivity */

                "", /* validateActivity */ false, /* enableSetting */ true);

        verifyActivityDetails(null);

        assertThat(mController.isEnabledForLockScreenButton()).isFalse();

        assertThat(mController.isEnabledForQuickSettings()).isFalse();

    }



    @Test

    public void verifyDeviceConfigChange_withDefaultActivity() {

        setUpLocal(/* deviceConfigActivity */ null, /* defaultActivity */

                "abc/.def", /* validateActivity */ true, /* enableSetting */true);

        verifyActivityDetails("abc/.def");

        assertThat(mController.isEnabledForLockScreenButton()).isTrue();

        assertThat(mController.isEnabledForQuickSettings()).isTrue();



        mProxyFake.setProperty(DeviceConfig.NAMESPACE_SYSTEMUI,

                SystemUiDeviceConfigFlags.DEFAULT_QR_CODE_SCANNER,

                "def/.ijk", false);

        verifyActivityDetails("def/.ijk");

        assertThat(mController.isEnabledForLockScreenButton()).isTrue();

        assertThat(mController.isEnabledForQuickSettings()).isTrue();



        mProxyFake.setProperty(DeviceConfig.NAMESPACE_SYSTEMUI,

                SystemUiDeviceConfigFlags.DEFAULT_QR_CODE_SCANNER,

                null, false);

        verifyActivityDetails("abc/.def");

        assertThat(mController.isEnabledForLockScreenButton()).isTrue();

        assertThat(mController.isEnabledForQuickSettings()).isTrue();



        // Once from setup + twice from this function

        verify(mCallback, times(3)).onQRCodeScannerActivityChanged();

    }



    @Test

    public void verifyDeviceConfigChange_withoutDefaultActivity() {

        setUpLocal(/* deviceConfigActivity */ null, /* defaultActivity */

                "", /* validateActivity */ true, /* enableSetting */ true);

        verifyActivityDetails(null);

        assertThat(mController.isEnabledForLockScreenButton()).isFalse();

        assertThat(mController.isEnabledForQuickSettings()).isFalse();



        mProxyFake.setProperty(DeviceConfig.NAMESPACE_SYSTEMUI,

                SystemUiDeviceConfigFlags.DEFAULT_QR_CODE_SCANNER,

                "def/.ijk", false);



        verifyActivityDetails("def/.ijk");

        assertThat(mController.isEnabledForLockScreenButton()).isTrue();

        assertThat(mController.isEnabledForQuickSettings()).isTrue();



        mProxyFake.setProperty(DeviceConfig.NAMESPACE_SYSTEMUI,

                SystemUiDeviceConfigFlags.DEFAULT_QR_CODE_SCANNER,

                null, false);

        verifyActivityDetails(null);

        assertThat(mController.isEnabledForLockScreenButton()).isFalse();

        assertThat(mController.isEnabledForQuickSettings()).isFalse();

        verify(mCallback, times(2)).onQRCodeScannerActivityChanged();

    }



    @Test

    public void verifyDeviceConfigChangeToSameValue() {

        setUpLocal(/* deviceConfigActivity */ null, /* defaultActivity */

                "", /* validateActivity */ true, /* enableSetting */true);



        mProxyFake.setProperty(DeviceConfig.NAMESPACE_SYSTEMUI,

                SystemUiDeviceConfigFlags.DEFAULT_QR_CODE_SCANNER,

                "def/.ijk", false);

        mProxyFake.setProperty(DeviceConfig.NAMESPACE_SYSTEMUI,

                SystemUiDeviceConfigFlags.DEFAULT_QR_CODE_SCANNER,

                "def/.ijk", false);



        mProxyFake.setProperty(DeviceConfig.NAMESPACE_SYSTEMUI,

                SystemUiDeviceConfigFlags.DEFAULT_QR_CODE_SCANNER,

                null, false);

        mProxyFake.setProperty(DeviceConfig.NAMESPACE_SYSTEMUI,

                SystemUiDeviceConfigFlags.DEFAULT_QR_CODE_SCANNER,

                null, false);



        verify(mCallback, times(2)).onQRCodeScannerActivityChanged();

    }



    @Test

    public void verifyPreferenceChange() {

        setUpLocal(/* deviceConfigActivity */ null, /* defaultActivity */

                "abc/.def", /* validateActivity */ true, /* enableSetting */true);

        mSecureSettings.putStringForUser(LOCK_SCREEN_SHOW_QR_CODE_SCANNER, "0",

                UserHandle.USER_CURRENT);

        mSecureSettings.putStringForUser(LOCK_SCREEN_SHOW_QR_CODE_SCANNER, "0",

                UserHandle.USER_CURRENT);



        mSecureSettings.putStringForUser(LOCK_SCREEN_SHOW_QR_CODE_SCANNER, "1",

                UserHandle.USER_CURRENT);

        mSecureSettings.putStringForUser(LOCK_SCREEN_SHOW_QR_CODE_SCANNER, "1",

                UserHandle.USER_CURRENT);

        // Once from setup + twice from this function

        verify(mCallback, times(3)).onQRCodeScannerPreferenceChanged();

    }



    @Test

    public void verifyPreferenceChangeToSameValue() {

        setUpLocal(/* deviceConfigActivity */ null, /* defaultActivity */

                "abc/.def", /* validateActivity */ true, /* enableSetting */true);

        verifyActivityDetails("abc/.def");

        assertThat(mController.isEnabledForLockScreenButton()).isTrue();

        assertThat(mController.isEnabledForQuickSettings()).isTrue();



        mSecureSettings.putStringForUser(LOCK_SCREEN_SHOW_QR_CODE_SCANNER, "0",

                UserHandle.USER_CURRENT);

        verifyActivityDetails("abc/.def");

        assertThat(mController.isEnabledForLockScreenButton()).isFalse();

        assertThat(mController.isEnabledForQuickSettings()).isTrue();



        mSecureSettings.putStringForUser(LOCK_SCREEN_SHOW_QR_CODE_SCANNER, "1",

                UserHandle.USER_CURRENT);

        verifyActivityDetails("abc/.def");

        assertThat(mController.isEnabledForLockScreenButton()).isTrue();

        assertThat(mController.isEnabledForQuickSettings()).isTrue();

        // Once from setup + twice from this function

        verify(mCallback, times(3)).onQRCodeScannerPreferenceChanged();

    }



    @Test

    public void verifyUnregisterRegisterChangeObservers() {

        setUpLocal(/* deviceConfigActivity */ null, /* defaultActivity */

                "abc/.def", /* validateActivity */ true, /* enableSetting */true);

        verifyActivityDetails("abc/.def");

        assertThat(mController.isEnabledForLockScreenButton()).isTrue();

        assertThat(mController.isEnabledForQuickSettings()).isTrue();



        mController.unregisterQRCodeScannerChangeObservers(DEFAULT_QR_CODE_SCANNER_CHANGE,

                QR_CODE_SCANNER_PREFERENCE_CHANGE);

        verifyActivityDetails(null);

        assertThat(mController.isEnabledForLockScreenButton()).isFalse();

        assertThat(mController.isEnabledForQuickSettings()).isFalse();



        // Unregister once again and make sure, it affect affect the next register event

        mController.unregisterQRCodeScannerChangeObservers(DEFAULT_QR_CODE_SCANNER_CHANGE,

                QR_CODE_SCANNER_PREFERENCE_CHANGE);

        mController.registerQRCodeScannerChangeObservers(DEFAULT_QR_CODE_SCANNER_CHANGE,

                QR_CODE_SCANNER_PREFERENCE_CHANGE);

        verifyActivityDetails("abc/.def");

        assertThat(mController.isEnabledForLockScreenButton()).isTrue();

        assertThat(mController.isEnabledForQuickSettings()).isTrue();

    }

}