1 /* 2 * Copyright (C) 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.wallpaper.wrapper 18 19 import android.content.Context 20 import com.android.wallpaper.picker.category.data.repository.WallpaperCategoryRepository 21 import com.android.wallpaper.picker.category.wrapper.DefaultWallpaperCategoryWrapper 22 import com.google.common.truth.Truth.assertThat 23 import dagger.hilt.android.qualifiers.ApplicationContext 24 import dagger.hilt.android.testing.HiltAndroidRule 25 import dagger.hilt.android.testing.HiltAndroidTest 26 import javax.inject.Inject 27 import kotlinx.coroutines.test.runTest 28 import org.junit.Before 29 import org.junit.Rule 30 import org.junit.Test 31 import org.junit.runner.RunWith 32 import org.robolectric.RobolectricTestRunner 33 34 @RunWith(RobolectricTestRunner::class) 35 @HiltAndroidTest 36 class DefaultWallpaperCategoryWrapperTest { 37 @get:Rule var hiltRule = HiltAndroidRule(this) 38 39 @Inject @ApplicationContext lateinit var context: Context 40 41 @Inject lateinit var fakeDefaultWallpaperCategoryRepository: WallpaperCategoryRepository 42 43 @Inject lateinit var defaultWallpaperCategoryWrapper: DefaultWallpaperCategoryWrapper 44 45 @Before setUpnull46 fun setUp() { 47 hiltRule.inject() 48 } 49 50 @Test <lambda>null51 fun testGetCategories() = runTest { 52 val categories = defaultWallpaperCategoryWrapper.getCategories(false) 53 assertThat(categories).isNotEmpty() 54 assertThat(categories.size).isEqualTo(1) 55 } 56 } 57