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
Loading
Loading
Loading
Loading
+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