1 /*
2  * Copyright (C) 2016 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.server.wifi;
18 
19 import android.content.BroadcastReceiver;
20 import android.content.Context;
21 import android.content.Intent;
22 import android.net.ConnectivityManager;
23 import android.net.NetworkInfo;
24 import android.net.wifi.WifiInfo;
25 import android.net.wifi.WifiManager;
26 import android.os.PowerManager;
27 
28 import java.util.ArrayList;
29 import java.util.BitSet;
30 
31 /**
32  * Utils for wifi tests.
33  */
34 public class TestUtil {
35     /**
36      * Send {@link WifiManager#NETWORK_STATE_CHANGED_ACTION} broadcast.
37      */
sendNetworkStateChanged(BroadcastReceiver broadcastReceiver, Context context, NetworkInfo.DetailedState detailedState)38     public static void sendNetworkStateChanged(BroadcastReceiver broadcastReceiver,
39             Context context, NetworkInfo.DetailedState detailedState) {
40         Intent intent = new Intent(WifiManager.NETWORK_STATE_CHANGED_ACTION);
41         NetworkInfo networkInfo = new NetworkInfo(0, 0, "", "");
42         networkInfo.setDetailedState(detailedState, "", "");
43         intent.putExtra(WifiManager.EXTRA_NETWORK_INFO, networkInfo);
44         broadcastReceiver.onReceive(context, intent);
45     }
46 
47     /**
48      * Send {@link WifiManager#NETWORK_STATE_CHANGED_ACTION} broadcast.
49      */
sendNetworkStateChanged(BroadcastReceiver broadcastReceiver, Context context, NetworkInfo nwInfo, WifiInfo wifiInfo)50     public static void sendNetworkStateChanged(BroadcastReceiver broadcastReceiver,
51             Context context, NetworkInfo nwInfo, WifiInfo wifiInfo) {
52         Intent intent = new Intent(WifiManager.NETWORK_STATE_CHANGED_ACTION);
53         intent.putExtra(WifiManager.EXTRA_NETWORK_INFO, nwInfo);
54         intent.putExtra(WifiManager.EXTRA_WIFI_INFO, wifiInfo);
55         broadcastReceiver.onReceive(context, intent);
56     }
57 
58     /**
59      * Send {@link WifiManager#SCAN_RESULTS_AVAILABLE_ACTION} broadcast.
60      */
sendScanResultsAvailable(BroadcastReceiver broadcastReceiver, Context context)61     public static void sendScanResultsAvailable(BroadcastReceiver broadcastReceiver,
62             Context context) {
63         Intent intent = new Intent(WifiManager.SCAN_RESULTS_AVAILABLE_ACTION);
64         broadcastReceiver.onReceive(context, intent);
65     }
66 
67     /**
68      * Send {@link WifiManager#WIFI_STATE_CHANGED} broadcast.
69      */
sendWifiStateChanged(BroadcastReceiver broadcastReceiver, Context context, int wifiState)70     public static void sendWifiStateChanged(BroadcastReceiver broadcastReceiver,
71             Context context, int wifiState) {
72         Intent intent = new Intent(WifiManager.WIFI_STATE_CHANGED_ACTION);
73         intent.putExtra(WifiManager.EXTRA_WIFI_STATE, wifiState);
74         broadcastReceiver.onReceive(context, intent);
75     }
76 
77     /**
78      * Send {@link WifiManager#WIFI_AP_STATE_CHANGED} broadcast.
79      */
sendWifiApStateChanged(BroadcastReceiver broadcastReceiver, Context context, int apState, int previousState, int error, String ifaceName, int mode)80     public static void sendWifiApStateChanged(BroadcastReceiver broadcastReceiver,
81             Context context, int apState, int previousState, int error, String ifaceName,
82             int mode) {
83         Intent intent = new Intent(WifiManager.WIFI_AP_STATE_CHANGED_ACTION);
84         intent.putExtra(WifiManager.EXTRA_WIFI_AP_STATE, apState);
85         intent.putExtra(WifiManager.EXTRA_PREVIOUS_WIFI_AP_STATE, previousState);
86         if (apState == WifiManager.WIFI_AP_STATE_FAILED) {
87             // only set reason number when softAP start failed
88             intent.putExtra(WifiManager.EXTRA_WIFI_AP_FAILURE_REASON, error);
89         }
90         intent.putExtra(WifiManager.EXTRA_WIFI_AP_INTERFACE_NAME, ifaceName);
91         intent.putExtra(WifiManager.EXTRA_WIFI_AP_MODE, mode);
92         broadcastReceiver.onReceive(context, intent);
93     }
94 
95     /**
96      * Send {@link ConnectivityManager#ACTION_TETHER_STATE_CHANGED} broadcast.
97      */
sendTetherStateChanged(BroadcastReceiver broadcastReceiver, Context context, ArrayList<String> available, ArrayList<String> active)98     public static void sendTetherStateChanged(BroadcastReceiver broadcastReceiver,
99             Context context, ArrayList<String> available, ArrayList<String> active) {
100         Intent intent = new Intent(ConnectivityManager.ACTION_TETHER_STATE_CHANGED);
101         intent.putExtra(ConnectivityManager.EXTRA_AVAILABLE_TETHER, available);
102         intent.putExtra(ConnectivityManager.EXTRA_ACTIVE_TETHER, active);
103         broadcastReceiver.onReceive(context, intent);
104     }
105 
sendIdleModeChanged(BroadcastReceiver broadcastReceiver, Context context)106     public static void sendIdleModeChanged(BroadcastReceiver broadcastReceiver, Context context) {
107         Intent intent = new Intent(PowerManager.ACTION_DEVICE_IDLE_MODE_CHANGED);
108         broadcastReceiver.onReceive(context, intent);
109     }
110 
111     /**
112      * Create a new capability BitSet containing the provided capabilities.
113      *
114      * @param capabilities One or more WifiManager.WIFI_FEATURE_* capabilities
115      */
createCapabilityBitset(int... capabilities)116     public static BitSet createCapabilityBitset(int... capabilities) {
117         BitSet capabilityBitset = new BitSet();
118         for (int capability : capabilities) {
119             capabilityBitset.set(capability);
120         }
121         return capabilityBitset;
122     }
123 
124     /**
125      * Add additional capabilities to the provided BitSet.
126      *
127      * @param bitset BitSet that the capabilities should be added to
128      * @param capabilities One or more WifiManager.WIFI_FEATURE_* capabilities
129      */
addCapabilitiesToBitset(BitSet bitset, int... capabilities)130     public static BitSet addCapabilitiesToBitset(BitSet bitset, int... capabilities) {
131         // Clone to avoid modifying the input BitSet
132         BitSet clonedBitset = (BitSet) bitset.clone();
133         for (int capability : capabilities) {
134             clonedBitset.set(capability);
135         }
136         return clonedBitset;
137     }
138 
139     /**
140      * Combine several BitSets using an OR operation.
141      *
142      * @param bitsets BitSets that should be combined
143      */
combineBitsets(BitSet... bitsets)144     public static BitSet combineBitsets(BitSet... bitsets) {
145         BitSet combinedBitset = new BitSet();
146         for (BitSet bitset : bitsets) {
147             combinedBitset.or(bitset);
148         }
149         return combinedBitset;
150     }
151 }
152