/
UserState.kt
43 lines (36 loc) · 1.22 KB
/
UserState.kt
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
package com.example.android.coroutines.testing.scope
import kotlinx.coroutines.CoroutineScope
import kotlinx.coroutines.delay
import kotlinx.coroutines.flow.MutableStateFlow
import kotlinx.coroutines.flow.StateFlow
import kotlinx.coroutines.flow.asStateFlow
import kotlinx.coroutines.flow.update
import kotlinx.coroutines.launch
interface UserRepository {
suspend fun getAllUsers(): List<String>
suspend fun register(name: String)
}
class FakeUserRepository : UserRepository {
private var users = listOf<String>()
override suspend fun getAllUsers(): List<String> = users
override suspend fun register(name: String) {
delay(100L) // Simulate work
users = users + name
println("Registered $name")
}
}
// [START android_coroutine_test_user_state]
class UserState(
private val userRepository: UserRepository,
private val scope: CoroutineScope,
) {
private val _users = MutableStateFlow(emptyList<String>())
val users: StateFlow<List<String>> = _users.asStateFlow()
fun registerUser(name: String) {
scope.launch {
userRepository.register(name)
_users.update { userRepository.getAllUsers() }
}
}
}
// [END android_coroutine_test_user_state]