1 /*
2  * Copyright (C) 2022 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 android.graphics.Bitmap
20 import android.net.Uri
21 import com.android.intentresolver.contentpreview.ImageLoader
22 import java.util.function.Consumer
23 import kotlinx.coroutines.CoroutineScope
24 
25 class FakeImageLoader(initialBitmaps: Map<Uri, Bitmap> = emptyMap()) : ImageLoader {
<lambda>null26     private val bitmaps = HashMap<Uri, Bitmap>().apply { putAll(initialBitmaps) }
27 
loadImagenull28     override fun loadImage(callerScope: CoroutineScope, uri: Uri, callback: Consumer<Bitmap?>) {
29         callback.accept(bitmaps[uri])
30     }
31 
invokenull32     override suspend fun invoke(uri: Uri, caching: Boolean): Bitmap? = bitmaps[uri]
33 
34     override fun prePopulate(uris: List<Uri>) = Unit
35 
36     fun setBitmap(uri: Uri, bitmap: Bitmap) {
37         bitmaps[uri] = bitmap
38     }
39 }
40