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.systemui.volume;
18 
19 import androidx.mediarouter.media.MediaRouteSelector;
20 import androidx.mediarouter.media.MediaRouter;
21 
22 import java.util.List;
23 
24 /**
25  * Wrapper for final class MediaRouter, for testing.
26  */
27 public class MediaRouterWrapper {
28 
29     private final MediaRouter mRouter;
30 
MediaRouterWrapper(MediaRouter router)31     public MediaRouterWrapper(MediaRouter router)
32     {
33         mRouter = router;
34     }
35 
addCallback(MediaRouteSelector selector, MediaRouter.Callback callback, int flags)36     public void addCallback(MediaRouteSelector selector, MediaRouter.Callback callback, int flags) {
37         mRouter.addCallback(selector, callback, flags);
38     }
39 
removeCallback(MediaRouter.Callback callback)40     public void removeCallback(MediaRouter.Callback callback) {
41         mRouter.removeCallback(callback);
42     }
43 
unselect(int reason)44     public void unselect(int reason) {
45         mRouter.unselect(reason);
46     }
47 
getRoutes()48     public List<MediaRouter.RouteInfo> getRoutes() {
49         return mRouter.getRoutes();
50     }
51 }