diff options
| -rwxr-xr-x | services/tests/uiservicestests/src/com/android/server/notification/NotificationManagerServiceTest.java | 38 |
1 files changed, 38 insertions, 0 deletions
diff --git a/services/tests/uiservicestests/src/com/android/server/notification/NotificationManagerServiceTest.java b/services/tests/uiservicestests/src/com/android/server/notification/NotificationManagerServiceTest.java index 4d0cfbc6ad11..21de668e10c3 100755 --- a/services/tests/uiservicestests/src/com/android/server/notification/NotificationManagerServiceTest.java +++ b/services/tests/uiservicestests/src/com/android/server/notification/NotificationManagerServiceTest.java @@ -57,6 +57,7 @@ import static junit.framework.Assert.assertNull; import static junit.framework.Assert.assertTrue; import static junit.framework.Assert.fail; +import static org.mockito.ArgumentMatchers.isNull; import static org.mockito.Matchers.anyBoolean; import static org.mockito.Matchers.anyLong; import static org.mockito.Matchers.anyString; @@ -658,6 +659,43 @@ public class NotificationManagerServiceTest extends UiServiceTestCase { } @Test + public void testDefaultAssistant_overrideDefault() { + final int userId = 0; + final String testComponent = "package/class"; + final List<UserInfo> userInfos = new ArrayList<>(); + userInfos.add(new UserInfo(0, "", 0)); + final ArraySet<ComponentName> validAssistants = new ArraySet<>(); + validAssistants.add(ComponentName.unflattenFromString(testComponent)); + final String originalComponent = DeviceConfig.getProperty( + DeviceConfig.NAMESPACE_SYSTEMUI, + SystemUiDeviceConfigFlags.NAS_DEFAULT_SERVICE + ); + DeviceConfig.setProperty( + DeviceConfig.NAMESPACE_SYSTEMUI, + SystemUiDeviceConfigFlags.NAS_DEFAULT_SERVICE, + testComponent, + false + ); + when(mActivityManager.isLowRamDevice()).thenReturn(false); + when(mAssistants.queryPackageForServices(isNull(), anyInt(), anyInt())) + .thenReturn(validAssistants); + when(mAssistants.getDefaultComponents()).thenReturn(new ArraySet<>()); + when(mUm.getEnabledProfiles(anyInt())).thenReturn(userInfos); + + mService.setDefaultAssistantForUser(userId); + + verify(mAssistants).setPackageOrComponentEnabled( + eq(testComponent), eq(userId), eq(true), eq(true)); + + DeviceConfig.setProperty( + DeviceConfig.NAMESPACE_SYSTEMUI, + SystemUiDeviceConfigFlags.NAS_DEFAULT_SERVICE, + originalComponent, + false + ); + } + + @Test public void testCreateNotificationChannels_SingleChannel() throws Exception { final NotificationChannel channel = new NotificationChannel("id", "name", IMPORTANCE_DEFAULT); |