diff options
Diffstat (limited to 'sdksandbox/tests/testutils/src/android/app/sdksandbox/hosttestutils')
2 files changed, 219 insertions, 0 deletions
diff --git a/sdksandbox/tests/testutils/src/android/app/sdksandbox/hosttestutils/AdoptableStorageUtils.java b/sdksandbox/tests/testutils/src/android/app/sdksandbox/hosttestutils/AdoptableStorageUtils.java new file mode 100644 index 0000000000..7e62875f01 --- /dev/null +++ b/sdksandbox/tests/testutils/src/android/app/sdksandbox/hosttestutils/AdoptableStorageUtils.java @@ -0,0 +1,135 @@ +/* + * Copyright (C) 2022 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.app.sdksandbox.hosttestutils; + +import com.android.tradefed.log.LogUtil; +import com.android.tradefed.testtype.junit4.BaseHostJUnit4Test; + +import java.util.Arrays; + +public class AdoptableStorageUtils { + + private final BaseHostJUnit4Test mTest; + + private String mDiskId; + + public AdoptableStorageUtils(BaseHostJUnit4Test test) { + mTest = test; + } + + public boolean isAdoptableStorageSupported() throws Exception { + boolean hasFeature = + mTest.getDevice().hasFeature("feature:android.software.adoptable_storage"); + boolean hasFstab = + Boolean.parseBoolean( + mTest.getDevice().executeShellCommand("sm has-adoptable").trim()); + return hasFeature && hasFstab; + } + + // Creates a new volume in adoptable storage and returns its uuid + public String createNewVolume() throws Exception { + mDiskId = getAdoptionDisk(); + assertEmpty(mTest.getDevice().executeShellCommand("sm partition " + mDiskId + " private")); + final LocalVolumeInfo vol = getAdoptionVolume(); + return vol.uuid; + } + + // Destroy the volume created before + public void cleanUpVolume() throws Exception { + mTest.getDevice().executeShellCommand("sm partition " + mDiskId + " public"); + mTest.getDevice().executeShellCommand("sm forget all"); + } + + private String getAdoptionDisk() throws Exception { + // In the case where we run multiple test we cleanup the state of the device. This + // results in the execution of sm forget all which causes the MountService to "reset" + // all its knowledge about available drives. This can cause the adoptable drive to + // become temporarily unavailable. + int attempt = 0; + String disks = mTest.getDevice().executeShellCommand("sm list-disks adoptable"); + while ((disks == null || disks.isEmpty()) && attempt++ < 15) { + Thread.sleep(1000); + disks = mTest.getDevice().executeShellCommand("sm list-disks adoptable"); + } + + if (disks == null || disks.isEmpty()) { + throw new AssertionError( + "Devices that claim to support adoptable storage must have " + + "adoptable media inserted during CTS to verify correct behavior"); + } + return disks.split("\n")[0].trim(); + } + + private static void assertEmpty(String str) { + if (str != null && str.trim().length() > 0) { + throw new AssertionError("Expected empty string but found " + str); + } + } + + private static class LocalVolumeInfo { + public String volId; + public String state; + public String uuid; + + LocalVolumeInfo(String line) { + final String[] split = line.split(" "); + volId = split[0]; + state = split[1]; + uuid = split[2]; + } + } + + private LocalVolumeInfo getAdoptionVolume() throws Exception { + String[] lines = null; + int attempt = 0; + int mounted_count = 0; + while (attempt++ < 15) { + lines = mTest.getDevice().executeShellCommand("sm list-volumes private").split("\n"); + LogUtil.CLog.w("getAdoptionVolume(): " + Arrays.toString(lines)); + for (String line : lines) { + final LocalVolumeInfo info = new LocalVolumeInfo(line.trim()); + if (!"private".equals(info.volId)) { + if ("mounted".equals(info.state)) { + // make sure the storage is mounted and stable for a while + mounted_count++; + attempt--; + if (mounted_count >= 3) { + return waitForVolumeReady(info); + } + } else { + mounted_count = 0; + } + } + } + Thread.sleep(1000); + } + throw new AssertionError("Expected private volume; found " + Arrays.toString(lines)); + } + + private LocalVolumeInfo waitForVolumeReady(LocalVolumeInfo vol) throws Exception { + int attempt = 0; + while (attempt++ < 15) { + if (mTest.getDevice() + .executeShellCommand("dumpsys package volumes") + .contains(vol.volId)) { + return vol; + } + Thread.sleep(1000); + } + throw new AssertionError("Volume not ready " + vol.volId); + } +} diff --git a/sdksandbox/tests/testutils/src/android/app/sdksandbox/hosttestutils/SecondaryUserUtils.java b/sdksandbox/tests/testutils/src/android/app/sdksandbox/hosttestutils/SecondaryUserUtils.java new file mode 100644 index 0000000000..d063b89d8a --- /dev/null +++ b/sdksandbox/tests/testutils/src/android/app/sdksandbox/hosttestutils/SecondaryUserUtils.java @@ -0,0 +1,84 @@ +/* + * Copyright (C) 2022 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.app.sdksandbox.hosttestutils; + +import static com.google.common.truth.Truth.assertThat; + +import static org.junit.Assert.fail; + +import com.android.tradefed.testtype.junit4.BaseHostJUnit4Test; + +public class SecondaryUserUtils { + + private static final long NUMBER_OF_POLLS = 2 * 60; + private static final long POLL_INTERVAL_IN_MILLIS = 1000; + + private final BaseHostJUnit4Test mTest; + + private int mOriginalUserId = -1; + private int mSecondaryUserId = -1; + + public SecondaryUserUtils(BaseHostJUnit4Test test) { + mTest = test; + } + + public int createAndStartSecondaryUser() throws Exception { + if (mSecondaryUserId != -1) { + throw new IllegalStateException("Cannot create secondary user, it already exists"); + } + mOriginalUserId = mTest.getDevice().getCurrentUser(); + String name = "SdkSandboxStorageHost_User" + System.currentTimeMillis(); + mSecondaryUserId = mTest.getDevice().createUser(name); + // Note we can't install apps on a locked user, so we wait + mTest.getDevice().startUser(mSecondaryUserId, /*waitFlag=*/ true); + return mSecondaryUserId; + } + + public void removeSecondaryUserIfNecessary() throws Exception { + removeSecondaryUserIfNecessary(/*waitForUserDataDeletion=*/ false); + } + + public void removeSecondaryUserIfNecessary(boolean waitForUserDataDeletion) throws Exception { + if (mSecondaryUserId == -1) { + return; + } + + final int userBeingRemoved = mSecondaryUserId; + // Set to -1 so that we can create new users later, even when removal goes wrong + mSecondaryUserId = -1; + + if (mOriginalUserId != -1 && userBeingRemoved != -1) { + // Can't remove the 2nd user without switching out of it + assertThat(mTest.getDevice().switchUser(mOriginalUserId)).isTrue(); + mTest.getDevice().removeUser(userBeingRemoved); + if (waitForUserDataDeletion) { + waitForUserDataDeletion(userBeingRemoved); + } + } + } + + private void waitForUserDataDeletion(int userId) throws Exception { + final String deSdkSandboxDataRootPath = "/data/misc_de/" + userId + "/sdksandbox"; + for (int i = 0; i < NUMBER_OF_POLLS; ++i) { + if (!mTest.getDevice().isDirectory(deSdkSandboxDataRootPath)) { + return; + } + Thread.sleep(POLL_INTERVAL_IN_MILLIS); + } + fail("User data was not deleted for UserId " + userId); + } +} |