Home | History | Annotate | Download | only in shadow 
      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 static android.media.AudioManager.STREAM_ACCESSIBILITY;
     20 import static android.media.AudioManager.STREAM_ALARM;
     21 import static android.media.AudioManager.STREAM_MUSIC;
     22 import static android.media.AudioManager.STREAM_NOTIFICATION;
     23 import static android.media.AudioManager.STREAM_RING;
     24 import static android.media.AudioManager.STREAM_SYSTEM;
     25 import static android.media.AudioManager.STREAM_VOICE_CALL;
     26 import static android.media.AudioManager.STREAM_DTMF;
     27 
     28 import static org.robolectric.RuntimeEnvironment.application;
     29 
     30 import android.media.AudioDeviceCallback;
     31 import android.media.AudioManager;
     32 import android.os.Handler;
     33 
     34 import org.robolectric.annotation.Implementation;
     35 import org.robolectric.annotation.Implements;
     36 import org.robolectric.annotation.Resetter;
     37 import org.robolectric.shadow.api.Shadow;
     38 
     39 import java.util.ArrayList;
     40 
     41 @Implements(value = AudioManager.class, inheritImplementationMethods = true)
     42 public class ShadowAudioManager extends org.robolectric.shadows.ShadowAudioManager {
     43     private int mRingerMode;
     44     private int mDeviceCodes;
     45     private boolean mMusicActiveRemotely;
     46     private boolean mBluetoothScoOn;
     47     private ArrayList<AudioDeviceCallback> mDeviceCallbacks = new ArrayList();
     48 
     49     @Implementation
     50     private int getRingerModeInternal() {
     51         return mRingerMode;
     52     }
     53 
     54     public static ShadowAudioManager getShadow() {
     55         return Shadow.extract(application.getSystemService(AudioManager.class));
     56     }
     57 
     58     public void setRingerModeInternal(int mode) {
     59         mRingerMode = mode;
     60     }
     61 
     62     @Implementation
     63     public void registerAudioDeviceCallback(AudioDeviceCallback callback, Handler handler) {
     64         mDeviceCallbacks.add(callback);
     65     }
     66 
     67     @Implementation
     68     public void unregisterAudioDeviceCallback(AudioDeviceCallback callback) {
     69         if (mDeviceCallbacks.contains(callback)) {
     70             mDeviceCallbacks.remove(callback);
     71         }
     72     }
     73 
     74     public void setMusicActiveRemotely(boolean flag) {
     75         mMusicActiveRemotely = flag;
     76     }
     77 
     78     @Implementation
     79     public boolean isMusicActiveRemotely() {
     80         return mMusicActiveRemotely;
     81     }
     82 
     83     public void setOutputDevice(int deviceCodes) {
     84         mDeviceCodes = deviceCodes;
     85     }
     86 
     87     @Implementation
     88     public int getDevicesForStream(int streamType) {
     89         switch (streamType) {
     90             case STREAM_VOICE_CALL:
     91             case STREAM_SYSTEM:
     92             case STREAM_RING:
     93             case STREAM_MUSIC:
     94             case STREAM_ALARM:
     95             case STREAM_NOTIFICATION:
     96             case STREAM_DTMF:
     97             case STREAM_ACCESSIBILITY:
     98                 return mDeviceCodes;
     99             default:
    100                 return 0;
    101         }
    102     }
    103 
    104     @Resetter
    105     public void reset() {
    106         mDeviceCallbacks.clear();
    107     }
    108 
    109     public void setBluetoothScoOn(boolean bluetoothScoOn) {
    110         mBluetoothScoOn = bluetoothScoOn;
    111     }
    112 
    113     @Implementation
    114     public boolean isBluetoothScoOn() { return mBluetoothScoOn; }
    115 }
    116