1 /* 2 * Copyright (C) 2022 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.wifi.hal; 18 19 import static junit.framework.Assert.assertEquals; 20 21 import static org.mockito.Mockito.when; 22 23 import java.util.function.Supplier; 24 25 public class HalTestUtils { 26 /** 27 * Check that we get the expected return value when the specified method is called. 28 * 29 * @param calledMethod Method to call on mDut. 30 * @param mockedMethod Method called by mDut to retrieve the value. 31 * @param value Value that the mockedMethod should return. 32 */ verifyReturnValue(Supplier<T> calledMethod, T mockedMethod, T value)33 public static <T> void verifyReturnValue(Supplier<T> calledMethod, T mockedMethod, T value) { 34 when(mockedMethod).thenReturn(value); 35 T retrievedValue = calledMethod.get(); 36 assertEquals(value, retrievedValue); 37 } 38 } 39