From 6d9de7840f117cd68cb29e97848b9f8cfd96276f Mon Sep 17 00:00:00 2001
From: Jacky Wang <jiannan@google.com>
Date: Mon, 23 Dec 2024 10:20:39 +0800
Subject: [PATCH] [Catalyst] Simplify RemoveAnimationsPreferenceTest

Bug: 373451690
Flag: TEST_ONLY
Test: atest
Change-Id: Idc64a811d97b5fd3047ad4167b98da5b97c0fee0
---
 .../accessibility/RemoveAnimationsPreferenceTest.kt   | 11 ++---------
 1 file changed, 2 insertions(+), 9 deletions(-)

diff --git a/tests/robotests/src/com/android/settings/accessibility/RemoveAnimationsPreferenceTest.kt b/tests/robotests/src/com/android/settings/accessibility/RemoveAnimationsPreferenceTest.kt
index 294632ff482..9df7ce72b75 100644
--- a/tests/robotests/src/com/android/settings/accessibility/RemoveAnimationsPreferenceTest.kt
+++ b/tests/robotests/src/com/android/settings/accessibility/RemoveAnimationsPreferenceTest.kt
@@ -20,11 +20,9 @@ import android.content.Context
 import androidx.preference.SwitchPreferenceCompat
 import androidx.test.core.app.ApplicationProvider
 import androidx.test.ext.junit.runners.AndroidJUnit4
-import com.android.settings.R
 import com.android.settings.accessibility.RemoveAnimationsPreference.Companion.ANIMATION_ON_VALUE
 import com.android.settingslib.datastore.SettingsGlobalStore
-import com.android.settingslib.preference.PreferenceScreenBindingHelper
-import com.android.settingslib.preference.PreferenceScreenFactory
+import com.android.settingslib.preference.createAndBindWidget
 import com.google.common.truth.Truth.assertThat
 import org.junit.Test
 import org.junit.runner.RunWith
@@ -35,12 +33,7 @@ class RemoveAnimationsPreferenceTest {
     private val appContext: Context = ApplicationProvider.getApplicationContext()
 
     private fun getRemoveAnimationsSwitchPreference(): SwitchPreferenceCompat =
-        PreferenceScreenFactory(appContext).let {
-            val preferenceScreen = it.inflate(R.xml.accessibility_color_and_motion)!!
-            it.preferenceManager.setPreferences(preferenceScreen)
-            PreferenceScreenBindingHelper.bind(preferenceScreen)
-            preferenceScreen.findPreference(RemoveAnimationsPreference.KEY)!!
-        }
+        RemoveAnimationsPreference().createAndBindWidget(appContext)
 
     @Test
     fun animationOff_switchPreferenceIsChecked() {