Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit fd6a4147 authored by Jacky Wang's avatar Jacky Wang
Browse files

Allow start custom activity for catalyst test 

Bug: 370394101
Flag: EXEMPT test
Test: atest
Change-Id: I38e3954a607e2fbb8525abc46b59b34e58d224cf
parent 0a884b6d

packages/SettingsLib/Preference/testutils/com/android/settingslib/preference/CatalystScreenTestCase.kt

+8 −5
Original line number Diff line number Diff line
@@ -99,15 +99,18 @@ abstract class CatalystScreenTestCase { 
        @Suppress("UNCHECKED_CAST")

        val clazz = preferenceScreenCreator.fragmentClass() as Class<PreferenceFragmentCompat>

        val builder = StringBuilder()

        launchFragmentScenario(clazz).use {

            it.onFragment { fragment ->

        launchFragment(clazz) { fragment ->

            taskFinished.set(true)

            fragment.preferenceScreen.toString(builder)

        }

        }

        return builder.toString()

    }



    protected open fun launchFragment(

        fragmentClass: Class<PreferenceFragmentCompat>,

        action: (PreferenceFragmentCompat) -> Unit,

    ): Unit = launchFragmentScenario(fragmentClass).use { it.onFragment(action) }



    protected open fun launchFragmentScenario(fragmentClass: Class<PreferenceFragmentCompat>) =

        FragmentScenario.launch(fragmentClass)