| /* |
| * Copyright (C) 2020 The Android Open Source 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 android.car.watchdoglib; |
| |
| import static android.car.test.mocks.AndroidMockitoHelper.mockQueryService; |
| |
| import static com.android.dx.mockito.inline.extended.ExtendedMockito.mockitoSession; |
| |
| import static com.google.common.truth.Truth.assertThat; |
| |
| import static org.mockito.Mockito.eq; |
| import static org.mockito.Mockito.never; |
| import static org.mockito.Mockito.verify; |
| import static org.mockito.Mockito.when; |
| import static org.testng.Assert.assertThrows; |
| |
| import android.automotive.watchdog.internal.ComponentType; |
| import android.automotive.watchdog.internal.ICarWatchdog; |
| import android.automotive.watchdog.internal.ICarWatchdogMonitor; |
| import android.automotive.watchdog.internal.ICarWatchdogServiceForSystem; |
| import android.automotive.watchdog.internal.PackageInfo; |
| import android.automotive.watchdog.internal.PackageIoOveruseStats; |
| import android.automotive.watchdog.internal.PackageResourceOveruseAction; |
| import android.automotive.watchdog.internal.PowerCycle; |
| import android.automotive.watchdog.internal.ResourceOveruseConfiguration; |
| import android.automotive.watchdog.internal.StateType; |
| import android.os.Binder; |
| import android.os.IBinder; |
| import android.os.RemoteException; |
| import android.os.ServiceManager; |
| |
| import org.junit.After; |
| import org.junit.Before; |
| import org.junit.Test; |
| import org.mockito.Mock; |
| import org.mockito.MockitoSession; |
| import org.mockito.Spy; |
| import org.mockito.quality.Strictness; |
| |
| import java.util.ArrayList; |
| import java.util.Collections; |
| import java.util.List; |
| |
| /** |
| * <p>This class contains unit tests for the {@link CarWatchdogDaemonHelper}. |
| */ |
| public class CarWatchdogDaemonHelperTest { |
| |
| private static final String CAR_WATCHDOG_DAEMON_INTERFACE = "carwatchdogd_system"; |
| |
| @Mock CarWatchdogDaemonHelper.OnConnectionChangeListener mListener; |
| @Mock private IBinder mBinder = new Binder(); |
| @Spy private ICarWatchdog mFakeCarWatchdog = new FakeCarWatchdog(); |
| private CarWatchdogDaemonHelper mCarWatchdogDaemonHelper; |
| private MockitoSession mMockSession; |
| |
| @Before |
| public void setUp() { |
| mMockSession = mockitoSession() |
| .initMocks(this) |
| .strictness(Strictness.LENIENT) |
| .spyStatic(ServiceManager.class) |
| .startMocking(); |
| mockQueryService(CAR_WATCHDOG_DAEMON_INTERFACE, mBinder, mFakeCarWatchdog); |
| mCarWatchdogDaemonHelper = new CarWatchdogDaemonHelper(); |
| mCarWatchdogDaemonHelper.connect(); |
| } |
| |
| @After |
| public void tearDown() { |
| mMockSession.finishMocking(); |
| } |
| |
| @Test |
| public void testConnection() { |
| CarWatchdogDaemonHelper carWatchdogDaemonHelper = new CarWatchdogDaemonHelper(); |
| carWatchdogDaemonHelper.addOnConnectionChangeListener(mListener); |
| |
| carWatchdogDaemonHelper.connect(); |
| |
| verify(mListener).onConnectionChange(true); |
| } |
| |
| @Test |
| public void testRemoveConnectionChangeListener() { |
| CarWatchdogDaemonHelper carWatchdogDaemonHelper = new CarWatchdogDaemonHelper(); |
| carWatchdogDaemonHelper.addOnConnectionChangeListener(mListener); |
| carWatchdogDaemonHelper.removeOnConnectionChangeListener(mListener); |
| carWatchdogDaemonHelper.connect(); |
| |
| verify(mListener, never()).onConnectionChange(true); |
| } |
| |
| @Test |
| public void testIndirectCall_RegisterUnregisterMediator() throws Exception { |
| ICarWatchdogServiceForSystem service = new ICarWatchdogServiceForSystem.Default(); |
| |
| mCarWatchdogDaemonHelper.registerCarWatchdogService(service); |
| |
| verify(mFakeCarWatchdog).registerCarWatchdogService(service); |
| |
| mCarWatchdogDaemonHelper.unregisterCarWatchdogService(service); |
| |
| verify(mFakeCarWatchdog).unregisterCarWatchdogService(service); |
| } |
| |
| @Test |
| public void testIndirectCall_RegisterUnregisterMonitor() throws Exception { |
| ICarWatchdogMonitor monitor = new ICarWatchdogMonitor.Default(); |
| |
| mCarWatchdogDaemonHelper.registerMonitor(monitor); |
| |
| verify(mFakeCarWatchdog).registerMonitor(monitor); |
| |
| mCarWatchdogDaemonHelper.unregisterMonitor(monitor); |
| |
| verify(mFakeCarWatchdog).unregisterMonitor(monitor); |
| } |
| |
| @Test |
| public void testIndirectCall_TellCarWatchdogServiceAlive() throws Exception { |
| ICarWatchdogServiceForSystem service = new ICarWatchdogServiceForSystem.Default(); |
| int[] pids = new int[]{111}; |
| |
| mCarWatchdogDaemonHelper.tellCarWatchdogServiceAlive(service, pids, 123456); |
| |
| verify(mFakeCarWatchdog).tellCarWatchdogServiceAlive(service, pids, 123456); |
| } |
| |
| @Test |
| public void testIndirectCall_TellDumpFinished() throws Exception { |
| ICarWatchdogMonitor monitor = new ICarWatchdogMonitor.Default(); |
| |
| mCarWatchdogDaemonHelper.tellDumpFinished(monitor, 123456); |
| |
| verify(mFakeCarWatchdog).tellDumpFinished(monitor, 123456); |
| } |
| |
| @Test |
| public void testIndirectCall_NotifySystemStateChange() throws Exception { |
| mCarWatchdogDaemonHelper.notifySystemStateChange(StateType.POWER_CYCLE, |
| PowerCycle.POWER_CYCLE_SHUTDOWN_PREPARE, -1); |
| |
| verify(mFakeCarWatchdog).notifySystemStateChange(StateType.POWER_CYCLE, |
| PowerCycle.POWER_CYCLE_SHUTDOWN_PREPARE, -1); |
| } |
| |
| @Test |
| public void testIndirectCall_updateResourceOveruseConfigurations() throws Exception { |
| ResourceOveruseConfiguration config = new ResourceOveruseConfiguration(); |
| config.componentType = ComponentType.SYSTEM; |
| List<ResourceOveruseConfiguration> configs = new ArrayList<>(Collections.singleton(config)); |
| |
| mCarWatchdogDaemonHelper.updateResourceOveruseConfigurations(configs); |
| |
| verify(mFakeCarWatchdog).updateResourceOveruseConfigurations(eq(configs)); |
| } |
| |
| @Test |
| public void testIndirectCall_getResourceOveruseConfigurations() throws Exception { |
| ResourceOveruseConfiguration config = new ResourceOveruseConfiguration(); |
| config.componentType = ComponentType.SYSTEM; |
| List<ResourceOveruseConfiguration> expected = |
| new ArrayList<>(Collections.singleton(config)); |
| when(mFakeCarWatchdog.getResourceOveruseConfigurations()).thenReturn(expected); |
| |
| List<ResourceOveruseConfiguration> actual = |
| mCarWatchdogDaemonHelper.getResourceOveruseConfigurations(); |
| |
| assertThat(actual).isEqualTo(expected); |
| } |
| |
| @Test |
| public void testIndirectCall_actionTakenOnResourceOveruse() throws Exception { |
| List<PackageResourceOveruseAction> actions = new ArrayList<>(); |
| |
| mCarWatchdogDaemonHelper.actionTakenOnResourceOveruse(actions); |
| |
| verify(mFakeCarWatchdog).actionTakenOnResourceOveruse(eq(actions)); |
| } |
| |
| /* |
| * Test that the {@link CarWatchdogDaemonHelper} throws {@code IllegalArgumentException} when |
| * trying to register already-registered service again. |
| */ |
| @Test |
| public void testMultipleRegistration() throws Exception { |
| ICarWatchdogServiceForSystem service = new ICarWatchdogServiceForSystemImpl(); |
| |
| mCarWatchdogDaemonHelper.registerCarWatchdogService(service); |
| |
| assertThrows(IllegalArgumentException.class, |
| () -> mCarWatchdogDaemonHelper.registerCarWatchdogService(service)); |
| } |
| |
| /* |
| * Test that the {@link CarWatchdogDaemonHelper} throws {@code IllegalArgumentException} when |
| * trying to unregister not-registered service. |
| */ |
| @Test |
| public void testInvalidUnregistration() throws Exception { |
| ICarWatchdogServiceForSystem service = new ICarWatchdogServiceForSystemImpl(); |
| |
| assertThrows(IllegalArgumentException.class, |
| () -> mCarWatchdogDaemonHelper.unregisterCarWatchdogService(service)); |
| } |
| |
| // FakeCarWatchdog mimics ICarWatchdog daemon in local process. |
| private final class FakeCarWatchdog extends ICarWatchdog.Default { |
| |
| private final ArrayList<ICarWatchdogServiceForSystem> mServices = new ArrayList<>(); |
| |
| @Override |
| public void registerCarWatchdogService( |
| ICarWatchdogServiceForSystem service) throws RemoteException { |
| for (ICarWatchdogServiceForSystem curService : mServices) { |
| if (curService == service) { |
| throw new IllegalArgumentException("Already registered service"); |
| } |
| } |
| mServices.add(service); |
| } |
| |
| @Override |
| public void unregisterCarWatchdogService( |
| ICarWatchdogServiceForSystem service) throws RemoteException { |
| for (ICarWatchdogServiceForSystem curService : mServices) { |
| if (curService == service) { |
| mServices.remove(service); |
| return; |
| } |
| } |
| throw new IllegalArgumentException("Not registered service"); |
| } |
| } |
| |
| private final class ICarWatchdogServiceForSystemImpl extends ICarWatchdogServiceForSystem.Stub { |
| @Override |
| public void checkIfAlive(int sessionId, int timeout) {} |
| |
| @Override |
| public void prepareProcessTermination() {} |
| |
| @Override |
| public List<PackageInfo> getPackageInfosForUids( |
| int[] uids, List<String> vendorPackagePrefixes) { |
| return new ArrayList<>(); |
| } |
| |
| @Override |
| public void latestIoOveruseStats(List<PackageIoOveruseStats> ioOveruseStats) {} |
| |
| @Override |
| public void resetResourceOveruseStats(List<String> packageNames) {} |
| } |
| } |