From 76b2fcbe2381d8c72faa8fd23f28f51b2ff5b07f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Mat=C3=ADas=20Hern=C3=A1ndez?= <matiashe@google.com>
Date: Tue, 7 Mar 2023 16:29:56 +0100
Subject: [PATCH] Allow tests to verify the result of
 PrimarySwitchPreference.setSwitchEnabled()

Bug: 265064188
Test: atest AppChannelsBypassingDndPreferenceControllerTest
Change-Id: Ic6f921fc0d0d6da03b05be8b2c2b0439241ebdd9
---
 .../com/android/settingslib/PrimarySwitchPreference.java    | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/packages/SettingsLib/src/com/android/settingslib/PrimarySwitchPreference.java b/packages/SettingsLib/src/com/android/settingslib/PrimarySwitchPreference.java
index fb06976ebfe3e..3ec5ebad0de64 100644
--- a/packages/SettingsLib/src/com/android/settingslib/PrimarySwitchPreference.java
+++ b/packages/SettingsLib/src/com/android/settingslib/PrimarySwitchPreference.java
@@ -23,6 +23,7 @@ import android.widget.Switch;
 
 import androidx.annotation.Keep;
 import androidx.annotation.Nullable;
+import androidx.annotation.VisibleForTesting;
 import androidx.preference.PreferenceViewHolder;
 
 import com.android.settingslib.RestrictedLockUtils.EnforcedAdmin;
@@ -132,6 +133,11 @@ public class PrimarySwitchPreference extends RestrictedPreference {
         }
     }
 
+    @VisibleForTesting(otherwise = VisibleForTesting.NONE)
+    public boolean isSwitchEnabled() {
+        return mEnableSwitch;
+    }
+
     /**
      * If admin is not null, disables the switch.
      * Otherwise, keep it enabled.