Add tests for StatusBarManagerService
These tests only test the requestAddTile functionality. This is a setup CL before adding tests for new functionality. NoBroadcastContextWrapper is part of that setup. Bug: 208671346 Test: atest StatusBarManagerServiceTileRequestTest Change-Id: Ie1e5c19c0fb03735efc808e4da6febb5f536811f
This commit is contained in:
Fabian Kozynski
@@ -0,0 +1,129 @@
|
||||
/*
|
||||
* Copyright (C) 2021 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 com.android.server.statusbar;
|
||||
|
||||
import android.content.BroadcastReceiver;
|
||||
import android.content.Context;
|
||||
import android.content.ContextWrapper;
|
||||
import android.content.Intent;
|
||||
import android.content.IntentFilter;
|
||||
import android.os.Handler;
|
||||
import android.os.UserHandle;
|
||||
|
||||
import androidx.annotation.NonNull;
|
||||
import androidx.annotation.Nullable;
|
||||
|
||||
import java.util.ArrayList;
|
||||
|
||||
/**
|
||||
* {@link ContextWrapper} that doesn't register {@link BroadcastReceiver}.
|
||||
*
|
||||
* Instead, it keeps a list of the registrations for querying.
|
||||
*/
|
||||
class NoBroadcastContextWrapper extends ContextWrapper {
|
||||
|
||||
private ArrayList<BroadcastReceiverRegistration> mRegistrationList =
|
||||
new ArrayList<>();
|
||||
|
||||
NoBroadcastContextWrapper(Context context) {
|
||||
super(context);
|
||||
}
|
||||
|
||||
@Override
|
||||
public Intent registerReceiver(@Nullable BroadcastReceiver receiver, IntentFilter filter) {
|
||||
return registerReceiver(receiver, filter, 0);
|
||||
}
|
||||
|
||||
@Override
|
||||
public Intent registerReceiver(@Nullable BroadcastReceiver receiver, IntentFilter filter,
|
||||
int flags) {
|
||||
return registerReceiver(receiver, filter, null, null, flags);
|
||||
}
|
||||
|
||||
@Override
|
||||
public Intent registerReceiver(@Nullable BroadcastReceiver receiver, IntentFilter filter,
|
||||
@Nullable String broadcastPermission, @Nullable Handler scheduler) {
|
||||
return registerReceiver(receiver, filter, broadcastPermission, scheduler, 0);
|
||||
}
|
||||
|
||||
@Override
|
||||
public Intent registerReceiver(@Nullable BroadcastReceiver receiver, IntentFilter filter,
|
||||
@Nullable String broadcastPermission, @Nullable Handler scheduler, int flags) {
|
||||
return registerReceiverAsUser(receiver, getUser(), filter, broadcastPermission, scheduler,
|
||||
flags);
|
||||
}
|
||||
|
||||
@Nullable
|
||||
@Override
|
||||
public Intent registerReceiverForAllUsers(@Nullable BroadcastReceiver receiver,
|
||||
@NonNull IntentFilter filter, @Nullable String broadcastPermission,
|
||||
@Nullable Handler scheduler) {
|
||||
return registerReceiverForAllUsers(receiver, filter, broadcastPermission, scheduler, 0);
|
||||
}
|
||||
|
||||
@Nullable
|
||||
@Override
|
||||
public Intent registerReceiverForAllUsers(@Nullable BroadcastReceiver receiver,
|
||||
@NonNull IntentFilter filter, @Nullable String broadcastPermission,
|
||||
@Nullable Handler scheduler, int flags) {
|
||||
return registerReceiverAsUser(receiver, UserHandle.ALL, filter, broadcastPermission,
|
||||
scheduler, flags);
|
||||
}
|
||||
|
||||
@Override
|
||||
public Intent registerReceiverAsUser(@Nullable BroadcastReceiver receiver, UserHandle user,
|
||||
IntentFilter filter, @Nullable String broadcastPermission,
|
||||
@Nullable Handler scheduler) {
|
||||
return registerReceiverAsUser(receiver, user, filter, broadcastPermission,
|
||||
scheduler, 0);
|
||||
}
|
||||
|
||||
@Override
|
||||
public Intent registerReceiverAsUser(@Nullable BroadcastReceiver receiver, UserHandle user,
|
||||
IntentFilter filter, @Nullable String broadcastPermission,
|
||||
@Nullable Handler scheduler, int flags) {
|
||||
BroadcastReceiverRegistration reg = new BroadcastReceiverRegistration(
|
||||
receiver, user, filter, broadcastPermission, scheduler, flags
|
||||
);
|
||||
mRegistrationList.add(reg);
|
||||
return null;
|
||||
}
|
||||
|
||||
@Override
|
||||
public void unregisterReceiver(BroadcastReceiver receiver) {
|
||||
mRegistrationList.removeIf((reg) -> reg.mReceiver == receiver);
|
||||
}
|
||||
|
||||
static class BroadcastReceiverRegistration {
|
||||
final BroadcastReceiver mReceiver;
|
||||
final UserHandle mUser;
|
||||
final IntentFilter mIntentFilter;
|
||||
final String mBroadcastPermission;
|
||||
final Handler mHandler;
|
||||
final int mFlags;
|
||||
|
||||
BroadcastReceiverRegistration(BroadcastReceiver receiver, UserHandle user,
|
||||
IntentFilter intentFilter, String broadcastPermission, Handler handler, int flags) {
|
||||
mReceiver = receiver;
|
||||
mUser = user;
|
||||
mIntentFilter = intentFilter;
|
||||
mBroadcastPermission = broadcastPermission;
|
||||
mHandler = handler;
|
||||
mFlags = flags;
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,603 @@
|
||||
/*
|
||||
* Copyright (C) 2021 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 com.android.server.statusbar;
|
||||
|
||||
import static android.app.ActivityManager.PROCESS_STATE_FOREGROUND_SERVICE;
|
||||
import static android.app.ActivityManager.PROCESS_STATE_TOP;
|
||||
|
||||
import static org.junit.Assert.assertEquals;
|
||||
import static org.junit.Assert.assertNotEquals;
|
||||
import static org.junit.Assert.fail;
|
||||
import static org.mockito.ArgumentMatchers.nullable;
|
||||
import static org.mockito.Mockito.any;
|
||||
import static org.mockito.Mockito.anyInt;
|
||||
import static org.mockito.Mockito.anyString;
|
||||
import static org.mockito.Mockito.argThat;
|
||||
import static org.mockito.Mockito.eq;
|
||||
import static org.mockito.Mockito.verify;
|
||||
import static org.mockito.Mockito.when;
|
||||
|
||||
import android.Manifest;
|
||||
import android.app.ActivityManagerInternal;
|
||||
import android.app.StatusBarManager;
|
||||
import android.content.ComponentName;
|
||||
import android.content.Intent;
|
||||
import android.content.pm.ApplicationInfo;
|
||||
import android.content.pm.PackageManager;
|
||||
import android.content.pm.PackageManagerInternal;
|
||||
import android.content.pm.ResolveInfo;
|
||||
import android.content.pm.ServiceInfo;
|
||||
import android.graphics.drawable.Icon;
|
||||
import android.hardware.display.DisplayManager;
|
||||
import android.os.Binder;
|
||||
import android.os.Looper;
|
||||
import android.os.RemoteException;
|
||||
import android.os.UserHandle;
|
||||
import android.service.quicksettings.TileService;
|
||||
import android.testing.TestableContext;
|
||||
|
||||
import androidx.test.InstrumentationRegistry;
|
||||
|
||||
import com.android.internal.statusbar.IAddTileResultCallback;
|
||||
import com.android.internal.statusbar.IStatusBar;
|
||||
import com.android.server.LocalServices;
|
||||
import com.android.server.policy.GlobalActionsProvider;
|
||||
import com.android.server.wm.ActivityTaskManagerInternal;
|
||||
|
||||
import org.junit.Before;
|
||||
import org.junit.BeforeClass;
|
||||
import org.junit.Rule;
|
||||
import org.junit.Test;
|
||||
import org.junit.runner.RunWith;
|
||||
import org.junit.runners.JUnit4;
|
||||
import org.mockito.ArgumentCaptor;
|
||||
import org.mockito.ArgumentMatcher;
|
||||
import org.mockito.Captor;
|
||||
import org.mockito.Mock;
|
||||
import org.mockito.MockitoAnnotations;
|
||||
|
||||
@RunWith(JUnit4.class)
|
||||
public class StatusBarManagerServiceTest {
|
||||
|
||||
private static final String TEST_PACKAGE = "test_pkg";
|
||||
private static final String TEST_SERVICE = "test_svc";
|
||||
private static final ComponentName TEST_COMPONENT = new ComponentName(TEST_PACKAGE,
|
||||
TEST_SERVICE);
|
||||
private static final CharSequence APP_NAME = "AppName";
|
||||
private static final CharSequence TILE_LABEL = "Tile label";
|
||||
|
||||
@Rule
|
||||
public final TestableContext mContext =
|
||||
new TestableContext(
|
||||
new NoBroadcastContextWrapper(InstrumentationRegistry.getContext()), null);
|
||||
|
||||
@Mock
|
||||
private ActivityTaskManagerInternal mActivityTaskManagerInternal;
|
||||
@Mock
|
||||
private PackageManagerInternal mPackageManagerInternal;
|
||||
@Mock
|
||||
private ActivityManagerInternal mActivityManagerInternal;
|
||||
@Mock
|
||||
private ApplicationInfo mApplicationInfo;
|
||||
@Mock
|
||||
private IStatusBar.Stub mMockStatusBar;
|
||||
@Captor
|
||||
private ArgumentCaptor<IAddTileResultCallback> mAddTileResultCallbackCaptor;
|
||||
|
||||
private Icon mIcon;
|
||||
private StatusBarManagerService mStatusBarManagerService;
|
||||
|
||||
@BeforeClass
|
||||
public static void oneTimeInitialization() {
|
||||
if (Looper.myLooper() == null) {
|
||||
Looper.prepare();
|
||||
}
|
||||
}
|
||||
|
||||
@Before
|
||||
public void setUp() {
|
||||
MockitoAnnotations.initMocks(this);
|
||||
LocalServices.removeServiceForTest(ActivityTaskManagerInternal.class);
|
||||
LocalServices.addService(ActivityTaskManagerInternal.class, mActivityTaskManagerInternal);
|
||||
LocalServices.removeServiceForTest(ActivityManagerInternal.class);
|
||||
LocalServices.addService(ActivityManagerInternal.class, mActivityManagerInternal);
|
||||
LocalServices.removeServiceForTest(PackageManagerInternal.class);
|
||||
LocalServices.addService(PackageManagerInternal.class, mPackageManagerInternal);
|
||||
|
||||
when(mMockStatusBar.asBinder()).thenReturn(mMockStatusBar);
|
||||
when(mApplicationInfo.loadLabel(any())).thenReturn(APP_NAME);
|
||||
|
||||
mStatusBarManagerService = new StatusBarManagerService(mContext);
|
||||
LocalServices.removeServiceForTest(StatusBarManagerInternal.class);
|
||||
LocalServices.removeServiceForTest(GlobalActionsProvider.class);
|
||||
|
||||
mContext.getSystemService(DisplayManager.class).unregisterDisplayListener(
|
||||
mStatusBarManagerService);
|
||||
|
||||
mStatusBarManagerService.registerStatusBar(mMockStatusBar);
|
||||
|
||||
mIcon = Icon.createWithResource(mContext, android.R.drawable.btn_plus);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testHandleIncomingUserCalled() {
|
||||
int fakeUser = 17;
|
||||
try {
|
||||
mStatusBarManagerService.requestAddTile(
|
||||
TEST_COMPONENT,
|
||||
TILE_LABEL,
|
||||
mIcon,
|
||||
fakeUser,
|
||||
new Callback()
|
||||
);
|
||||
fail("Should have SecurityException from uid check");
|
||||
} catch (SecurityException e) {
|
||||
verify(mActivityManagerInternal).handleIncomingUser(
|
||||
eq(Binder.getCallingPid()),
|
||||
eq(Binder.getCallingUid()),
|
||||
eq(fakeUser),
|
||||
eq(false),
|
||||
eq(ActivityManagerInternal.ALLOW_NON_FULL),
|
||||
anyString(),
|
||||
eq(TEST_PACKAGE)
|
||||
);
|
||||
}
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testCheckUid_pass() {
|
||||
when(mPackageManagerInternal.getPackageUid(TEST_PACKAGE, 0, mContext.getUserId()))
|
||||
.thenReturn(Binder.getCallingUid());
|
||||
try {
|
||||
mStatusBarManagerService.requestAddTile(
|
||||
TEST_COMPONENT,
|
||||
TILE_LABEL,
|
||||
mIcon,
|
||||
mContext.getUserId(),
|
||||
new Callback()
|
||||
);
|
||||
} catch (SecurityException e) {
|
||||
fail("No SecurityException should be thrown");
|
||||
}
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testCheckUid_pass_differentUser() {
|
||||
int otherUserUid = UserHandle.getUid(17, UserHandle.getAppId(Binder.getCallingUid()));
|
||||
when(mPackageManagerInternal.getPackageUid(TEST_PACKAGE, 0, mContext.getUserId()))
|
||||
.thenReturn(otherUserUid);
|
||||
try {
|
||||
mStatusBarManagerService.requestAddTile(
|
||||
TEST_COMPONENT,
|
||||
TILE_LABEL,
|
||||
mIcon,
|
||||
mContext.getUserId(),
|
||||
new Callback()
|
||||
);
|
||||
} catch (SecurityException e) {
|
||||
fail("No SecurityException should be thrown");
|
||||
}
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testCheckUid_fail() {
|
||||
when(mPackageManagerInternal.getPackageUid(TEST_PACKAGE, 0, mContext.getUserId()))
|
||||
.thenReturn(Binder.getCallingUid() + 1);
|
||||
try {
|
||||
mStatusBarManagerService.requestAddTile(
|
||||
TEST_COMPONENT,
|
||||
TILE_LABEL,
|
||||
mIcon,
|
||||
mContext.getUserId(),
|
||||
new Callback()
|
||||
);
|
||||
fail("Should throw SecurityException");
|
||||
} catch (SecurityException e) {
|
||||
// pass
|
||||
}
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testCurrentUser_fail() {
|
||||
mockUidCheck();
|
||||
int user = 0;
|
||||
when(mActivityManagerInternal.getCurrentUserId()).thenReturn(user + 1);
|
||||
|
||||
Callback callback = new Callback();
|
||||
mStatusBarManagerService.requestAddTile(TEST_COMPONENT, TILE_LABEL, mIcon, user, callback);
|
||||
|
||||
assertEquals(StatusBarManager.TILE_ADD_REQUEST_ERROR_NOT_CURRENT_USER,
|
||||
callback.mUserResponse);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testCurrentUser_pass() {
|
||||
mockUidCheck();
|
||||
int user = 0;
|
||||
when(mActivityManagerInternal.getCurrentUserId()).thenReturn(user);
|
||||
|
||||
Callback callback = new Callback();
|
||||
mStatusBarManagerService.requestAddTile(TEST_COMPONENT, TILE_LABEL, mIcon, user, callback);
|
||||
|
||||
assertNotEquals(StatusBarManager.TILE_ADD_REQUEST_ERROR_NOT_CURRENT_USER,
|
||||
callback.mUserResponse);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testValidComponent_fail_noComponentFound() {
|
||||
int user = 10;
|
||||
mockUidCheck();
|
||||
mockCurrentUserCheck(user);
|
||||
IntentMatcher im = new IntentMatcher(
|
||||
new Intent(TileService.ACTION_QS_TILE).setComponent(TEST_COMPONENT));
|
||||
when(mPackageManagerInternal.resolveService(argThat(im), nullable(String.class), eq(0),
|
||||
eq(user), anyInt())).thenReturn(null);
|
||||
|
||||
Callback callback = new Callback();
|
||||
mStatusBarManagerService.requestAddTile(TEST_COMPONENT, TILE_LABEL, mIcon, user, callback);
|
||||
|
||||
assertEquals(StatusBarManager.TILE_ADD_REQUEST_ERROR_BAD_COMPONENT, callback.mUserResponse);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testValidComponent_fail_notEnabled() {
|
||||
int user = 10;
|
||||
mockUidCheck();
|
||||
mockCurrentUserCheck(user);
|
||||
|
||||
ResolveInfo r = makeResolveInfo();
|
||||
r.serviceInfo.permission = Manifest.permission.BIND_QUICK_SETTINGS_TILE;
|
||||
|
||||
IntentMatcher im = new IntentMatcher(
|
||||
new Intent(TileService.ACTION_QS_TILE).setComponent(TEST_COMPONENT));
|
||||
when(mPackageManagerInternal.resolveService(argThat(im), nullable(String.class), eq(0),
|
||||
eq(user), anyInt())).thenReturn(r);
|
||||
when(mPackageManagerInternal.getComponentEnabledSetting(TEST_COMPONENT,
|
||||
Binder.getCallingUid(), user)).thenReturn(
|
||||
PackageManager.COMPONENT_ENABLED_STATE_DISABLED);
|
||||
|
||||
Callback callback = new Callback();
|
||||
mStatusBarManagerService.requestAddTile(TEST_COMPONENT, TILE_LABEL, mIcon, user, callback);
|
||||
|
||||
assertEquals(StatusBarManager.TILE_ADD_REQUEST_ERROR_BAD_COMPONENT, callback.mUserResponse);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testValidComponent_fail_noPermission() {
|
||||
int user = 10;
|
||||
mockUidCheck();
|
||||
mockCurrentUserCheck(user);
|
||||
|
||||
ResolveInfo r = makeResolveInfo();
|
||||
|
||||
IntentMatcher im = new IntentMatcher(
|
||||
new Intent(TileService.ACTION_QS_TILE).setComponent(TEST_COMPONENT));
|
||||
when(mPackageManagerInternal.resolveService(argThat(im), nullable(String.class), eq(0),
|
||||
eq(user), anyInt())).thenReturn(r);
|
||||
when(mPackageManagerInternal.getComponentEnabledSetting(TEST_COMPONENT,
|
||||
Binder.getCallingUid(), user)).thenReturn(
|
||||
PackageManager.COMPONENT_ENABLED_STATE_ENABLED);
|
||||
|
||||
Callback callback = new Callback();
|
||||
mStatusBarManagerService.requestAddTile(TEST_COMPONENT, TILE_LABEL, mIcon, user, callback);
|
||||
|
||||
assertEquals(StatusBarManager.TILE_ADD_REQUEST_ERROR_BAD_COMPONENT, callback.mUserResponse);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testValidComponent_fail_notExported() {
|
||||
int user = 10;
|
||||
mockUidCheck();
|
||||
mockCurrentUserCheck(user);
|
||||
|
||||
ResolveInfo r = makeResolveInfo();
|
||||
r.serviceInfo.permission = Manifest.permission.BIND_QUICK_SETTINGS_TILE;
|
||||
r.serviceInfo.exported = false;
|
||||
|
||||
IntentMatcher im = new IntentMatcher(
|
||||
new Intent(TileService.ACTION_QS_TILE).setComponent(TEST_COMPONENT));
|
||||
when(mPackageManagerInternal.resolveService(argThat(im), nullable(String.class), eq(0),
|
||||
eq(user), anyInt())).thenReturn(r);
|
||||
when(mPackageManagerInternal.getComponentEnabledSetting(TEST_COMPONENT,
|
||||
Binder.getCallingUid(), user)).thenReturn(
|
||||
PackageManager.COMPONENT_ENABLED_STATE_ENABLED);
|
||||
|
||||
Callback callback = new Callback();
|
||||
mStatusBarManagerService.requestAddTile(TEST_COMPONENT, TILE_LABEL, mIcon, user, callback);
|
||||
|
||||
assertEquals(StatusBarManager.TILE_ADD_REQUEST_ERROR_BAD_COMPONENT, callback.mUserResponse);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testValidComponent_pass() {
|
||||
int user = 10;
|
||||
mockUidCheck();
|
||||
mockCurrentUserCheck(user);
|
||||
|
||||
ResolveInfo r = makeResolveInfo();
|
||||
r.serviceInfo.permission = Manifest.permission.BIND_QUICK_SETTINGS_TILE;
|
||||
r.serviceInfo.exported = true;
|
||||
|
||||
IntentMatcher im = new IntentMatcher(
|
||||
new Intent(TileService.ACTION_QS_TILE).setComponent(TEST_COMPONENT));
|
||||
when(mPackageManagerInternal.resolveService(argThat(im), nullable(String.class), eq(0),
|
||||
eq(user), anyInt())).thenReturn(r);
|
||||
when(mPackageManagerInternal.getComponentEnabledSetting(TEST_COMPONENT,
|
||||
Binder.getCallingUid(), user)).thenReturn(
|
||||
PackageManager.COMPONENT_ENABLED_STATE_ENABLED);
|
||||
|
||||
Callback callback = new Callback();
|
||||
mStatusBarManagerService.requestAddTile(TEST_COMPONENT, TILE_LABEL, mIcon, user, callback);
|
||||
|
||||
assertNotEquals(StatusBarManager.TILE_ADD_REQUEST_ERROR_BAD_COMPONENT,
|
||||
callback.mUserResponse);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testAppInForeground_fail() {
|
||||
int user = 10;
|
||||
mockUidCheck();
|
||||
mockCurrentUserCheck(user);
|
||||
mockComponentInfo(user);
|
||||
|
||||
when(mActivityManagerInternal.getUidProcessState(Binder.getCallingUid())).thenReturn(
|
||||
PROCESS_STATE_FOREGROUND_SERVICE);
|
||||
|
||||
Callback callback = new Callback();
|
||||
mStatusBarManagerService.requestAddTile(TEST_COMPONENT, TILE_LABEL, mIcon, user, callback);
|
||||
|
||||
assertEquals(StatusBarManager.TILE_ADD_REQUEST_ERROR_APP_NOT_IN_FOREGROUND,
|
||||
callback.mUserResponse);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testAppInForeground_pass() {
|
||||
int user = 10;
|
||||
mockUidCheck();
|
||||
mockCurrentUserCheck(user);
|
||||
mockComponentInfo(user);
|
||||
|
||||
when(mActivityManagerInternal.getUidProcessState(Binder.getCallingUid())).thenReturn(
|
||||
PROCESS_STATE_TOP);
|
||||
|
||||
Callback callback = new Callback();
|
||||
mStatusBarManagerService.requestAddTile(TEST_COMPONENT, TILE_LABEL, mIcon, user, callback);
|
||||
|
||||
assertNotEquals(StatusBarManager.TILE_ADD_REQUEST_ERROR_APP_NOT_IN_FOREGROUND,
|
||||
callback.mUserResponse);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testRequestToStatusBar() throws RemoteException {
|
||||
int user = 10;
|
||||
mockEverything(user);
|
||||
|
||||
mStatusBarManagerService.requestAddTile(TEST_COMPONENT, TILE_LABEL, mIcon, user,
|
||||
new Callback());
|
||||
|
||||
verify(mMockStatusBar).requestAddTile(
|
||||
eq(TEST_COMPONENT),
|
||||
eq(APP_NAME),
|
||||
eq(TILE_LABEL),
|
||||
eq(mIcon),
|
||||
any()
|
||||
);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testRequestInProgress_samePackage() throws RemoteException {
|
||||
int user = 10;
|
||||
mockEverything(user);
|
||||
|
||||
mStatusBarManagerService.requestAddTile(TEST_COMPONENT, TILE_LABEL, mIcon, user,
|
||||
new Callback());
|
||||
|
||||
Callback callback = new Callback();
|
||||
mStatusBarManagerService.requestAddTile(TEST_COMPONENT, TILE_LABEL, mIcon, user, callback);
|
||||
|
||||
assertEquals(StatusBarManager.TILE_ADD_REQUEST_ERROR_REQUEST_IN_PROGRESS,
|
||||
callback.mUserResponse);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testRequestInProgress_differentPackage() throws RemoteException {
|
||||
int user = 10;
|
||||
mockEverything(user);
|
||||
ComponentName otherComponent = new ComponentName("a", "b");
|
||||
mockUidCheck(otherComponent.getPackageName());
|
||||
mockComponentInfo(user, otherComponent);
|
||||
|
||||
mStatusBarManagerService.requestAddTile(TEST_COMPONENT, TILE_LABEL, mIcon, user,
|
||||
new Callback());
|
||||
|
||||
Callback callback = new Callback();
|
||||
mStatusBarManagerService.requestAddTile(otherComponent, TILE_LABEL, mIcon, user, callback);
|
||||
|
||||
assertNotEquals(StatusBarManager.TILE_ADD_REQUEST_ERROR_REQUEST_IN_PROGRESS,
|
||||
callback.mUserResponse);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testResponseForwardedToCallback_tileAdded() throws RemoteException {
|
||||
int user = 10;
|
||||
mockEverything(user);
|
||||
|
||||
Callback callback = new Callback();
|
||||
mStatusBarManagerService.requestAddTile(TEST_COMPONENT, TILE_LABEL, mIcon, user, callback);
|
||||
|
||||
verify(mMockStatusBar).requestAddTile(
|
||||
eq(TEST_COMPONENT),
|
||||
eq(APP_NAME),
|
||||
eq(TILE_LABEL),
|
||||
eq(mIcon),
|
||||
mAddTileResultCallbackCaptor.capture()
|
||||
);
|
||||
|
||||
mAddTileResultCallbackCaptor.getValue().onTileRequest(
|
||||
StatusBarManager.TILE_ADD_REQUEST_RESULT_TILE_ADDED);
|
||||
assertEquals(StatusBarManager.TILE_ADD_REQUEST_RESULT_TILE_ADDED, callback.mUserResponse);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testResponseForwardedToCallback_tileNotAdded() throws RemoteException {
|
||||
int user = 10;
|
||||
mockEverything(user);
|
||||
|
||||
Callback callback = new Callback();
|
||||
mStatusBarManagerService.requestAddTile(TEST_COMPONENT, TILE_LABEL, mIcon, user, callback);
|
||||
|
||||
verify(mMockStatusBar).requestAddTile(
|
||||
eq(TEST_COMPONENT),
|
||||
eq(APP_NAME),
|
||||
eq(TILE_LABEL),
|
||||
eq(mIcon),
|
||||
mAddTileResultCallbackCaptor.capture()
|
||||
);
|
||||
|
||||
mAddTileResultCallbackCaptor.getValue().onTileRequest(
|
||||
StatusBarManager.TILE_ADD_REQUEST_RESULT_TILE_NOT_ADDED);
|
||||
assertEquals(StatusBarManager.TILE_ADD_REQUEST_RESULT_TILE_NOT_ADDED,
|
||||
callback.mUserResponse);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testResponseForwardedToCallback_tileAlreadyAdded() throws RemoteException {
|
||||
int user = 10;
|
||||
mockEverything(user);
|
||||
|
||||
Callback callback = new Callback();
|
||||
mStatusBarManagerService.requestAddTile(TEST_COMPONENT, TILE_LABEL, mIcon, user, callback);
|
||||
|
||||
verify(mMockStatusBar).requestAddTile(
|
||||
eq(TEST_COMPONENT),
|
||||
eq(APP_NAME),
|
||||
eq(TILE_LABEL),
|
||||
eq(mIcon),
|
||||
mAddTileResultCallbackCaptor.capture()
|
||||
);
|
||||
|
||||
mAddTileResultCallbackCaptor.getValue().onTileRequest(
|
||||
StatusBarManager.TILE_ADD_REQUEST_RESULT_TILE_ALREADY_ADDED);
|
||||
assertEquals(StatusBarManager.TILE_ADD_REQUEST_RESULT_TILE_ALREADY_ADDED,
|
||||
callback.mUserResponse);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testResponseForwardedToCallback_dialogDismissed() throws RemoteException {
|
||||
int user = 10;
|
||||
mockEverything(user);
|
||||
|
||||
Callback callback = new Callback();
|
||||
mStatusBarManagerService.requestAddTile(TEST_COMPONENT, TILE_LABEL, mIcon, user, callback);
|
||||
|
||||
verify(mMockStatusBar).requestAddTile(
|
||||
eq(TEST_COMPONENT),
|
||||
eq(APP_NAME),
|
||||
eq(TILE_LABEL),
|
||||
eq(mIcon),
|
||||
mAddTileResultCallbackCaptor.capture()
|
||||
);
|
||||
|
||||
mAddTileResultCallbackCaptor.getValue().onTileRequest(
|
||||
StatusBarManager.TILE_ADD_REQUEST_RESULT_DIALOG_DISMISSED);
|
||||
// This gets translated to TILE_NOT_ADDED
|
||||
assertEquals(StatusBarManager.TILE_ADD_REQUEST_RESULT_TILE_NOT_ADDED,
|
||||
callback.mUserResponse);
|
||||
}
|
||||
|
||||
private void mockUidCheck() {
|
||||
mockUidCheck(TEST_PACKAGE);
|
||||
}
|
||||
|
||||
private void mockUidCheck(String packageName) {
|
||||
when(mPackageManagerInternal.getPackageUid(eq(packageName), anyInt(), anyInt()))
|
||||
.thenReturn(Binder.getCallingUid());
|
||||
}
|
||||
|
||||
private void mockCurrentUserCheck(int user) {
|
||||
when(mActivityManagerInternal.getCurrentUserId()).thenReturn(user);
|
||||
}
|
||||
|
||||
private void mockComponentInfo(int user) {
|
||||
mockComponentInfo(user, TEST_COMPONENT);
|
||||
}
|
||||
|
||||
private ResolveInfo makeResolveInfo() {
|
||||
ResolveInfo r = new ResolveInfo();
|
||||
r.serviceInfo = new ServiceInfo();
|
||||
r.serviceInfo.applicationInfo = mApplicationInfo;
|
||||
return r;
|
||||
}
|
||||
|
||||
private void mockComponentInfo(int user, ComponentName componentName) {
|
||||
ResolveInfo r = makeResolveInfo();
|
||||
r.serviceInfo.exported = true;
|
||||
r.serviceInfo.permission = Manifest.permission.BIND_QUICK_SETTINGS_TILE;
|
||||
|
||||
IntentMatcher im = new IntentMatcher(
|
||||
new Intent(TileService.ACTION_QS_TILE).setComponent(componentName));
|
||||
when(mPackageManagerInternal.resolveService(argThat(im), nullable(String.class), eq(0),
|
||||
eq(user), anyInt())).thenReturn(r);
|
||||
when(mPackageManagerInternal.getComponentEnabledSetting(componentName,
|
||||
Binder.getCallingUid(), user)).thenReturn(
|
||||
PackageManager.COMPONENT_ENABLED_STATE_ENABLED);
|
||||
}
|
||||
|
||||
private void mockProcessState() {
|
||||
when(mActivityManagerInternal.getUidProcessState(Binder.getCallingUid())).thenReturn(
|
||||
PROCESS_STATE_TOP);
|
||||
}
|
||||
|
||||
private void mockEverything(int user) {
|
||||
mockUidCheck();
|
||||
mockCurrentUserCheck(user);
|
||||
mockComponentInfo(user);
|
||||
mockProcessState();
|
||||
}
|
||||
|
||||
private static class Callback extends IAddTileResultCallback.Stub {
|
||||
int mUserResponse = -1;
|
||||
|
||||
@Override
|
||||
public void onTileRequest(int userResponse) throws RemoteException {
|
||||
if (mUserResponse != -1) {
|
||||
throw new IllegalStateException(
|
||||
"Setting response to " + userResponse + " but it already has "
|
||||
+ mUserResponse);
|
||||
}
|
||||
mUserResponse = userResponse;
|
||||
}
|
||||
}
|
||||
|
||||
private static class IntentMatcher implements ArgumentMatcher<Intent> {
|
||||
private final Intent mIntent;
|
||||
|
||||
IntentMatcher(Intent intent) {
|
||||
mIntent = intent;
|
||||
}
|
||||
|
||||
@Override
|
||||
public boolean matches(Intent argument) {
|
||||
return argument != null && argument.filterEquals(mIntent);
|
||||
}
|
||||
|
||||
@Override
|
||||
public String toString() {
|
||||
return "Expected: " + mIntent;
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user