Replace direct usage of update origin by TestModeBuilder.setEnabled() overload

Bug: 333527800
Test: atest ZenModesListItemPreferenceTest
Flag: android.app.modes_ui
Change-Id: Ice6144b8a15641703495d18b57f5b56cda91e92a
This commit is contained in:
Matías Hernández
2024-08-02 19:01:04 +02:00
parent 7a1039fea4
commit 0721f5c5b2

View File

@@ -22,7 +22,6 @@ import android.app.Flags;
import android.content.Context; import android.content.Context;
import android.platform.test.annotations.EnableFlags; import android.platform.test.annotations.EnableFlags;
import android.platform.test.flag.junit.SetFlagsRule; import android.platform.test.flag.junit.SetFlagsRule;
import android.service.notification.ZenModeConfig;
import com.android.settingslib.notification.modes.TestModeBuilder; import com.android.settingslib.notification.modes.TestModeBuilder;
import com.android.settingslib.notification.modes.ZenMode; import com.android.settingslib.notification.modes.ZenMode;
@@ -94,14 +93,10 @@ public class ZenModesListItemPreferenceTest {
@Test @Test
public void setZenMode_modeDisabledByApp() { public void setZenMode_modeDisabledByApp() {
ZenModeConfig.ZenRule configRule = new ZenModeConfig.ZenRule();
configRule.enabled = false;
configRule.disabledOrigin = ZenModeConfig.UPDATE_ORIGIN_APP;
ZenMode mode = new TestModeBuilder() ZenMode mode = new TestModeBuilder()
.setName("Mode disabled by app") .setName("Mode disabled by app")
.setTriggerDescription("When the cat's away") .setTriggerDescription("When the cat's away")
.setEnabled(false) .setEnabled(false, /* byUser= */ false)
.setConfigZenRule(configRule)
.build(); .build();
ZenModesListItemPreference preference = new ZenModesListItemPreference(mContext, mode); ZenModesListItemPreference preference = new ZenModesListItemPreference(mContext, mode);
@@ -114,14 +109,10 @@ public class ZenModesListItemPreferenceTest {
@Test @Test
public void setZenMode_modeDisabledByUser() { public void setZenMode_modeDisabledByUser() {
ZenModeConfig.ZenRule configRule = new ZenModeConfig.ZenRule();
configRule.enabled = false;
configRule.disabledOrigin = ZenModeConfig.UPDATE_ORIGIN_USER;
ZenMode mode = new TestModeBuilder() ZenMode mode = new TestModeBuilder()
.setName("Mode disabled by user") .setName("Mode disabled by user")
.setTriggerDescription("When the Levee Breaks") .setTriggerDescription("When the Levee Breaks")
.setEnabled(false) .setEnabled(false, /* byUser= */ true)
.setConfigZenRule(configRule)
.build(); .build();
ZenModesListItemPreference preference = new ZenModesListItemPreference(mContext, mode); ZenModesListItemPreference preference = new ZenModesListItemPreference(mContext, mode);