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

Commit ad62861b authored by Hieu Dang's avatar Hieu Dang Committed by Gerrit Code Review
Browse files

Merge "Add BluetoothOppBtEnablingActivityTest"

parents 33e1d946 259cceda
Loading
Loading
Loading
Loading
+0 −1
Original line number Original line Diff line number Diff line
@@ -111,7 +111,6 @@ public class BluetoothMethodProxy {
        return contentResolver.openFileDescriptor(uri, mode);
        return contentResolver.openFileDescriptor(uri, mode);
    }
    }



    /**
    /**
     * Proxies {@link HeaderSet#getHeader}.
     * Proxies {@link HeaderSet#getHeader}.
     */
     */
+8 −4
Original line number Original line Diff line number Diff line
@@ -48,7 +48,9 @@ import android.view.KeyEvent;
import android.view.View;
import android.view.View;
import android.widget.TextView;
import android.widget.TextView;


import com.android.bluetooth.BluetoothMethodProxy;
import com.android.bluetooth.R;
import com.android.bluetooth.R;
import com.android.internal.annotations.VisibleForTesting;


/**
/**
 * This class is designed to show BT enabling progress.
 * This class is designed to show BT enabling progress.
@@ -62,7 +64,8 @@ public class BluetoothOppBtEnablingActivity extends AlertActivity {


    private static final int BT_ENABLING_TIMEOUT = 0;
    private static final int BT_ENABLING_TIMEOUT = 0;


    private static final int BT_ENABLING_TIMEOUT_VALUE = 20000;
    @VisibleForTesting
    static int sBtEnablingTimeoutMs = 20000;


    private boolean mRegistered = false;
    private boolean mRegistered = false;


@@ -73,7 +76,7 @@ public class BluetoothOppBtEnablingActivity extends AlertActivity {
        getWindow().addSystemFlags(SYSTEM_FLAG_HIDE_NON_SYSTEM_OVERLAY_WINDOWS);
        getWindow().addSystemFlags(SYSTEM_FLAG_HIDE_NON_SYSTEM_OVERLAY_WINDOWS);
        // If BT is already enabled jus return.
        // If BT is already enabled jus return.
        BluetoothAdapter adapter = BluetoothAdapter.getDefaultAdapter();
        BluetoothAdapter adapter = BluetoothAdapter.getDefaultAdapter();
        if (adapter.isEnabled()) {
        if (BluetoothMethodProxy.getInstance().bluetoothAdapterIsEnabled(adapter)) {
            finish();
            finish();
            return;
            return;
        }
        }
@@ -88,7 +91,7 @@ public class BluetoothOppBtEnablingActivity extends AlertActivity {


        // Add timeout for enabling progress
        // Add timeout for enabling progress
        mTimeoutHandler.sendMessageDelayed(mTimeoutHandler.obtainMessage(BT_ENABLING_TIMEOUT),
        mTimeoutHandler.sendMessageDelayed(mTimeoutHandler.obtainMessage(BT_ENABLING_TIMEOUT),
                BT_ENABLING_TIMEOUT_VALUE);
                sBtEnablingTimeoutMs);
    }
    }


    private View createView() {
    private View createView() {
@@ -119,7 +122,8 @@ public class BluetoothOppBtEnablingActivity extends AlertActivity {
        }
        }
    }
    }


    private final Handler mTimeoutHandler = new Handler() {
    @VisibleForTesting
    final Handler mTimeoutHandler = new Handler() {
        @Override
        @Override
        public void handleMessage(Message msg) {
        public void handleMessage(Message msg) {
            switch (msg.what) {
            switch (msg.what) {
+151 −0
Original line number Original line Diff line number Diff line
/*
 * Copyright (C) 2022 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.bluetooth.opp;

import static android.content.pm.PackageManager.COMPONENT_ENABLED_STATE_DEFAULT;
import static android.content.pm.PackageManager.COMPONENT_ENABLED_STATE_ENABLED;
import static android.content.pm.PackageManager.DONT_KILL_APP;

import static androidx.lifecycle.Lifecycle.State.DESTROYED;

import static com.android.bluetooth.opp.BluetoothOppBtEnablingActivity.sBtEnablingTimeoutMs;

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

import static org.mockito.ArgumentMatchers.any;
import static org.mockito.Mockito.doReturn;
import static org.mockito.Mockito.spy;
import static org.mockito.Mockito.timeout;
import static org.mockito.Mockito.verify;

import android.content.ComponentName;
import android.content.Context;
import android.content.Intent;
import android.os.Looper;
import android.view.KeyEvent;

import androidx.lifecycle.Lifecycle;
import androidx.test.core.app.ActivityScenario;
import androidx.test.platform.app.InstrumentationRegistry;
import androidx.test.runner.AndroidJUnit4;

import com.android.bluetooth.BluetoothMethodProxy;
import com.android.bluetooth.TestUtils;

import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.Mockito;
import org.mockito.MockitoAnnotations;
import org.mockito.Spy;

@RunWith(AndroidJUnit4.class)
public class BluetoothOppBtEnablingActivityTest {
    @Spy
    BluetoothMethodProxy mBluetoothMethodProxy;

    Intent mIntent;
    Context mTargetContext;

    int mRealTimeoutValue;

    @Before
    public void setUp() {
        MockitoAnnotations.initMocks(this);
        mBluetoothMethodProxy = Mockito.spy(BluetoothMethodProxy.getInstance());
        BluetoothMethodProxy.setInstanceForTesting(mBluetoothMethodProxy);

        mTargetContext = InstrumentationRegistry.getInstrumentation().getTargetContext();

        mIntent = new Intent();
        mIntent.setClass(mTargetContext, BluetoothOppBtEnablingActivity.class);

        mRealTimeoutValue = BluetoothOppBtEnablingActivity.sBtEnablingTimeoutMs;
        enableActivity(true);
    }

    @After
    public void tearDown() {
        BluetoothMethodProxy.setInstanceForTesting(null);
        BluetoothOppBtEnablingActivity.sBtEnablingTimeoutMs = mRealTimeoutValue;
        enableActivity(false);
    }

    @Test
    public void onCreate_bluetoothEnableTimeout_finishAfterTimeout() throws Exception {
        int spedUpTimeoutValue = 1000;
        // To speed up the test
        BluetoothOppBtEnablingActivity.sBtEnablingTimeoutMs = spedUpTimeoutValue;
        doReturn(false).when(mBluetoothMethodProxy).bluetoothAdapterIsEnabled(any());

        ActivityScenario<BluetoothOppBtEnablingActivity> activityScenario = ActivityScenario.launch(
                mIntent);
        final BluetoothOppManager[] mOppManager = new BluetoothOppManager[1];
        activityScenario.onActivity(activity -> {
            // Should be cancelled after timeout
            mOppManager[0] = BluetoothOppManager.getInstance(activity);
        });
        Thread.sleep(spedUpTimeoutValue);
        assertThat(mOppManager[0].mSendingFlag).isEqualTo(false);
        assertActivityState(activityScenario, DESTROYED);
    }

    @Test
    public void onKeyDown_cancelProgress() throws Exception {
        doReturn(false).when(mBluetoothMethodProxy).bluetoothAdapterIsEnabled(any());
        ActivityScenario<BluetoothOppBtEnablingActivity> activityScenario = ActivityScenario.launch(
                mIntent);

        activityScenario.onActivity(activity -> {
            activity.onKeyDown(KeyEvent.KEYCODE_BACK,
                    new KeyEvent(KeyEvent.ACTION_DOWN, KeyEvent.KEYCODE_BACK));
            // Should be cancelled immediately
            BluetoothOppManager mOppManager = BluetoothOppManager.getInstance(activity);
            assertThat(mOppManager.mSendingFlag).isEqualTo(false);
        });
        assertActivityState(activityScenario, DESTROYED);
    }

    @Test
    public void onCreate_bluetoothAlreadyEnabled_finishImmediately() throws Exception {
        doReturn(true).when(mBluetoothMethodProxy).bluetoothAdapterIsEnabled(any());
        ActivityScenario<BluetoothOppBtEnablingActivity> activityScenario = ActivityScenario.launch(
                mIntent);
        assertActivityState(activityScenario, DESTROYED);
    }

    private void assertActivityState(ActivityScenario activityScenario, Lifecycle.State state)
      throws Exception {
        // TODO: Change this into an event driven systems
        Thread.sleep(3_000);
        assertThat(activityScenario.getState()).isEqualTo(state);
    }

    private void enableActivity(boolean enable) {
        int enabledState = enable ? COMPONENT_ENABLED_STATE_ENABLED
                : COMPONENT_ENABLED_STATE_DEFAULT;

        mTargetContext.getPackageManager().setApplicationEnabledSetting(
                mTargetContext.getPackageName(), enabledState, DONT_KILL_APP);

        ComponentName activityName = new ComponentName(mTargetContext,
                BluetoothOppTransferActivity.class);
        mTargetContext.getPackageManager().setComponentEnabledSetting(
                activityName, enabledState, DONT_KILL_APP);
    }
}