1 /* 2 * Copyright (C) 2018 The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17 package com.android.settings.testutils.shadow; 18 19 import android.content.Context; 20 21 import androidx.annotation.NonNull; 22 23 import com.android.settings.bluetooth.Utils; 24 import com.android.settingslib.bluetooth.CachedBluetoothDevice; 25 import com.android.settingslib.bluetooth.LocalBluetoothManager; 26 27 import org.robolectric.annotation.Implementation; 28 import org.robolectric.annotation.Implements; 29 import org.robolectric.annotation.Resetter; 30 31 import java.util.HashMap; 32 import java.util.Map; 33 34 /** Robolectric shadow for the bluetooth utils. */ 35 @Implements(Utils.class) 36 public class ShadowBluetoothUtils { 37 38 public static LocalBluetoothManager sLocalBluetoothManager; 39 private static final Map<CachedBluetoothDevice, Boolean> sLeAudioState = new HashMap<>(); 40 41 @Implementation getLocalBtManager(Context context)42 protected static LocalBluetoothManager getLocalBtManager(Context context) { 43 return sLocalBluetoothManager; 44 } 45 46 /** Sets le audio state for the device. */ 47 @Implementation setLeAudioEnabled( @onNull LocalBluetoothManager manager, @NonNull CachedBluetoothDevice cachedDevice, boolean enable)48 public static void setLeAudioEnabled( 49 @NonNull LocalBluetoothManager manager, 50 @NonNull CachedBluetoothDevice cachedDevice, 51 boolean enable) { 52 sLeAudioState.put(cachedDevice, enable); 53 } 54 55 /** Checks whether le audio is enabled for the device. */ isLeAudioEnabled(@onNull CachedBluetoothDevice cachedDevice)56 public static boolean isLeAudioEnabled(@NonNull CachedBluetoothDevice cachedDevice) { 57 if (sLeAudioState.containsKey(cachedDevice)) { 58 return sLeAudioState.get(cachedDevice); 59 } 60 return false; 61 } 62 63 /** Resets the local bluetooth manager to null. */ 64 @Resetter reset()65 public static void reset() { 66 sLocalBluetoothManager = null; 67 } 68 } 69