/*
 * Copyright (C) 2025 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.biometrics.face

import android.app.Activity
import android.content.Intent
import com.android.settings.overlay.FeatureFactory
import com.android.settings.testutils.FakeFeatureFactory
import com.google.common.truth.Truth.assertThat
import org.junit.Before
import org.junit.Test
import org.junit.runner.RunWith
import org.mockito.Mockito.`when` as whenever
import org.robolectric.Robolectric
import org.robolectric.RobolectricTestRunner
import org.robolectric.Shadows

@RunWith(RobolectricTestRunner::class)
class FaceEnrollTest {
    private lateinit var featureFactory: FeatureFactory

    private companion object {
        const val INTENT_KEY = "testKey"
        const val INTENT_VALUE = "testValue"
        val INTENT = Intent().apply {
            putExtra(INTENT_KEY, INTENT_VALUE)
            putExtra(Intent.EXTRA_USER_ID, 11)
        }
    }

    private val activityProvider = FaceEnrollActivityClassProvider()

    @Before
    fun setUp() {
        featureFactory = FakeFeatureFactory.setupForTest()
        whenever(featureFactory.faceFeatureProvider.enrollActivityClassProvider)
            .thenReturn(activityProvider)
    }

    private fun setupActivity(
        activityClass: Class<out FaceEnroll>,
        launchedFrom: String? = null,
    ): FaceEnroll {
        val activityController = Robolectric.buildActivity(activityClass, INTENT)
        activityController.get().launchedFromProvider = {
            launchedFrom ?: activityController.get().packageName
        }
        return activityController.create().get()
    }

    @Test
    fun testFinishAndLaunchDefaultActivity() {
        val activity = setupActivity(FaceEnroll::class.java)

        val nextActivityIntent = activity.asShadow().nextStartedActivity
        assertThat(nextActivityIntent.component!!.className).isEqualTo(activityProvider.next.name)
        assertThat(nextActivityIntent.extras!!.size()).isEqualTo(2)
        assertThat(nextActivityIntent.getStringExtra(INTENT_KEY)).isEqualTo(INTENT_VALUE)
        assertThat(nextActivityIntent.hasExtra(Intent.EXTRA_USER_ID)).isTrue()
    }

    @Test
    fun testFinishAndLaunchDefaultActivity_fromExternal_dropUser() {
        val activity = setupActivity(
            FaceEnroll::class.java,
            launchedFrom = "com.foo.bar"
        )

        val nextActivityIntent = activity.asShadow().nextStartedActivity
        assertThat(nextActivityIntent.component!!.className).isEqualTo(activityProvider.next.name)
        assertThat(nextActivityIntent.extras!!.size()).isEqualTo(1)
        assertThat(nextActivityIntent.getStringExtra(INTENT_KEY)).isEqualTo(INTENT_VALUE)
        assertThat(nextActivityIntent.hasExtra(Intent.EXTRA_USER_ID)).isFalse()
    }
}

private fun Activity.asShadow() = Shadows.shadowOf(this)