diff options
Diffstat (limited to 'tests/PhotoPicker/src/android/photopicker/cts/util/PhotoPickerFilesUtils.java')
-rw-r--r-- | tests/PhotoPicker/src/android/photopicker/cts/util/PhotoPickerFilesUtils.java | 116 |
1 files changed, 116 insertions, 0 deletions
diff --git a/tests/PhotoPicker/src/android/photopicker/cts/util/PhotoPickerFilesUtils.java b/tests/PhotoPicker/src/android/photopicker/cts/util/PhotoPickerFilesUtils.java new file mode 100644 index 00000000000..37a44f99915 --- /dev/null +++ b/tests/PhotoPicker/src/android/photopicker/cts/util/PhotoPickerFilesUtils.java @@ -0,0 +1,116 @@ +/* + * 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.photopicker.cts.util; + +import android.app.UiAutomation; +import android.content.Context; +import android.net.Uri; +import android.os.FileUtils; +import android.os.UserHandle; +import android.photopicker.cts.R; +import android.provider.MediaStore; +import android.provider.cts.media.MediaStoreUtils; + +import androidx.test.InstrumentationRegistry; + +import com.android.compatibility.common.util.ShellUtils; + +import java.io.IOException; +import java.io.InputStream; +import java.io.OutputStream; +import java.util.List; + +/** + * Photo Picker Utility methods for media files creation and deletion. + */ +public class PhotoPickerFilesUtils { + public static final String DISPLAY_NAME_PREFIX = "ctsPhotoPicker"; + + public static void createImages(int count, int userId, List<Uri> uriList) + throws Exception { + for (int i = 0; i < count; i++) { + final Uri uri = createImage(userId); + uriList.add(uri); + clearMediaOwner(uri, userId); + } + } + + public static void createVideos(int count, int userId, List<Uri> uriList) + throws Exception { + for (int i = 0; i < count; i++) { + final Uri uri = createVideo(userId); + uriList.add(uri); + clearMediaOwner(uri, userId); + } + } + + public static void deleteMedia(Uri uri, int userId) throws Exception { + final String cmd = String.format("content delete --uri %s --user %d", uri, userId); + ShellUtils.runShellCommand(cmd); + } + + private static void clearMediaOwner(Uri uri, int userId) throws Exception { + final String cmd = String.format( + "content update --uri %s --user %d --bind owner_package_name:n:", uri, userId); + ShellUtils.runShellCommand(cmd); + } + + private static Uri createVideo(int userId) throws Exception { + final Uri uri = stageMedia(R.raw.testvideo_meta, + MediaStore.Video.Media.EXTERNAL_CONTENT_URI, "video/mp4", userId); + return uri; + } + + private static Uri createImage(int userId) throws Exception { + final Uri uri = stageMedia(R.raw.lg_g4_iso_800_jpg, + MediaStore.Images.Media.EXTERNAL_CONTENT_URI, "image/jpeg", userId); + return uri; + } + + private static Uri stageMedia(int resId, Uri collectionUri, String mimeType, int userId) throws + Exception { + UiAutomation uiAutomation = InstrumentationRegistry.getInstrumentation().getUiAutomation(); + uiAutomation.adoptShellPermissionIdentity( + android.Manifest.permission.INTERACT_ACROSS_USERS, + android.Manifest.permission.INTERACT_ACROSS_USERS_FULL); + try { + final Context context = InstrumentationRegistry.getTargetContext(); + final Context userContext = userId == context.getUserId() ? context : + context.createPackageContextAsUser("android", /* flags= */ 0, + UserHandle.of(userId)); + return stageMedia(resId, collectionUri, mimeType, userContext); + } finally { + uiAutomation.dropShellPermissionIdentity(); + } + } + + private static Uri stageMedia(int resId, Uri collectionUri, String mimeType, Context context) + throws IOException { + final String displayName = DISPLAY_NAME_PREFIX + System.nanoTime(); + final MediaStoreUtils.PendingParams params = new MediaStoreUtils.PendingParams( + collectionUri, displayName, mimeType); + final Uri pendingUri = MediaStoreUtils.createPending(context, params); + try (MediaStoreUtils.PendingSession session = MediaStoreUtils.openPending(context, + pendingUri)) { + try (InputStream source = context.getResources().openRawResource(resId); + OutputStream target = session.openOutputStream()) { + FileUtils.copy(source, target); + } + return session.publish(); + } + } +} |