package android.platform.test.rule

import android.platform.uiautomatorhelpers.DeviceHelpers.context
import android.provider.Settings

/** Base rule to set values in [Settings.Secure]. The value is then reset at the end of the test. */
open class SecureSettingRule<T : Any>(private val settingName: String, initialValue: T? = null) :
    SettingRule<T>(initialValue) {

    override fun getSettingValueAsString(): String? =
        Settings.Secure.getString(context.contentResolver, settingName)

    override fun setSettingValueAsString(value: String?) {
        Settings.Secure.putString(context.contentResolver, settingName, value)
    }
}