Add an rpc to check if wifi is enabled. (#65)

1 files changed, 11 insertions, 3 deletions

diff --git a/src/main/java/com/google/android/mobly/snippet/bundled/WifiManagerSnippet.java b/src/main/java/com/google/android/mobly/snippet/bundled/WifiManagerSnippet.java
index 1911b9f..13a39cd 100644
--- a/src/main/java/com/google/android/mobly/snippet/bundled/WifiManagerSnippet.java
+++ b/src/main/java/com/google/android/mobly/snippet/bundled/WifiManagerSnippet.java
@@ -59,14 +59,17 @@ public class WifiManagerSnippet implements Snippet {
         mWifiManager = (WifiManager) mContext.getSystemService(Context.WIFI_SERVICE);
     }
 
-    @Rpc(description = "Clears all configured networks. This will only work if all configured "
-        + "networks were added through this MBS instance")
+    @Rpc(
+        description =
+                "Clears all configured networks. This will only work if all configured "
+                        + "networks were added through this MBS instance"
+    )
     public void wifiClearConfiguredNetworks() throws WifiManagerSnippetException {
         List<WifiConfiguration> unremovedConfigs = mWifiManager.getConfiguredNetworks();
         List<WifiConfiguration> failedConfigs = new ArrayList<>();
         if (unremovedConfigs == null) {
             throw new WifiManagerSnippetException(
-                "Failed to get a list of configured networks. Is wifi disabled?");
+                    "Failed to get a list of configured networks. Is wifi disabled?");
         }
         for (WifiConfiguration config : unremovedConfigs) {
             if (!mWifiManager.removeNetwork(config.networkId)) {
@@ -100,6 +103,11 @@ public class WifiManagerSnippet implements Snippet {
         }
     }
 
+    @Rpc(description = "Checks if Wi-Fi is enabled.")
+    public boolean wifiIsEnabled() {
+        return mWifiManager.getWifiState() == WifiManager.WIFI_STATE_ENABLED;
+    }
+
     @Rpc(description = "Trigger Wi-Fi scan.")
     public void wifiStartScan() throws WifiManagerSnippetException {
         if (!mWifiManager.startScan()) {