Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit fea80252 authored by Jay Aliomer's avatar Jay Aliomer
Browse files

Unit tests for Override default assistant 

unit test were removed by acccident in CL https://googleplex-android-review.googlesource.com/c/platform/frameworks/base/+/9764495

Test: atest NotificationManagerServiceTest
Change-Id: I6de44390c6225a68ecf58fde278733307d7f70ab
parent a91e08f6

services/tests/uiservicestests/src/com/android/server/notification/NotificationManagerServiceTest.java

+38 −0
Original line number Diff line number Diff line
@@ -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;
@@ -657,6 +658,43 @@ public class NotificationManagerServiceTest extends UiServiceTestCase { 
        return nrSummary;

    }



    @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 =