1 files changed, 31 insertions, 0 deletions
diff --git a/app/src/androidTest/legacy/ConnectionManager.java b/app/src/androidTest/legacy/ConnectionManager.java
new file mode 100644
index 00000000..6fb7d8a9
--- /dev/null
+++ b/app/src/androidTest/legacy/ConnectionManager.java
@@ -0,0 +1,31 @@
+package se.leap.bitmaskclient.test;
+
+import android.content.*;
+import android.net.*;
+import android.net.wifi.*;
+
+import java.lang.reflect.*;
+
+public class ConnectionManager {
+    static void setMobileDataEnabled(boolean enabled, Context context) {
+        final ConnectivityManager conman = (ConnectivityManager) context.getSystemService(Context.CONNECTIVITY_SERVICE);
+        Method[] methods = conman.getClass().getMethods();
+        for (Method method : methods) {
+            if (method.getName().equals("setMobileDataEnabled")) {
+                method.setAccessible(true);
+                try {
+                    method.invoke(conman, enabled);
+                } catch (InvocationTargetException e) {
+                    //e.printStackTrace();
+                } catch (IllegalAccessException e) {
+                    //e.printStackTrace();
+                }
+            }
+        }
+    }
+
+    static void setWifiEnabled(boolean enabled, Context context) {
+        WifiManager wifiManager = (WifiManager) context.getSystemService(Context.WIFI_SERVICE);
+        wifiManager.setWifiEnabled(enabled);
+    }
+}