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

Commit 8a4b4494 authored by Behnam Heydarshahi's avatar Behnam Heydarshahi Committed by Android (Google) Code Review
Browse files

Merge "Migrate Work Mode Tile" into main

parents 375471bb e59a3f01

packages/SystemUI/multivalentTests/src/com/android/systemui/qs/tiles/impl/work/domain/interactor/WorkModeTileDataInteractorTest.kt

0 → 100644
+93 −0
Original line number Diff line number Diff line 
/*

 * 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.systemui.qs.tiles.impl.work.domain.interactor



import android.os.UserHandle

import android.platform.test.annotations.EnabledOnRavenwood

import android.testing.LeakCheck

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

import androidx.test.filters.SmallTest

import com.android.systemui.SysuiTestCase

import com.android.systemui.coroutines.collectLastValue

import com.android.systemui.qs.tiles.base.interactor.DataUpdateTrigger

import com.android.systemui.qs.tiles.impl.work.domain.model.WorkModeTileModel

import com.android.systemui.utils.leaks.FakeManagedProfileController

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

import kotlinx.coroutines.flow.flowOf

import kotlinx.coroutines.test.runTest

import org.junit.Test

import org.junit.runner.RunWith



@SmallTest

@EnabledOnRavenwood

@RunWith(AndroidJUnit4::class)

class WorkModeTileDataInteractorTest : SysuiTestCase() {

    private val controller = FakeManagedProfileController(LeakCheck())

    private val underTest: WorkModeTileDataInteractor = WorkModeTileDataInteractor(controller)



    @Test

    fun availability_matchesControllerHasActiveProfiles() = runTest {

        val availability by collectLastValue(underTest.availability(TEST_USER))



        assertThat(availability).isFalse()



        controller.setHasActiveProfile(true)

        assertThat(availability).isTrue()



        controller.setHasActiveProfile(false)

        assertThat(availability).isFalse()

    }



    @Test

    fun tileData_whenHasActiveProfile_matchesControllerIsEnabled() = runTest {

        controller.setHasActiveProfile(true)

        val data by

            collectLastValue(

                underTest.tileData(TEST_USER, flowOf(DataUpdateTrigger.InitialRequest))

            )



        assertThat(data).isInstanceOf(WorkModeTileModel.HasActiveProfile::class.java)

        assertThat((data as WorkModeTileModel.HasActiveProfile).isEnabled).isFalse()



        controller.isWorkModeEnabled = true

        assertThat(data).isInstanceOf(WorkModeTileModel.HasActiveProfile::class.java)

        assertThat((data as WorkModeTileModel.HasActiveProfile).isEnabled).isTrue()



        controller.isWorkModeEnabled = false

        assertThat(data).isInstanceOf(WorkModeTileModel.HasActiveProfile::class.java)

        assertThat((data as WorkModeTileModel.HasActiveProfile).isEnabled).isFalse()

    }



    @Test

    fun tileData_matchesControllerHasActiveProfile() = runTest {

        val data by

            collectLastValue(

                underTest.tileData(TEST_USER, flowOf(DataUpdateTrigger.InitialRequest))

            )

        assertThat(data).isInstanceOf(WorkModeTileModel.NoActiveProfile::class.java)



        controller.setHasActiveProfile(true)

        assertThat(data).isInstanceOf(WorkModeTileModel.HasActiveProfile::class.java)



        controller.setHasActiveProfile(false)

        assertThat(data).isInstanceOf(WorkModeTileModel.NoActiveProfile::class.java)

    }



    private companion object {

        val TEST_USER = UserHandle.of(1)!!

    }

}

packages/SystemUI/multivalentTests/src/com/android/systemui/qs/tiles/impl/work/domain/interactor/WorkModeTileUserActionInteractorTest.kt

0 → 100644
+115 −0
Original line number Diff line number Diff line 
/*

 * 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.systemui.qs.tiles.impl.work.domain.interactor



import android.platform.test.annotations.EnabledOnRavenwood

import android.provider.Settings

import android.testing.LeakCheck

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

import androidx.test.filters.SmallTest

import com.android.systemui.SysuiTestCase

import com.android.systemui.qs.tiles.base.actions.FakeQSTileIntentUserInputHandler

import com.android.systemui.qs.tiles.base.actions.QSTileIntentUserInputHandlerSubject

import com.android.systemui.qs.tiles.base.interactor.QSTileInputTestKtx

import com.android.systemui.qs.tiles.impl.work.domain.model.WorkModeTileModel

import com.android.systemui.utils.leaks.FakeManagedProfileController

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

import kotlinx.coroutines.test.runTest

import org.junit.Test

import org.junit.runner.RunWith



@SmallTest

@EnabledOnRavenwood

@RunWith(AndroidJUnit4::class)

class WorkModeTileUserActionInteractorTest : SysuiTestCase() {



    private val inputHandler = FakeQSTileIntentUserInputHandler()

    private val profileController = FakeManagedProfileController(LeakCheck())



    private val underTest =

        WorkModeTileUserActionInteractor(

            profileController,

            inputHandler,

        )



    @Test

    fun handleClickWhenEnabled() = runTest {

        val wasEnabled = true

        profileController.isWorkModeEnabled = wasEnabled



        underTest.handleInput(

            QSTileInputTestKtx.click(WorkModeTileModel.HasActiveProfile(wasEnabled))

        )



        assertThat(profileController.isWorkModeEnabled).isEqualTo(!wasEnabled)

    }



    @Test

    fun handleClickWhenDisabled() = runTest {

        val wasEnabled = false

        profileController.isWorkModeEnabled = wasEnabled



        underTest.handleInput(

            QSTileInputTestKtx.click(WorkModeTileModel.HasActiveProfile(wasEnabled))

        )



        assertThat(profileController.isWorkModeEnabled).isEqualTo(!wasEnabled)

    }



    @Test

    fun handleClickWhenUnavailable() = runTest {

        val wasEnabled = false

        profileController.isWorkModeEnabled = wasEnabled



        underTest.handleInput(QSTileInputTestKtx.click(WorkModeTileModel.NoActiveProfile))



        assertThat(profileController.isWorkModeEnabled).isEqualTo(wasEnabled)

    }



    @Test

    fun handleLongClickWhenDisabled() = runTest {

        val enabled = false



        underTest.handleInput(

            QSTileInputTestKtx.longClick(WorkModeTileModel.HasActiveProfile(enabled))

        )



        QSTileIntentUserInputHandlerSubject.assertThat(inputHandler).handledOneIntentInput {

            assertThat(it.intent.action).isEqualTo(Settings.ACTION_MANAGED_PROFILE_SETTINGS)

        }

    }



    @Test

    fun handleLongClickWhenEnabled() = runTest {

        val enabled = true



        underTest.handleInput(

            QSTileInputTestKtx.longClick(WorkModeTileModel.HasActiveProfile(enabled))

        )



        QSTileIntentUserInputHandlerSubject.assertThat(inputHandler).handledOneIntentInput {

            assertThat(it.intent.action).isEqualTo(Settings.ACTION_MANAGED_PROFILE_SETTINGS)

        }

    }



    @Test

    fun handleLongClickWhenUnavailable() = runTest {

        underTest.handleInput(QSTileInputTestKtx.longClick(WorkModeTileModel.NoActiveProfile))



        QSTileIntentUserInputHandlerSubject.assertThat(inputHandler).handledNoInputs()

    }

}

packages/SystemUI/src/com/android/systemui/qs/tiles/impl/work/domain/interactor/WorkModeTileDataInteractor.kt

0 → 100644
+49 −0
Original line number Diff line number Diff line 
/*

 * 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.systemui.qs.tiles.impl.work.domain.interactor



import android.os.UserHandle

import com.android.systemui.qs.tiles.base.interactor.DataUpdateTrigger

import com.android.systemui.qs.tiles.base.interactor.QSTileDataInteractor

import com.android.systemui.qs.tiles.impl.work.domain.model.WorkModeTileModel

import com.android.systemui.statusbar.phone.ManagedProfileController

import com.android.systemui.util.kotlin.hasActiveWorkProfile

import javax.inject.Inject

import kotlinx.coroutines.flow.Flow

import kotlinx.coroutines.flow.map



/** Observes data saver state changes providing the [WorkModeTileModel]. */

class WorkModeTileDataInteractor

@Inject

constructor(

    private val profileController: ManagedProfileController,

) : QSTileDataInteractor<WorkModeTileModel> {

    override fun tileData(

        user: UserHandle,

        triggers: Flow<DataUpdateTrigger>

    ): Flow<WorkModeTileModel> =

        profileController.hasActiveWorkProfile.map { hasActiveWorkProfile: Boolean ->

            if (hasActiveWorkProfile) {

                WorkModeTileModel.HasActiveProfile(profileController.isWorkModeEnabled)

            } else {

                WorkModeTileModel.NoActiveProfile

            }

        }



    override fun availability(user: UserHandle): Flow<Boolean> =

        profileController.hasActiveWorkProfile

}

packages/SystemUI/src/com/android/systemui/qs/tiles/impl/work/domain/interactor/WorkModeTileUserActionInteractor.kt

0 → 100644
+54 −0
Original line number Diff line number Diff line 
/*

 * 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.systemui.qs.tiles.impl.work.domain.interactor



import android.content.Intent

import android.provider.Settings

import com.android.systemui.qs.tiles.base.actions.QSTileIntentUserInputHandler

import com.android.systemui.qs.tiles.base.interactor.QSTileInput

import com.android.systemui.qs.tiles.base.interactor.QSTileUserActionInteractor

import com.android.systemui.qs.tiles.impl.work.domain.model.WorkModeTileModel

import com.android.systemui.qs.tiles.viewmodel.QSTileUserAction

import com.android.systemui.statusbar.phone.ManagedProfileController

import javax.inject.Inject



/** Handles airplane mode tile clicks and long clicks. */

class WorkModeTileUserActionInteractor

@Inject

constructor(

    private val profileController: ManagedProfileController,

    private val qsTileIntentUserActionHandler: QSTileIntentUserInputHandler,

) : QSTileUserActionInteractor<WorkModeTileModel> {

    override suspend fun handleInput(input: QSTileInput<WorkModeTileModel>) =

        with(input) {

            when (action) {

                is QSTileUserAction.Click -> {

                    if (data is WorkModeTileModel.HasActiveProfile) {

                        profileController.setWorkModeEnabled(!data.isEnabled)

                    }

                }

                is QSTileUserAction.LongClick -> {

                    if (data is WorkModeTileModel.HasActiveProfile) {

                        qsTileIntentUserActionHandler.handle(

                            action.view,

                            Intent(Settings.ACTION_MANAGED_PROFILE_SETTINGS)

                        )

                    }

                }

            }

        }

}

packages/SystemUI/src/com/android/systemui/qs/tiles/impl/work/domain/model/WorkModeTileModel.kt

0 → 100644
+24 −0
Original line number Diff line number Diff line 
/*

 * 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.systemui.qs.tiles.impl.work.domain.model



/** Work mode tile model. */

sealed interface WorkModeTileModel {

    /** @param isEnabled is true when the work mode is enabled */

    data class HasActiveProfile(val isEnabled: Boolean) : WorkModeTileModel

    data object NoActiveProfile : WorkModeTileModel

}