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

Commit 6fa19828 authored by hieudz's avatar hieudz
Browse files

Add BluetoothOppTransferActivityTest 

Test: atest BluetoothOppTransferActivityTest
Bug: 244534014
Tag: #refactor
Change-Id: I069555a174d9837f9d77a1dc99435d9593e55dd2
parent 04f609bd

android/app/src/com/android/bluetooth/BluetoothMethodProxy.java

+9 −0
Original line number Diff line number Diff line
@@ -17,6 +17,7 @@ 
package com.android.bluetooth;



import android.content.ContentResolver;

import android.content.ContentValues;

import android.content.Context;

import android.database.Cursor;

import android.net.Uri;
@@ -83,6 +84,14 @@ public class BluetoothMethodProxy { 
        return contentResolver.delete(url, where, selectionArgs);

    }



    /**

     * Proxies {@link ContentResolver#update(Uri, ContentValues, String, String[])}.

     */

    public int contentResolverUpdate(ContentResolver contentResolver, final Uri contentUri,

            final ContentValues contentValues, String where, String[] selectionArgs) {

        return contentResolver.update(contentUri, contentValues, where, selectionArgs);

    }



    /**

     * Proxies {@link HeaderSet#getHeader}.

     */

android/app/src/com/android/bluetooth/opp/BluetoothOppTransferActivity.java

+4 −1
Original line number Diff line number Diff line
@@ -52,6 +52,8 @@ import android.widget.Toast; 


import com.android.bluetooth.R;



import com.google.common.annotations.VisibleForTesting;



/**

 * Handle all transfer related dialogs: -Ongoing transfer -Receiving one file

 * dialog -Sending one file dialog -sending multiple files dialog -Complete
@@ -83,7 +85,8 @@ public class BluetoothOppTransferActivity extends AlertActivity 


    private TextView mLine1View, mLine2View, mLine3View, mLine5View;



    private int mWhichDialog;

    @VisibleForTesting

    int mWhichDialog;



    private BluetoothAdapter mAdapter;

android/app/src/com/android/bluetooth/opp/BluetoothOppUtility.java

+18 −8
Original line number Diff line number Diff line
@@ -50,6 +50,7 @@ import android.os.ParcelFileDescriptor; 
import android.os.SystemProperties;

import android.util.Log;



import com.android.bluetooth.BluetoothMethodProxy;

import com.android.bluetooth.R;



import java.io.File;
@@ -83,7 +84,9 @@ public class BluetoothOppUtility { 


    public static BluetoothOppTransferInfo queryRecord(Context context, Uri uri) {

        BluetoothOppTransferInfo info = new BluetoothOppTransferInfo();

        Cursor cursor = context.getContentResolver().query(uri, null, null, null, null);

        Cursor cursor = BluetoothMethodProxy.getInstance().contentResolverQuery(

                context.getContentResolver(), uri, null, null, null, null

        );

        if (cursor != null) {

            if (cursor.moveToFirst()) {

                fillRecord(context, cursor, info);
@@ -152,10 +155,14 @@ public class BluetoothOppUtility { 
    public static ArrayList<String> queryTransfersInBatch(Context context, Long timeStamp) {

        ArrayList<String> uris = new ArrayList();

        final String where = BluetoothShare.TIMESTAMP + " == " + timeStamp;

        Cursor metadataCursor =

                context.getContentResolver().query(BluetoothShare.CONTENT_URI, new String[]{

                        BluetoothShare._DATA

                }, where, null, BluetoothShare._ID);

        Cursor metadataCursor = BluetoothMethodProxy.getInstance().contentResolverQuery(

                context.getContentResolver(),

                BluetoothShare.CONTENT_URI,

                new String[]{BluetoothShare._DATA},

                where,

                null,

                BluetoothShare._ID

        );



        if (metadataCursor == null) {

            return null;
@@ -195,8 +202,10 @@ public class BluetoothOppUtility { 
        }



        Uri path = null;

        Cursor metadataCursor = context.getContentResolver().query(uri, new String[]{

                BluetoothShare.URI}, null, null, null);

        Cursor metadataCursor = BluetoothMethodProxy.getInstance().contentResolverQuery(

                context.getContentResolver(), uri, new String[]{BluetoothShare.URI},

                null, null, null

        );

        if (metadataCursor != null) {

            try {

                if (metadataCursor.moveToFirst()) {
@@ -302,7 +311,8 @@ public class BluetoothOppUtility { 
    public static void updateVisibilityToHidden(Context context, Uri uri) {

        ContentValues updateValues = new ContentValues();

        updateValues.put(BluetoothShare.VISIBILITY, BluetoothShare.VISIBILITY_HIDDEN);

        context.getContentResolver().update(uri, updateValues, null, null);

        BluetoothMethodProxy.getInstance().contentResolverUpdate(context.getContentResolver(), uri,

                updateValues, null, null);

    }



    /**

android/app/tests/unit/src/com/android/bluetooth/opp/BluetoothOppTestUtils.java

+7 −0
Original line number Diff line number Diff line
@@ -19,6 +19,7 @@ package com.android.bluetooth.opp; 
import static org.mockito.ArgumentMatchers.anyInt;

import static org.mockito.ArgumentMatchers.anyString;

import static org.mockito.Mockito.doAnswer;

import static org.mockito.Mockito.doReturn;



import android.database.Cursor;
@@ -98,6 +99,12 @@ public class BluetoothOppTestUtils { 
                    mockCursorData -> mockCursorData.mColumnIndex == index

            ).findFirst().orElse(new CursorMockData("", -1, null)).mValue;

        }).when(cursor).getString(anyInt());



        doReturn(true).when(cursor).moveToFirst();

        doReturn(true).when(cursor).moveToLast();

        doReturn(true).when(cursor).moveToNext();

        doReturn(true).when(cursor).moveToPrevious();

        doReturn(true).when(cursor).moveToPosition(anyInt());

    }

}

android/app/tests/unit/src/com/android/bluetooth/opp/BluetoothOppTransferActivityTest.java

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

 * Copyright 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.DONT_KILL_APP;

import static android.service.pm.PackageProto.UserInfoProto.COMPONENT_ENABLED_STATE_DEFAULT;

import static android.service.pm.PackageProto.UserInfoProto.COMPONENT_ENABLED_STATE_ENABLED;



import static com.android.bluetooth.opp.BluetoothOppTestUtils.CursorMockData;

import static com.android.bluetooth.opp.BluetoothOppTransferActivity.DIALOG_RECEIVE_COMPLETE_FAIL;

import static com.android.bluetooth.opp.BluetoothOppTransferActivity.DIALOG_RECEIVE_COMPLETE_SUCCESS;

import static com.android.bluetooth.opp.BluetoothOppTransferActivity.DIALOG_RECEIVE_ONGOING;

import static com.android.bluetooth.opp.BluetoothOppTransferActivity.DIALOG_SEND_COMPLETE_FAIL;

import static com.android.bluetooth.opp.BluetoothOppTransferActivity.DIALOG_SEND_COMPLETE_SUCCESS;

import static com.android.bluetooth.opp.BluetoothOppTransferActivity.DIALOG_SEND_ONGOING;



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



import static org.mockito.Mockito.any;

import static org.mockito.Mockito.doReturn;

import static org.mockito.Mockito.eq;



import android.content.ComponentName;

import android.content.Context;

import android.content.Intent;

import android.database.Cursor;

import android.net.Uri;



import androidx.test.core.app.ActivityScenario;

import androidx.test.filters.MediumTest;

import androidx.test.platform.app.InstrumentationRegistry;

import androidx.test.runner.AndroidJUnit4;



import com.android.bluetooth.BluetoothMethodProxy;

import com.android.bluetooth.pbap.BluetoothPbapActivity;



import org.junit.After;

import org.junit.Before;

import org.junit.Test;

import org.junit.runner.RunWith;

import org.mockito.Mock;

import org.mockito.Mockito;

import org.mockito.MockitoAnnotations;

import org.mockito.Spy;



import java.util.ArrayList;

import java.util.List;



@MediumTest

@RunWith(AndroidJUnit4.class)

public class BluetoothOppTransferActivityTest {

  @Mock

  Cursor mCursor;

  @Spy

  BluetoothMethodProxy mBluetoothMethodProxy;



  List<CursorMockData> mCursorMockDataList;



  Intent mIntent;

  Context mTargetContext;



  @Before

  public void setUp() {

    MockitoAnnotations.initMocks(this);

    mBluetoothMethodProxy = Mockito.spy(BluetoothMethodProxy.getInstance());

    BluetoothMethodProxy.setInstanceForTesting(mBluetoothMethodProxy);



    Uri dataUrl = Uri.parse("content://com.android.bluetooth.opp.test/random");



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



    mIntent = new Intent();

    mIntent.setClass(mTargetContext, BluetoothOppTransferActivity.class);

    mIntent.setData(dataUrl);



    doReturn(mCursor).when(mBluetoothMethodProxy).contentResolverQuery(any(), eq(dataUrl),

            eq(null), eq(null),

            eq(null), eq(null));



    doReturn(1).when(mBluetoothMethodProxy).contentResolverUpdate(any(), eq(dataUrl),

            any(), eq(null), eq(null));



    int idValue = 1234;

    Long timestampValue = 123456789L;

    String destinationValue = "AA:BB:CC:00:11:22";

    String fileTypeValue = "text/plain";



    mCursorMockDataList = new ArrayList<>(List.of(

            new CursorMockData(BluetoothShare._ID, 0, idValue),

            new CursorMockData(BluetoothShare.MIMETYPE, 5, fileTypeValue),

            new CursorMockData(BluetoothShare.TIMESTAMP, 6, timestampValue),

            new CursorMockData(BluetoothShare.DESTINATION, 7, destinationValue),

            new CursorMockData(BluetoothShare._DATA, 8, null),

            new CursorMockData(BluetoothShare.FILENAME_HINT, 9, null),

            new CursorMockData(BluetoothShare.URI, 10, "content://textfile.txt"),

            new CursorMockData(BluetoothShare.USER_CONFIRMATION, 11,

                    BluetoothShare.USER_CONFIRMATION_HANDOVER_CONFIRMED)

    ));



    enableActivity(true);

  }



  @After

  public void tearDown() {

    BluetoothMethodProxy.setInstanceForTesting(null);

    enableActivity(false);

  }



  @Test

  public void onCreate_showSendOnGoingDialog() {

    mCursorMockDataList.add(

            new CursorMockData(BluetoothShare.STATUS, 1, BluetoothShare.STATUS_PENDING));

    mCursorMockDataList.add(

            new CursorMockData(BluetoothShare.DIRECTION, 2, BluetoothShare.DIRECTION_OUTBOUND)

    );

    mCursorMockDataList.add(new CursorMockData(BluetoothShare.TOTAL_BYTES, 3, 100));

    mCursorMockDataList.add(new CursorMockData(BluetoothShare.CURRENT_BYTES, 4, 0));

    BluetoothOppTestUtils.setUpMockCursor(mCursor, mCursorMockDataList);



    ActivityScenario<BluetoothOppTransferActivity> activityScenario = ActivityScenario.launch(

            mIntent);



    activityScenario.onActivity(activity -> {

      assertThat(activity.mWhichDialog).isEqualTo(DIALOG_SEND_ONGOING);

    });

  }



  @Test

  public void onCreate_showSendCompleteSuccessDialog() {

    mCursorMockDataList.add(

            new CursorMockData(BluetoothShare.STATUS, 1, BluetoothShare.STATUS_SUCCESS)

    );

    mCursorMockDataList.add(

            new CursorMockData(BluetoothShare.DIRECTION, 2, BluetoothShare.DIRECTION_OUTBOUND)

    );

    mCursorMockDataList.add(new CursorMockData(BluetoothShare.TOTAL_BYTES, 3, 100));

    mCursorMockDataList.add(

            new CursorMockData(BluetoothShare.CURRENT_BYTES, 4, 100));

    BluetoothOppTestUtils.setUpMockCursor(mCursor, mCursorMockDataList);



    ActivityScenario<BluetoothOppTransferActivity> activityScenario = ActivityScenario.launch(

            mIntent);



    activityScenario.onActivity(activity -> {

      assertThat(activity.mWhichDialog).isEqualTo(DIALOG_SEND_COMPLETE_SUCCESS);

    });

  }



  @Test

  public void onCreate_showSendCompleteFailDialog() {

    mCursorMockDataList.add(

            new CursorMockData(BluetoothShare.STATUS, 1, BluetoothShare.STATUS_FORBIDDEN));

    mCursorMockDataList.add(

            new CursorMockData(BluetoothShare.DIRECTION, 2, BluetoothShare.DIRECTION_OUTBOUND)

    );

    mCursorMockDataList.add(new CursorMockData(BluetoothShare.TOTAL_BYTES, 3, 100));

    mCursorMockDataList.add(new CursorMockData(BluetoothShare.CURRENT_BYTES, 4, 42));

    BluetoothOppTestUtils.setUpMockCursor(mCursor, mCursorMockDataList);



    ActivityScenario<BluetoothOppTransferActivity> activityScenario = ActivityScenario.launch(

            mIntent);



    activityScenario.onActivity(

            activity -> assertThat(activity.mWhichDialog).isEqualTo(DIALOG_SEND_COMPLETE_FAIL));

  }



  @Test

  public void onCreate_showReceiveOnGoingDialog() {

    mCursorMockDataList.add(

            new CursorMockData(BluetoothShare.STATUS, 1, BluetoothShare.STATUS_PENDING));

    mCursorMockDataList.add(

            new CursorMockData(BluetoothShare.DIRECTION, 2, BluetoothShare.DIRECTION_INBOUND)

    );

    mCursorMockDataList.add(new CursorMockData(BluetoothShare.TOTAL_BYTES, 3, 100));

    mCursorMockDataList.add(new CursorMockData(BluetoothShare.CURRENT_BYTES, 4, 0));

    BluetoothOppTestUtils.setUpMockCursor(mCursor, mCursorMockDataList);



    ActivityScenario<BluetoothOppTransferActivity> activityScenario = ActivityScenario.launch(

            mIntent);



    activityScenario.onActivity(

            activity -> assertThat(activity.mWhichDialog).isEqualTo(DIALOG_RECEIVE_ONGOING));

  }



  @Test

  public void onCreate_showReceiveCompleteSuccessDialog() {

    mCursorMockDataList.add(

            new CursorMockData(BluetoothShare.STATUS, 1, BluetoothShare.STATUS_SUCCESS));

    mCursorMockDataList.add(

            new CursorMockData(BluetoothShare.DIRECTION, 2, BluetoothShare.DIRECTION_INBOUND)

    );

    mCursorMockDataList.add(new CursorMockData(BluetoothShare.TOTAL_BYTES, 3, 100));

    mCursorMockDataList.add(

            new CursorMockData(BluetoothShare.CURRENT_BYTES, 4, 100)

    );



    BluetoothOppTestUtils.setUpMockCursor(mCursor, mCursorMockDataList);

    ActivityScenario<BluetoothOppTransferActivity> activityScenario = ActivityScenario.launch(

            mIntent);



    activityScenario.onActivity(activity -> assertThat(activity.mWhichDialog).isEqualTo(

            DIALOG_RECEIVE_COMPLETE_SUCCESS));

  }



  @Test

  public void onCreate_showReceiveCompleteFailDialog() {

    mCursorMockDataList.add(

            new CursorMockData(BluetoothShare.STATUS, 1, BluetoothShare.STATUS_FORBIDDEN));

    mCursorMockDataList.add(

            new CursorMockData(BluetoothShare.DIRECTION, 2, BluetoothShare.DIRECTION_INBOUND)

    );

    mCursorMockDataList.add(new CursorMockData(BluetoothShare.TOTAL_BYTES, 3, 100));

    mCursorMockDataList.add(new CursorMockData(BluetoothShare.CURRENT_BYTES, 4, 42));



    BluetoothOppTestUtils.setUpMockCursor(mCursor, mCursorMockDataList);

    ActivityScenario<BluetoothOppTransferActivity> activityScenario = ActivityScenario.launch(

            mIntent);



    activityScenario.onActivity(activity -> assertThat(activity.mWhichDialog).isEqualTo(

            DIALOG_RECEIVE_COMPLETE_FAIL));

  }



  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);

  }

}