1 /*
2  * Copyright (C) 2019 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 org.robolectric.shadows.ShadowMediaPlayer.State.INITIALIZED;
20 
21 import android.content.Context;
22 import android.media.MediaPlayer;
23 import android.net.Uri;
24 
25 import org.robolectric.annotation.Implementation;
26 import org.robolectric.annotation.Implements;
27 import org.robolectric.shadow.api.Shadow;
28 import org.robolectric.shadows.ShadowMediaPlayer;
29 import org.robolectric.shadows.util.DataSource;
30 
31 @Implements(MediaPlayer.class)
32 public class ShadowSettingsMediaPlayer extends ShadowMediaPlayer {
33 
34     @Implementation
create(Context context, Uri uri)35     public static MediaPlayer create(Context context, Uri uri) {
36         final DataSource ds = DataSource.toDataSource(context, uri);
37         addMediaInfo(ds, new ShadowMediaPlayer.MediaInfo());
38 
39         final MediaPlayer mp = new MediaPlayer();
40         final ShadowMediaPlayer shadow = Shadow.extract(mp);
41         try {
42             shadow.setDataSource(ds);
43             shadow.setState(INITIALIZED);
44             mp.prepare();
45         } catch (Exception e) {
46             return null;
47         }
48 
49         return mp;
50     }
51 }
52