Add an API to check if 5G Wi-Fi is supported. (#107)

1 files changed, 11 insertions, 0 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 3d96e4d..d848eb9 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
@@ -26,6 +26,7 @@ import android.net.wifi.WifiInfo;
 import android.net.wifi.WifiManager;
 import android.os.Build;
 import androidx.annotation.Nullable;
+import androidx.annotation.RequiresApi;
 import androidx.test.platform.app.InstrumentationRegistry;
 import com.google.android.mobly.snippet.Snippet;
 import com.google.android.mobly.snippet.bundled.utils.JsonDeserializer;
@@ -321,6 +322,16 @@ public class WifiManagerSnippet implements Snippet {
         return (boolean) Utils.invokeByReflection(mWifiManager, "isWifiApEnabled");
     }
 
+    @RequiresApi(Build.VERSION_CODES.LOLLIPOP)
+    @RpcMinSdk(Build.VERSION_CODES.LOLLIPOP)
+    @Rpc(
+            description =
+                    "Check whether this device supports 5 GHz band Wi-Fi. "
+                            + "Turn on Wi-Fi before calling.")
+    public boolean wifiIs5GHzBandSupported() {
+        return mWifiManager.is5GHzBandSupported();
+    }
+
     /**
      * Enable Wi-Fi Soft AP (hotspot).
      *