| /* |
| * 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 com.android.cts.appcloningtestapp; |
| |
| import static com.google.common.truth.Truth.assertThat; |
| |
| import android.Manifest; |
| import android.content.Context; |
| import android.graphics.Bitmap; |
| import android.graphics.Canvas; |
| import android.graphics.Color; |
| import android.graphics.Paint; |
| import android.os.Bundle; |
| import android.os.storage.StorageManager; |
| import android.os.storage.StorageVolume; |
| import android.util.Log; |
| |
| import androidx.test.InstrumentationRegistry; |
| |
| import com.android.cts.install.lib.InstallUtils; |
| |
| import org.junit.After; |
| import org.junit.Before; |
| import org.junit.Test; |
| import org.junit.runner.RunWith; |
| import org.junit.runners.JUnit4; |
| |
| import java.util.List; |
| |
| @RunWith(JUnit4.class) |
| public class AppCloningDeviceTest { |
| private static final String EMPTY_STRING = ""; |
| private static final String TAG = "AppCloningDeviceTest"; |
| |
| private Context mContext; |
| private StorageManager mStorageManager; |
| |
| @Before |
| public void setUp() throws Exception { |
| mContext = InstrumentationRegistry.getContext(); |
| mStorageManager = mContext.getSystemService(StorageManager.class); |
| |
| // Adopts common permission to read and write to shared storage |
| InstallUtils.adoptShellPermissionIdentity( |
| Manifest.permission.READ_EXTERNAL_STORAGE, |
| Manifest.permission.WRITE_EXTERNAL_STORAGE, |
| Manifest.permission.MANAGE_EXTERNAL_STORAGE); |
| } |
| |
| @After |
| public void tearDown() { |
| // Drops adopted shell permissions |
| InstallUtils.dropShellPermissionIdentity(); |
| } |
| |
| private String getTestArgumentValueForGivenKey(String testArgumentKey) { |
| final Bundle testArguments = InstrumentationRegistry.getArguments(); |
| String testArgumentValue = testArguments.getString(testArgumentKey, EMPTY_STRING); |
| |
| return testArgumentValue; |
| } |
| |
| private String getImageNameToBeDisplayed() { |
| return getTestArgumentValueForGivenKey("imageNameToBeDisplayed"); |
| } |
| |
| private String getImageNameToBeCreated() { |
| return getTestArgumentValueForGivenKey("imageNameToBeCreated"); |
| } |
| |
| private String getImageNameToBeVerifiedInOwnerProfile() { |
| return getTestArgumentValueForGivenKey("imageNameToBeVerifiedInOwnerProfile"); |
| } |
| |
| private String getImageNameToBeVerifiedInCloneProfile() { |
| return getTestArgumentValueForGivenKey("imageNameToBeVerifiedInCloneProfile"); |
| } |
| |
| private String getCloneUserId() { |
| return getTestArgumentValueForGivenKey("cloneUserId"); |
| } |
| |
| @Test |
| public void testMediaStoreManager_verifyCrossUserImagesInSharedStorage() throws Exception { |
| // This method will be called only after writing images in owner and clone profile |
| String imageNameToBeVerifiedInOwnerProfile = getImageNameToBeVerifiedInOwnerProfile(); |
| String imageNameToBeVerifiedInCloneProfile = getImageNameToBeVerifiedInCloneProfile(); |
| |
| List<Image> imageList = MediaStoreReadOperation.getImageFilesFromMediaStore(mContext); |
| boolean verifiedImageInOwnerProfile = false; |
| boolean verifiedImageInCloneProfile = false; |
| |
| for (Image image: imageList) { |
| |
| // Eg: Data: /storage/emulated/<user_id>/Pictures/<imageName>.jpg |
| // user_id will be 0 for owner |
| if (!verifiedImageInOwnerProfile && image.getData().contains("/emulated/0/") |
| && image.getDisplayName().startsWith(imageNameToBeVerifiedInOwnerProfile)) { |
| verifiedImageInOwnerProfile = true; |
| continue; |
| } |
| |
| if (!verifiedImageInCloneProfile && image.getData().contains("/emulated/" |
| + getCloneUserId() + "/") |
| && image.getDisplayName().startsWith(imageNameToBeVerifiedInCloneProfile)) { |
| verifiedImageInCloneProfile = true; |
| } |
| } |
| |
| assertThat(verifiedImageInOwnerProfile && verifiedImageInCloneProfile).isTrue(); |
| } |
| |
| @Test |
| public void testMediaStoreManager_writeImageToSharedStorage() throws Exception { |
| String imageNameToBeCreated = getImageNameToBeCreated(); |
| |
| int color = 0x00000000; |
| if (imageNameToBeCreated.equalsIgnoreCase("clone_profile_image")) { |
| // Blue represents clone profile image |
| color = Color.BLUE; |
| } else if (imageNameToBeCreated.equalsIgnoreCase("owner_profile_image")) { |
| // Green represents owner profile image |
| color = Color.GREEN; |
| } |
| |
| Bitmap bitmap = createImage(1000, 1000, color); |
| |
| assertThat(MediaStoreWriteOperation.createImageFileToMediaStore(mContext, |
| getImageNameToBeDisplayed(), bitmap)).isTrue(); |
| } |
| |
| /** |
| * Creates a bitmap unicolor image with the given width, height and color |
| * @param width |
| * @param height |
| * @param color |
| * @return A bitmap unicolor image with the given width and height |
| */ |
| public static Bitmap createImage(int width, int height, int color) { |
| Bitmap bitmap = Bitmap.createBitmap(width, height, Bitmap.Config.ARGB_8888); |
| Canvas canvas = new Canvas(bitmap); |
| Paint paint = new Paint(); |
| paint.setColor(color); |
| canvas.drawRect(0F, 0F, (float) width, (float) height, paint); |
| return bitmap; |
| } |
| |
| @Test |
| public void testStorageManager_verifyInclusionOfSharedProfileVolumes() throws Exception { |
| int cloneUserId = -1; |
| try { |
| cloneUserId = Integer.valueOf(getCloneUserId()); |
| } catch (NumberFormatException exception) { |
| Log.d(TAG, "Failed to get clone user ID - " + exception.getMessage()); |
| } |
| assertThat(cloneUserId).isNotEqualTo(-1); |
| List<StorageVolume> volumeList = mStorageManager.getStorageVolumes(); |
| List<StorageVolume> volumeListIncludingShared = |
| mStorageManager.getStorageVolumesIncludingSharedProfiles(); |
| |
| // should contain all volumes of volumeList |
| assertThat(volumeListIncludingShared.containsAll(volumeList)).isTrue(); |
| |
| // remove volumes that belong to owner profile |
| volumeListIncludingShared.removeAll(volumeList); |
| |
| // remaining volumes should belong to the clone user. |
| for (StorageVolume vol : volumeListIncludingShared) { |
| assertThat(vol.getOwner().getIdentifier()).isEqualTo(cloneUserId); |
| } |
| } |
| } |