1 /*
2  * Copyright 2024 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.photopicker.core.database
18 
19 import com.android.photopicker.core.banners.BannerStateDao
20 import org.mockito.Mockito.mock
21 
22 /**
23  * This is a test implementation of [DatabaseManager] that will isolate the device state and mock
24  * out any database interactions.
25  */
26 class DatabaseManagerTestImpl() : DatabaseManager {
27 
28     val bannerState = mock(BannerStateDao::class.java)
29 
30     @Suppress("UNCHECKED_CAST")
acquireDaonull31     override fun <T> acquireDao(daoClass: Class<T>): T {
32         with(daoClass) {
33             return when {
34                 isAssignableFrom(BannerStateDao::class.java) -> bannerState as T
35                 else ->
36                     throw IllegalArgumentException(
37                         "Cannot acquire ${daoClass.simpleName} from DatabaseManagerImpl"
38                     )
39             }
40         }
41     }
42 }
43