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

Commit 35754a0c authored by Julia Reynolds's avatar Julia Reynolds
Browse files

Update tests for new adjustment key defaults 

Test: NotificationAssistantsTest
Fixes: 423894558
Flag: android.app.nm_summarization_onboarding_ui
Change-Id: I8fd6dd30f9071cf7f8d61af71ffeca5fd77709af
parent b94ffc19

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

+11 −3
Original line number Diff line number Diff line
@@ -104,6 +104,8 @@ import java.util.ArrayList; 
import java.util.Arrays;

import java.util.List;

import java.util.Map;

import java.util.function.Predicate;

import java.util.stream.Stream;



@RunWith(AndroidJUnit4.class)

public class NotificationAssistantsTest extends UiServiceTestCase {
@@ -736,8 +738,8 @@ public class NotificationAssistantsTest extends UiServiceTestCase { 
        assertThat(mAssistants.isAdjustmentAllowed(11, Adjustment.KEY_TYPE)).isFalse();



        // Check this doesn't apply to other adjustments if they default to allowed

        mAssistants.allowAdjustmentType(mZero.id, Adjustment.KEY_SUMMARIZATION);

        assertThat(mAssistants.isAdjustmentAllowed(11, Adjustment.KEY_SUMMARIZATION)).isTrue();

        mAssistants.allowAdjustmentType(mZero.id, KEY_IMPORTANCE);

        assertThat(mAssistants.isAdjustmentAllowed(11, Adjustment.KEY_IMPORTANCE)).isTrue();



        // now turn on classification for the profile user directly

        mAssistants.allowAdjustmentType(11, Adjustment.KEY_TYPE);
@@ -818,8 +820,14 @@ public class NotificationAssistantsTest extends UiServiceTestCase { 


        writeXmlAndReload(USER_ALL);



        ArrayList<String> expected = new ArrayList<>(List.of(DEFAULT_ALLOWED_ADJUSTMENTS));



        if (android.app.Flags.nmSummarizationOnboardingUi()) {

            expected.remove(KEY_SUMMARIZATION);

        }



        assertThat(mAssistants.getAllowedAssistantAdjustments(mZero.id))

                .containsExactlyElementsIn(DEFAULT_ALLOWED_ADJUSTMENTS);

                .containsExactlyElementsIn(expected);

    }



    @Test