1 /* 2 * Copyright (C) 2023 The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17 package com.android.intentresolver 18 19 import androidx.lifecycle.ViewModel 20 import androidx.lifecycle.ViewModelProvider 21 import androidx.lifecycle.viewmodel.CreationExtras 22 import com.android.intentresolver.contentpreview.BasePreviewViewModel 23 import com.android.intentresolver.contentpreview.ImageLoader 24 import com.android.intentresolver.contentpreview.PreviewDataProvider 25 26 /** A test content preview model that supports image loader override. */ 27 class TestContentPreviewViewModel( 28 private val viewModel: BasePreviewViewModel, 29 private val imageLoader: ImageLoader? = null, 30 ) : BasePreviewViewModel() { createOrReuseProvidernull31 override fun createOrReuseProvider( 32 chooserRequest: ChooserRequestParameters 33 ): PreviewDataProvider = viewModel.createOrReuseProvider(chooserRequest) 34 35 override fun createOrReuseImageLoader(): ImageLoader = 36 imageLoader ?: viewModel.createOrReuseImageLoader() 37 38 companion object { 39 fun wrap( 40 factory: ViewModelProvider.Factory, 41 imageLoader: ImageLoader?, 42 ): ViewModelProvider.Factory = 43 object : ViewModelProvider.Factory { 44 @Suppress("UNCHECKED_CAST") 45 override fun <T : ViewModel> create( 46 modelClass: Class<T>, 47 extras: CreationExtras 48 ): T { 49 return TestContentPreviewViewModel( 50 factory.create(modelClass, extras) as BasePreviewViewModel, 51 imageLoader, 52 ) as T 53 } 54 } 55 } 56 } 57