1 /*
2  * Copyright (C) 2023 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 package com.android.systemui
17 
18 import com.android.systemui.dagger.qualifiers.Application
19 import com.android.systemui.dagger.qualifiers.Background
20 import com.android.systemui.dagger.qualifiers.Main
21 import dagger.Binds
22 import dagger.Module
23 import dagger.Provides
24 import kotlin.coroutines.ContinuationInterceptor
25 import kotlin.coroutines.CoroutineContext
26 import kotlinx.coroutines.CoroutineDispatcher
27 import kotlinx.coroutines.CoroutineScope
28 import kotlinx.coroutines.test.TestCoroutineScheduler
29 import kotlinx.coroutines.test.TestDispatcher
30 import kotlinx.coroutines.test.TestScope
31 
32 @Module(includes = [CoroutineTestScopeModule.Bindings::class])
33 class CoroutineTestScopeModule
34 private constructor(
35     @get:Provides val scope: TestScope,
36     @get:Provides val dispatcher: TestDispatcher,
37     @get:Provides val scheduler: TestCoroutineScheduler = dispatcher.scheduler,
38 ) {
39 
40     constructor() : this(TestScope())
41 
42     constructor(
43         scope: TestScope
44     ) : this(scope, scope.coroutineContext[ContinuationInterceptor] as TestDispatcher)
45 
46     constructor(context: CoroutineContext) : this(TestScope(context))
47 
48     @get:[Provides Application]
49     val appScope: CoroutineScope = scope.backgroundScope
50 
51     @get:[Provides Background]
52     val bgScope: CoroutineScope = scope.backgroundScope
53 
54     @Module
55     interface Bindings {
bindMainContextnull56         @Binds @Main fun bindMainContext(dispatcher: TestDispatcher): CoroutineContext
57         @Binds @Main fun bindMainDispatcher(dispatcher: TestDispatcher): CoroutineDispatcher
58         @Binds @Background fun bindBgContext(dispatcher: TestDispatcher): CoroutineContext
59         @Binds @Background fun bindBgDispatcher(dispatcher: TestDispatcher): CoroutineDispatcher
60     }
61 }
62