From b05854810a19da0b698b1c9fa6a9531acdcb8ffb Mon Sep 17 00:00:00 2001
From: Ang Li <angli@google.com>
Date: Wed, 12 Jun 2019 17:34:15 -0700
Subject: Handle Wi-Fi permission requirements in Q. (#112)

* Adopt shell permission in Wi-Fi snippet when it's Q.
* Fix ClassNotFound error caused by androidx

Tested locally and confirmed that the change works.
---
 .../com/google/android/mobly/snippet/bundled/WifiManagerSnippet.java | 5 +++++
 1 file changed, 5 insertions(+)

(limited to 'src/main/java/com')

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 0008c13..3c46708 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
@@ -62,6 +62,11 @@ public class WifiManagerSnippet implements Snippet {
         mWifiManager =
                 (WifiManager)
                         mContext.getApplicationContext().getSystemService(Context.WIFI_SERVICE);
+        if (Build.VERSION.SDK_INT >= 29) {
+            InstrumentationRegistry.getInstrumentation()
+                    .getUiAutomation()
+                    .adoptShellPermissionIdentity();
+        }
     }
 
     @Rpc(
-- 
cgit v1.2.3