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

Commit b23d36b4 authored by Svet Ganov's avatar Svet Ganov
Browse files

Ensure mock context returns an attribution source. 

bug:184050679

Test: atest BluetoothInstrumentationTests

Change-Id: Id6726b2b50000eecb887b33eb783d4743eef67e9
parent 8851741c

tests/unit/src/com/android/bluetooth/btservice/AdapterServiceTest.java

+5 −0
Original line number Diff line number Diff line
@@ -25,6 +25,7 @@ import android.app.admin.DevicePolicyManager; 
import android.bluetooth.BluetoothAdapter;

import android.bluetooth.BluetoothDevice;

import android.bluetooth.IBluetoothCallback;

import android.content.AttributionSource;

import android.content.Context;

import android.content.pm.ApplicationInfo;

import android.content.pm.PackageManager;
@@ -97,6 +98,9 @@ public class AdapterServiceTest { 
    private static final int NATIVE_INIT_MS = 8000;

    private static final int NATIVE_DISABLE_MS = 1000;



    private final AttributionSource mAttributionSource = new AttributionSource.Builder(

            Process.myUid()).build();



    private PowerManager mPowerManager;

    private PackageManager mMockPackageManager;

    private MockContentResolver mMockContentResolver;
@@ -152,6 +156,7 @@ public class AdapterServiceTest { 
        when(mMockContext.getSystemService(Context.POWER_SERVICE)).thenReturn(mPowerManager);

        when(mMockContext.getSystemService(Context.ALARM_SERVICE)).thenReturn(mMockAlarmManager);

        when(mMockContext.getSystemService(Context.AUDIO_SERVICE)).thenReturn(mAudioManager);

        when(mMockContext.getAttributionSource()).thenReturn(mAttributionSource);

        doAnswer(invocation -> {

            Object[] args = invocation.getArguments();

            return InstrumentationRegistry.getTargetContext().getDatabasePath((String) args[0]);