fix: 引入Settings的Module
This commit is contained in:
@@ -0,0 +1,165 @@
|
||||
/*
|
||||
* Copyright (C) 2021 The Android Open Source Project
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package com.android.settings.testutils;
|
||||
|
||||
import android.app.Activity;
|
||||
import android.app.Instrumentation;
|
||||
import android.content.Context;
|
||||
import android.graphics.Bitmap;
|
||||
import android.net.wifi.WifiManager;
|
||||
import android.os.Environment;
|
||||
import android.os.PowerManager;
|
||||
import android.os.StrictMode;
|
||||
import android.os.SystemClock;
|
||||
import android.util.Log;
|
||||
import android.view.View;
|
||||
|
||||
import androidx.test.platform.app.InstrumentationRegistry;
|
||||
|
||||
import java.io.BufferedReader;
|
||||
import java.io.File;
|
||||
import java.io.FileOutputStream;
|
||||
import java.io.InputStream;
|
||||
import java.io.InputStreamReader;
|
||||
import java.net.HttpURLConnection;
|
||||
import java.net.URL;
|
||||
|
||||
import javax.net.ssl.HttpsURLConnection;
|
||||
|
||||
public class CommonUtils {
|
||||
private static final String TAG = CommonUtils.class.getSimpleName();
|
||||
private static final Instrumentation sInstrumentation =
|
||||
InstrumentationRegistry.getInstrumentation();
|
||||
private static final WifiManager sWifiManager =
|
||||
(WifiManager) sInstrumentation.getTargetContext().getSystemService(
|
||||
Context.WIFI_SERVICE);
|
||||
private static final PowerManager sPowerManager =
|
||||
(PowerManager) sInstrumentation.getTargetContext().getSystemService(
|
||||
Context.POWER_SERVICE);
|
||||
|
||||
public static void takeScreenshot(Activity activity) {
|
||||
long now = System.currentTimeMillis();
|
||||
|
||||
try {
|
||||
// image naming and path to include sd card appending name you choose for file
|
||||
String mPath =
|
||||
Environment.getExternalStorageDirectory().toString() + "/" + now + ".jpg";
|
||||
Log.d(TAG, "screenshot path is " + mPath);
|
||||
|
||||
// create bitmap screen capture
|
||||
View v1 = activity.getWindow().getDecorView().getRootView();
|
||||
v1.setDrawingCacheEnabled(true);
|
||||
Bitmap bitmap = Bitmap.createBitmap(v1.getDrawingCache());
|
||||
v1.setDrawingCacheEnabled(false);
|
||||
|
||||
File imageFile = new File(mPath);
|
||||
|
||||
FileOutputStream outputStream = new FileOutputStream(imageFile);
|
||||
int quality = 100;
|
||||
bitmap.compress(Bitmap.CompressFormat.JPEG, quality, outputStream);
|
||||
outputStream.flush();
|
||||
outputStream.close();
|
||||
} catch (Throwable e) {
|
||||
// Several error may come out with file handling or DOM
|
||||
e.printStackTrace();
|
||||
}
|
||||
}
|
||||
|
||||
public static boolean connectToURL(URL url) {
|
||||
StrictMode.ThreadPolicy policy = new StrictMode.ThreadPolicy.Builder().permitAll().build();
|
||||
StrictMode.setThreadPolicy(policy);
|
||||
HttpURLConnection connection = null;
|
||||
try {
|
||||
connection = (HttpsURLConnection) url.openConnection();
|
||||
connection.setRequestMethod("GET");
|
||||
connection.setConnectTimeout(8000);
|
||||
connection.setReadTimeout(8000);
|
||||
connection.connect();
|
||||
if (HttpURLConnection.HTTP_OK == connection.getResponseCode()) {
|
||||
InputStream in = connection.getInputStream();
|
||||
BufferedReader reader = new BufferedReader(new InputStreamReader(in));
|
||||
StringBuilder response = new StringBuilder();
|
||||
String line;
|
||||
while (null != (line = reader.readLine())) {
|
||||
response.append(line);
|
||||
}
|
||||
Log.d(TAG, "Connection success! " + response.toString());
|
||||
return true;
|
||||
}
|
||||
} catch (Exception e) {
|
||||
Log.e(TAG, e.toString());
|
||||
e.printStackTrace();
|
||||
return false;
|
||||
} finally {
|
||||
if (null != connection) {
|
||||
connection.disconnect();
|
||||
}
|
||||
}
|
||||
Log.d(TAG, "End, return false.");
|
||||
return false;
|
||||
}
|
||||
|
||||
/**
|
||||
* Return a resource identifier for the given resource name in Settings app.
|
||||
*
|
||||
* @param name The name of the desired resource.
|
||||
* @return int The associated resource identifier. Returns 0 if no such resource was found. (0
|
||||
* is not a valid resource ID.)
|
||||
*/
|
||||
public static int getResId(String name) {
|
||||
return InstrumentationRegistry.getInstrumentation().getTargetContext().getResources()
|
||||
.getIdentifier(name, "id", Constants.SETTINGS_PACKAGE_NAME);
|
||||
}
|
||||
|
||||
public static void reopenScreen() {
|
||||
sPowerManager.goToSleep(SystemClock.uptimeMillis());
|
||||
// According to local test, we need to sleep to wait it fully processed.
|
||||
// 1000 ms is a good value to sleep, otherwise it might cause keyDispatchingTimedOut.
|
||||
try {
|
||||
Thread.sleep(1000);
|
||||
} catch (InterruptedException e) {
|
||||
e.printStackTrace();
|
||||
}
|
||||
UiUtils.waitUntilCondition(1000, () -> !sPowerManager.isInteractive());
|
||||
sPowerManager.wakeUp(SystemClock.uptimeMillis());
|
||||
UiUtils.waitUntilCondition(1000, () -> sPowerManager.isInteractive());
|
||||
|
||||
// After power on screen, need to unlock and goto home page.
|
||||
sPowerManager.wakeUp(1000, PowerManager.WAKE_REASON_POWER_BUTTON, "Wakeup");
|
||||
}
|
||||
|
||||
/**
|
||||
* Sets wifi status to given enable / disable via ADB command.
|
||||
*/
|
||||
public static void set_wifi_enabled(boolean enable) {
|
||||
final int timeoutMsec = 10000;
|
||||
Log.d(TAG, "Set wifi status to " + enable);
|
||||
if (sWifiManager.isWifiEnabled() != enable) {
|
||||
AdbUtils.shell("svc wifi " + (enable ? "enable" : "disable"));
|
||||
if (!UiUtils.waitUntilCondition(timeoutMsec,
|
||||
() -> sWifiManager.isWifiEnabled() == enable)) {
|
||||
Log.e(TAG, "Cannot set wifi to " + (enable ? "enabl" : "disable") + ", timeout "
|
||||
+ timeoutMsec + " (ms).");
|
||||
Log.e(TAG, "See logcat for more information.");
|
||||
}
|
||||
Log.d(TAG, "After configuration wifi status = " + sWifiManager.isWifiEnabled());
|
||||
} else {
|
||||
Log.d(TAG, "Wifi is enable is already " + enable + ", no need to change.");
|
||||
}
|
||||
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user