LineageOS/android_packages_modules_Connectivity
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge "Add CTS for BatteryStatsManager new API"

Browse Source
This commit is contained in:
Lorenzo Colitti
2021-06-24 02:26:53 +00:00
committed by Gerrit Code Review
parent 8691f3c286 f52a30b605
commit 7325c86c26
1 changed files with 195 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

195
tests/cts/net/src/android/net/cts/BatteryStatsManagerTest.java Normal file
View File

@@ -0,0 +1,195 @@
/*
* 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 android.net.cts;
import static android.Manifest.permission.UPDATE_DEVICE_STATS;
import static androidx.test.InstrumentationRegistry.getContext;
import static com.android.compatibility.common.util.SystemUtil.runShellCommand;
import static com.android.testutils.MiscAsserts.assertThrows;
import static com.android.testutils.TestPermissionUtil.runAsShell;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.fail;
import android.content.Context;
import android.net.ConnectivityManager;
import android.net.Network;
import android.net.cts.util.CtsNetUtils;
import android.os.BatteryStatsManager;
import android.os.Build;
import android.os.connectivity.CellularBatteryStats;
import android.os.connectivity.WifiBatteryStats;
import android.util.Log;
import androidx.test.runner.AndroidJUnit4;
import com.android.testutils.DevSdkIgnoreRule;
import com.android.testutils.SkipPresubmit;
import org.junit.Before;
import org.junit.Rule;
import org.junit.Test;
import org.junit.runner.RunWith;
import java.io.IOException;
import java.net.HttpURLConnection;
import java.net.URL;
import java.util.function.Predicate;
import java.util.function.Supplier;
/**
* Test for BatteryStatsManager.
*/
@RunWith(AndroidJUnit4.class)
public class BatteryStatsManagerTest{
@Rule
public final DevSdkIgnoreRule ignoreRule = new DevSdkIgnoreRule();
private static final String TAG = BatteryStatsManagerTest.class.getSimpleName();
private static final String TEST_URL = "https://connectivitycheck.gstatic.com/generate_204";
// This value should be the same as BatteryStatsManager.BATTERY_STATUS_DISCHARGING.
// TODO: Use the constant once it's available in all branches
private static final int BATTERY_STATUS_DISCHARGING = 3;
private Context mContext;
private BatteryStatsManager mBsm;
private ConnectivityManager mCm;
private CtsNetUtils mCtsNetUtils;
@Before
public void setUp() throws Exception {
mContext = getContext();
mBsm = mContext.getSystemService(BatteryStatsManager.class);
mCm = mContext.getSystemService(ConnectivityManager.class);
mCtsNetUtils = new CtsNetUtils(mContext);
}
@Test
@SkipPresubmit(reason = "Virtual hardware does not support wifi battery stats")
public void testReportNetworkInterfaceForTransports() throws Exception {
try {
final Network cellNetwork = mCtsNetUtils.connectToCell();
final URL url = new URL(TEST_URL);
// Make sure wifi is disabled.
mCtsNetUtils.ensureWifiDisconnected(null /* wifiNetworkToCheck */);
// Simulate the device being unplugged from charging.
executeShellCommand("dumpsys battery unplug");
executeShellCommand("dumpsys battery set status " + BATTERY_STATUS_DISCHARGING);
executeShellCommand("dumpsys batterystats enable pretend-screen-off");
// Get cellular battery stats
CellularBatteryStats cellularStatsBefore = runAsShell(UPDATE_DEVICE_STATS,
mBsm::getCellularBatteryStats);
// Generate traffic on cellular network.
generateNetworkTraffic(cellNetwork, url);
// The mobile battery stats are updated when a network stops being the default network.
// ConnectivityService will call BatteryStatsManager.reportMobileRadioPowerState when
// removing data activity tracking.
final Network wifiNetwork = mCtsNetUtils.ensureWifiConnected();
// Check cellular battery stats are updated.
runAsShell(UPDATE_DEVICE_STATS,
() -> assertStatsEventually(mBsm::getCellularBatteryStats,
cellularStatsAfter -> cellularBatteryStatsIncreased(
cellularStatsBefore, cellularStatsAfter)));
WifiBatteryStats wifiStatsBefore = runAsShell(UPDATE_DEVICE_STATS,
mBsm::getWifiBatteryStats);
// Generate traffic on wifi network.
generateNetworkTraffic(wifiNetwork, url);
// Wifi battery stats are updated when wifi on.
mCtsNetUtils.toggleWifi();
// Check wifi battery stats are updated.
runAsShell(UPDATE_DEVICE_STATS,
() -> assertStatsEventually(mBsm::getWifiBatteryStats,
wifiStatsAfter -> wifiBatteryStatsIncreased(wifiStatsBefore,
wifiStatsAfter)));
} finally {
// Reset battery settings.
executeShellCommand("dumpsys battery reset");
executeShellCommand("dumpsys batterystats disable pretend-screen-off");
}
}
@DevSdkIgnoreRule.IgnoreUpTo(Build.VERSION_CODES.R)
@Test
public void testReportNetworkInterfaceForTransports_throwsSecurityException()
throws Exception {
Network wifiNetwork = mCtsNetUtils.ensureWifiConnected();
final String iface = mCm.getLinkProperties(wifiNetwork).getInterfaceName();
final int[] transportType = mCm.getNetworkCapabilities(wifiNetwork).getTransportTypes();
assertThrows(SecurityException.class,
() -> mBsm.reportNetworkInterfaceForTransports(iface, transportType));
}
private void generateNetworkTraffic(Network network, URL url) throws IOException {
HttpURLConnection connection = null;
try {
connection = (HttpURLConnection) network.openConnection(url);
assertEquals(204, connection.getResponseCode());
} catch (IOException e) {
Log.e(TAG, "Generate traffic failed with exception " + e);
} finally {
if (connection != null) {
connection.disconnect();
}
}
}
private static <T> void assertStatsEventually(Supplier<T> statsGetter,
Predicate<T> statsChecker) throws Exception {
// Wait for updating mobile/wifi stats, and check stats every 10ms.
final int maxTries = 1000;
T result = null;
for (int i = 1; i <= maxTries; i++) {
result = statsGetter.get();
if (statsChecker.test(result)) return;
Thread.sleep(10);
}
final String stats = result instanceof CellularBatteryStats
? "Cellular" : "Wifi";
fail(stats + " battery stats did not increase.");
}
private static boolean cellularBatteryStatsIncreased(CellularBatteryStats before,
CellularBatteryStats after) {
return (after.getNumBytesTx() > before.getNumBytesTx())
&& (after.getNumBytesRx() > before.getNumBytesRx())
&& (after.getNumPacketsTx() > before.getNumPacketsTx())
&& (after.getNumPacketsRx() > before.getNumPacketsRx());
}
private static boolean wifiBatteryStatsIncreased(WifiBatteryStats before,
WifiBatteryStats after) {
return (after.getNumBytesTx() > before.getNumBytesTx())
&& (after.getNumBytesRx() > before.getNumBytesRx())
&& (after.getNumPacketsTx() > before.getNumPacketsTx())
&& (after.getNumPacketsRx() > before.getNumPacketsRx());
}
private static String executeShellCommand(String command) {
final String result = runShellCommand(command).trim();
Log.d(TAG, "Output of '" + command + "': '" + result + "'");
return result;
}
}