summaryrefslogtreecommitdiff
path: root/feature/preview/src/test/java/com/google/jetpackcamera/feature/preview/PreviewViewModelTest.kt
blob: 0ecd474853e4a4978637d4348152b9cfa88e5ac1 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
/*
 * Copyright (C) 2023 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 com.google.jetpackcamera.feature.preview

import android.content.ContentResolver
import androidx.camera.core.Preview.SurfaceProvider
import com.google.jetpackcamera.domain.camera.test.FakeCameraUseCase
import com.google.jetpackcamera.settings.model.FlashMode
import com.google.jetpackcamera.settings.test.FakeSettingsRepository
import junit.framework.TestCase.assertEquals
import kotlinx.coroutines.Dispatchers
import kotlinx.coroutines.ExperimentalCoroutinesApi
import kotlinx.coroutines.test.StandardTestDispatcher
import kotlinx.coroutines.test.advanceUntilIdle
import kotlinx.coroutines.test.runTest
import kotlinx.coroutines.test.setMain
import org.junit.Before
import org.junit.Test
import org.mockito.Mockito.mock

@OptIn(ExperimentalCoroutinesApi::class)
class PreviewViewModelTest {

    private val cameraUseCase = FakeCameraUseCase()
    private lateinit var previewViewModel: PreviewViewModel

    @Before
    fun setup() = runTest(StandardTestDispatcher()) {
        Dispatchers.setMain(StandardTestDispatcher())
        previewViewModel = PreviewViewModel(cameraUseCase, FakeSettingsRepository)
        advanceUntilIdle()
    }

    @Test
    fun getPreviewUiState() = runTest(StandardTestDispatcher()) {
        advanceUntilIdle()
        val uiState = previewViewModel.previewUiState.value
        assertEquals(CameraState.READY, uiState.cameraState)
    }

    @Test
    fun runCamera() = runTest(StandardTestDispatcher()) {
        val surfaceProvider: SurfaceProvider = mock(SurfaceProvider::class.java)
        previewViewModel.runCamera(surfaceProvider)
        advanceUntilIdle()

        assertEquals(cameraUseCase.previewStarted, true)
    }

    @Test
    fun captureImage() = runTest(StandardTestDispatcher()) {
        val surfaceProvider: SurfaceProvider = mock(SurfaceProvider::class.java)
        previewViewModel.runCamera(surfaceProvider)
        previewViewModel.captureImage()
        advanceUntilIdle()
        assertEquals(cameraUseCase.numPicturesTaken, 1)
    }

    @Test
    fun captureImageWithUri() = runTest(StandardTestDispatcher()) {
        val surfaceProvider: SurfaceProvider = mock(SurfaceProvider::class.java)
        val contentResolver: ContentResolver = mock(ContentResolver::class.java)
        previewViewModel.runCamera(surfaceProvider)
        previewViewModel.captureImage(contentResolver, null) {}
        advanceUntilIdle()
        assertEquals(cameraUseCase.numPicturesTaken, 1)
    }

    @Test
    fun startVideoRecording() = runTest(StandardTestDispatcher()) {
        previewViewModel.runCamera(mock(SurfaceProvider::class.java))
        previewViewModel.startVideoRecording()
        advanceUntilIdle()
        assertEquals(cameraUseCase.recordingInProgress, true)
    }

    @Test
    fun stopVideoRecording() = runTest(StandardTestDispatcher()) {
        previewViewModel.runCamera(mock(SurfaceProvider::class.java))
        previewViewModel.startVideoRecording()
        advanceUntilIdle()
        previewViewModel.stopVideoRecording()
        assertEquals(cameraUseCase.recordingInProgress, false)
    }

    @Test
    fun setFlash() = runTest(StandardTestDispatcher()) {
        previewViewModel.runCamera(mock(SurfaceProvider::class.java))
        previewViewModel.setFlash(FlashMode.AUTO)
        advanceUntilIdle()
        assertEquals(
            previewViewModel.previewUiState.value.currentCameraSettings.flashMode,
            FlashMode.AUTO
        )
    }

    @Test
    fun flipCamera() = runTest(StandardTestDispatcher()) {
        // initial default value should be back
        previewViewModel.runCamera(mock(SurfaceProvider::class.java))
        assertEquals(
            previewViewModel.previewUiState.value.currentCameraSettings.isFrontCameraFacing,
            false
        )
        previewViewModel.flipCamera()

        advanceUntilIdle()
        // ui state and camera should both be true now
        assertEquals(
            previewViewModel.previewUiState.value.currentCameraSettings.isFrontCameraFacing,
            true
        )
        assertEquals(true, cameraUseCase.isLensFacingFront)
    }
}