Merge "Part 1 of `use bluetooth` toggle catalyst migration." into main

/*

 * Copyright (C) 2024 The Android Open Source Project

 *

 * Licensed under the Apache License, Version 2.0 (the "License");

 * you may not use this file except in compliance with the License.

 * You may obtain a copy of the License at

 *

 *      http://www.apache.org/licenses/LICENSE-2.0

 *

 * Unless required by applicable law or agreed to in writing, software

 * distributed under the License is distributed on an "AS IS" BASIS,

 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

 * See the License for the specific language governing permissions and

 * limitations under the License.

 */

package com.android.settings.connecteddevice

import android.bluetooth.BluetoothAdapter

import android.content.BroadcastReceiver

import android.content.Context

import android.content.Intent

import android.content.IntentFilter

import com.android.settings.R

import com.android.settings.widget.MainSwitchBarMetadata

import com.android.settingslib.datastore.KeyValueStore

import com.android.settingslib.datastore.NoOpKeyedObservable

import com.android.settingslib.metadata.PreferenceLifecycleContext

import com.android.settingslib.metadata.PreferenceLifecycleProvider

import com.android.settingslib.metadata.ReadWritePermit

class BluetoothMainSwitchPreference(private val bluetoothAdapter: BluetoothAdapter?) :

    MainSwitchBarMetadata, PreferenceLifecycleProvider {

    private lateinit var broadcastReceiver: BroadcastReceiver

    override val key

        get() = "use_bluetooth"

    override val title

        get() = R.string.bluetooth_main_switch_title

    override fun getReadPermit(context: Context, myUid: Int, callingUid: Int) =

        ReadWritePermit.ALLOW

    override fun getWritePermit(context: Context, value: Boolean?, myUid: Int, callingUid: Int) =

        ReadWritePermit.ALLOW

    override fun storage(context: Context) = BluetoothStateStore(bluetoothAdapter)

    override fun onStart(context: PreferenceLifecycleContext) {

        broadcastReceiver =

            object : BroadcastReceiver() {

                override fun onReceive(receiverContext: Context, intent: Intent) {

                    context.notifyPreferenceChange(key)

                }

            }

        context.registerReceiver(

            broadcastReceiver,

            IntentFilter(BluetoothAdapter.ACTION_STATE_CHANGED),

            Context.RECEIVER_EXPORTED_UNAUDITED

        )

    }

    override fun onStop(context: PreferenceLifecycleContext) {

        if (::broadcastReceiver.isInitialized) {

            context.unregisterReceiver(broadcastReceiver)

        }

    }

    override fun isEnabled(context: Context): Boolean {

        return bluetoothAdapter?.state.let {

            it == BluetoothAdapter.STATE_ON || it == BluetoothAdapter.STATE_OFF

        }

    }

    @Suppress("UNCHECKED_CAST")

    class BluetoothStateStore(private val bluetoothAdapter: BluetoothAdapter?) :

        NoOpKeyedObservable<String>(), KeyValueStore {

        override fun contains(key: String) = true

        override fun <T : Any> getValue(key: String, valueType: Class<T>): T? {

            return (bluetoothAdapter?.state.let {

                it == BluetoothAdapter.STATE_ON || it == BluetoothAdapter.STATE_TURNING_ON

            }) as T

        }

        override fun <T : Any> setValue(key: String, valueType: Class<T>, value: T?) {

            if (value is Boolean) {

                if (value) {

                    bluetoothAdapter?.enable()

                } else {

                    bluetoothAdapter?.disable()

                }

            }

        }

    }

}

        assertThat(preferenceScreenCreator.key).isEqualTo(BluetoothDashboardScreen.KEY)

    }

    override fun migration() {}

    override fun launchFragment(

        fragmentClass: Class<PreferenceFragmentCompat>,

        action: (PreferenceFragmentCompat) -> Unit,

/*

 * Copyright (C) 2024 The Android Open Source Project

 *

 * Licensed under the Apache License, Version 2.0 (the "License");

 * you may not use this file except in compliance with the License.

 * You may obtain a copy of the License at

 *

 *      http://www.apache.org/licenses/LICENSE-2.0

 *

 * Unless required by applicable law or agreed to in writing, software

 * distributed under the License is distributed on an "AS IS" BASIS,

 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

 * See the License for the specific language governing permissions and

 * limitations under the License.

 */

package com.android.settings.connecteddevice

import android.bluetooth.BluetoothAdapter

import android.content.Context

import android.platform.test.flag.junit.SetFlagsRule

import androidx.test.core.app.ApplicationProvider

import androidx.test.ext.junit.runners.AndroidJUnit4

import com.google.common.truth.Truth.assertThat

import org.junit.Before

import org.junit.Rule

import org.junit.Test

import org.junit.runner.RunWith

import org.mockito.kotlin.spy

import org.mockito.kotlin.verify

import org.mockito.kotlin.whenever

import org.robolectric.RuntimeEnvironment

@RunWith(AndroidJUnit4::class)

class BluetoothMainSwitchPreferenceTest {

    @get:Rule

    val setFlagsRule = SetFlagsRule()

    private val context: Context = ApplicationProvider.getApplicationContext()

    private lateinit var bluetoothAdapter: BluetoothAdapter

    private lateinit var bluetoothMainSwitchPreference: BluetoothMainSwitchPreference

    @Before

    fun setUp() {

        bluetoothAdapter = spy(

            BluetoothAdapter.getDefaultAdapter()

        )

        whenever(bluetoothAdapter.state).thenReturn(BluetoothAdapter.STATE_ON)

        bluetoothMainSwitchPreference = BluetoothMainSwitchPreference(bluetoothAdapter)

    }

    @Test

    fun isEnabled_bluetoothOn_returnTrue() {

        assertThat(bluetoothMainSwitchPreference.isEnabled(context)).isTrue()

    }

    @Test

    fun isEnabled_bluetoothTurningOn_returnFalse() {

        whenever(bluetoothAdapter.state).thenReturn(BluetoothAdapter.STATE_TURNING_ON)

        assertThat(bluetoothMainSwitchPreference.isEnabled(context)).isFalse()

    }

    @Test

    fun storageSetOff_turnOff() {

        bluetoothMainSwitchPreference.storage(context).setValue(

            bluetoothMainSwitchPreference.key, Boolean::class.javaObjectType, false

        )

        verify(bluetoothAdapter).disable()

    }

    @Test

    fun storageSetOn_turnOn() {

        bluetoothMainSwitchPreference.storage(context).setValue(

            bluetoothMainSwitchPreference.key, Boolean::class.javaObjectType, true

        )

        verify(bluetoothAdapter).enable()

    }

}

 No newline at end of file