tests/src/org/lineageos/tests/common/MockIBinderStubForInterface.java


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

/**
  * Copyright (c) 2016, The CyanogenMod 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 org.lineageos.tests.common;

import android.os.IBinder;
import android.os.IInterface;
import android.os.RemoteCallbackList;
import org.junit.Assert;
import org.mockito.Mockito;

import java.lang.reflect.Field;

/**
 * Helper class to mock stubs for IInterfaces
 * Ensures that when querying the local interface
 * we return the instance itself as to preserve the mock instance
 */
public final class MockIBinderStubForInterface {
    private MockIBinderStubForInterface() {}

    private static <T extends IBinder> String getStubDescriptor(Class<T> stubClass) {
        String descriptor = null;
        try {
            Field f = stubClass.getDeclaredField("DESCRIPTOR");
            f.setAccessible(true);
            descriptor = (String) f.get(stubClass);
        } catch (NoSuchFieldException | IllegalAccessException e) {
            Assert.fail(e.getMessage());
        }
        return descriptor;
    }

    public static <T extends IBinder> T getMockInterface(Class<T> stub) {
        String descriptor = getStubDescriptor(stub);
        T mockInterface = Mockito.mock(stub);
        Mockito.doReturn(mockInterface)
                .when(mockInterface)
                .queryLocalInterface(descriptor == null ?
                        Mockito.anyString() : Mockito.eq(descriptor));
        Mockito.doReturn(Mockito.mock(IBinder.class))
                .when((IInterface) mockInterface)
                .asBinder();
        return mockInterface;
    }
}