Merge "Fix Flaky UiModeManagerServiceTest" into main

import java.time.LocalDateTime;

import java.time.LocalTime;

import java.time.ZoneId;

import java.util.LinkedHashMap;

import java.util.List;

import java.util.Map;

import java.util.function.Consumer;

                any(AlarmManager.OnAlarmListener.class), any(Handler.class));

        doAnswer(inv -> {

            mCustomListener = () -> {};

            mCustomListener = () -> {

            };

            return null;

        }).when(mAlarmManager).cancel(eq(mCustomListener));

        when(mContext.getSystemService(eq(Context.POWER_SERVICE)))

            assertFalse(mUiManagerService.getConfiguration().isNightModeActive());

        }

        // attention modes with expected night modes

        Map<Integer, Boolean> modes = Map.of(

                MODE_ATTENTION_THEME_OVERLAY_OFF, initialNightMode,

                MODE_ATTENTION_THEME_OVERLAY_DAY, false,

                MODE_ATTENTION_THEME_OVERLAY_NIGHT, true

        );

        // Attention modes with expected night modes.

        // Important to keep modes.put(MODE_ATTENTION_THEME_OVERLAY_OFF, initialNightMode) in the

        // first position, hence LinkedHashMap.

        Map<Integer, Boolean> modes = new LinkedHashMap<>();

        modes.put(MODE_ATTENTION_THEME_OVERLAY_OFF, initialNightMode);

        modes.put(MODE_ATTENTION_THEME_OVERLAY_DAY, false);

        modes.put(MODE_ATTENTION_THEME_OVERLAY_NIGHT, true);

        // test

        for (int attentionMode : modes.keySet()) {