1 /*
2  * Copyright (C) 2020 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.server.extendedtestutils
18 
19 import com.android.dx.mockito.inline.extended.ExtendedMockito
20 import com.android.dx.mockito.inline.extended.MockedMethod
21 import com.android.dx.mockito.inline.extended.StaticCapableStubber
22 import org.mockito.stubbing.Answer
23 
24 fun <T> StaticCapableStubber.wheneverStatic(methodCall: MockedMethod<T>) {
25     this.`when`(methodCall)
26 }
27 
28 fun <T> wheneverStatic(mockedMethod: MockedMethod<T>) = object : CustomStaticStubber<T> {
29     override fun thenAnswer(answer: Answer<T>) {
30         ExtendedMockito.doAnswer(answer).wheneverStatic(mockedMethod)
31     }
32 
33     override fun thenReturn(value: T) {
34         ExtendedMockito.doReturn(value).wheneverStatic(mockedMethod)
35     }
36 
37     override fun thenDoNothing() {
38         ExtendedMockito.doNothing().wheneverStatic(mockedMethod)
39     }
40 }
41 
42 interface CustomStaticStubber<T> {
43     fun thenAnswer(answer: Answer<T>)
44     fun thenReturn(value: T)
45     fun thenDoNothing()
46 }
47