/*
 * 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.audiosharing.audiostreams;

import static android.app.settings.SettingsEnums.AUDIO_STREAM_MAIN;

import static com.android.settings.connecteddevice.audiosharing.audiostreams.SourceAddedState.AUDIO_STREAM_SOURCE_ADDED_STATE_SUMMARY;

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

import static org.mockito.ArgumentMatchers.any;
import static org.mockito.ArgumentMatchers.anyInt;
import static org.mockito.ArgumentMatchers.anyString;
import static org.mockito.ArgumentMatchers.eq;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.never;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;

import android.app.settings.SettingsEnums;
import android.bluetooth.BluetoothLeBroadcastMetadata;
import android.content.Context;
import android.content.Intent;
import android.os.Bundle;
import android.platform.test.annotations.DisableFlags;
import android.platform.test.annotations.EnableFlags;
import android.platform.test.flag.junit.SetFlagsRule;

import androidx.fragment.app.FragmentActivity;
import androidx.preference.Preference;
import androidx.test.core.app.ApplicationProvider;

import com.android.settings.R;
import com.android.settings.SettingsActivity;
import com.android.settings.testutils.FakeFeatureFactory;
import com.android.settings.testutils.shadow.ShadowFragment;
import com.android.settingslib.core.instrumentation.MetricsFeatureProvider;
import com.android.settingslib.flags.Flags;

import org.junit.Before;
import org.junit.Rule;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.ArgumentCaptor;
import org.mockito.Mock;
import org.mockito.junit.MockitoJUnit;
import org.mockito.junit.MockitoRule;
import org.robolectric.RobolectricTestRunner;
import org.robolectric.annotation.Config;

@RunWith(RobolectricTestRunner.class)
@Config(
        shadows = {
            ShadowFragment.class,
        })
public class SourceAddedStateTest {
    @Rule public final MockitoRule mMockitoRule = MockitoJUnit.rule();
    @Rule public final SetFlagsRule mSetFlagsRule = new SetFlagsRule();
    private static final int BROADCAST_ID = 1;
    private static final String BROADCAST_TITLE = "title";
    private final Context mContext = ApplicationProvider.getApplicationContext();
    @Mock private AudioStreamPreference mPreference;
    @Mock private AudioStreamsProgressCategoryController mController;
    @Mock private AudioStreamsHelper mHelper;
    @Mock private AudioStreamScanHelper mScanHelper;
    @Mock private AudioStreamsRepository mRepository;
    @Mock private AudioStreamsDashboardFragment mFragment;
    @Mock private FragmentActivity mActivity;
    private FakeFeatureFactory mFeatureFactory;
    private SourceAddedState mInstance;

    @Before
    public void setUp() {
        when(mFragment.getActivity()).thenReturn(mActivity);
        mFeatureFactory = FakeFeatureFactory.setupForTest();
        mInstance = new SourceAddedState();
        when(mPreference.getAudioStreamBroadcastId()).thenReturn(BROADCAST_ID);
        when(mPreference.getTitle()).thenReturn(BROADCAST_TITLE);
    }

    @Test
    public void testGetInstance() {
        mInstance = SourceAddedState.getInstance();
        assertThat(mInstance).isNotNull();
        assertThat(mInstance).isInstanceOf(SourceAddedState.class);
    }

    @Test
    public void testGetSummary() {
        int summary = mInstance.getSummary();
        assertThat(summary).isEqualTo(AUDIO_STREAM_SOURCE_ADDED_STATE_SUMMARY);
    }

    @Test
    public void testGetStateEnum() {
        AudioStreamsProgressCategoryController.AudioStreamState stateEnum =
                mInstance.getStateEnum();
        assertThat(stateEnum)
                .isEqualTo(AudioStreamsProgressCategoryController.AudioStreamState.SOURCE_ADDED);
    }

    @Test
    @DisableFlags(Flags.FLAG_AUDIO_STREAM_MEDIA_SERVICE_BY_RECEIVE_STATE)
    public void testPerformAction_startService() {
        mInstance.setAudioStreamsRepositoryForTesting(mRepository);
        BluetoothLeBroadcastMetadata mockMetadata = mock(BluetoothLeBroadcastMetadata.class);
        when(mRepository.getCachedMetadata(anyInt())).thenReturn(mockMetadata);
        when(mPreference.getContext()).thenReturn(mContext);
        when(mPreference.getSourceOriginForLogging())
                .thenReturn(SourceOriginForLogging.QR_CODE_SCAN_SETTINGS);

        mInstance.onEnter(mPreference, mController, mHelper, mScanHelper);

        verify(mRepository).saveMetadata(eq(mContext), eq(mockMetadata));
        verify(mFeatureFactory.metricsFeatureProvider)
                .action(
                        eq(mContext),
                        eq(SettingsEnums.ACTION_AUDIO_STREAM_JOIN_SUCCEED),
                        eq(SourceOriginForLogging.QR_CODE_SCAN_SETTINGS.ordinal()));
        verify(mHelper).startMediaService(eq(mContext), eq(BROADCAST_ID), eq(BROADCAST_TITLE));
    }

    @Test
    @EnableFlags(Flags.FLAG_AUDIO_STREAM_MEDIA_SERVICE_BY_RECEIVE_STATE)
    public void testPerformAction_skipStartService() {
        mInstance.setAudioStreamsRepositoryForTesting(mRepository);
        BluetoothLeBroadcastMetadata mockMetadata = mock(BluetoothLeBroadcastMetadata.class);
        when(mRepository.getCachedMetadata(anyInt())).thenReturn(mockMetadata);
        when(mPreference.getContext()).thenReturn(mContext);
        when(mPreference.getSourceOriginForLogging())
                .thenReturn(SourceOriginForLogging.QR_CODE_SCAN_SETTINGS);

        mInstance.onEnter(mPreference, mController, mHelper, mScanHelper);

        verify(mRepository).saveMetadata(eq(mContext), eq(mockMetadata));
        verify(mFeatureFactory.metricsFeatureProvider)
                .action(
                        eq(mContext),
                        eq(SettingsEnums.ACTION_AUDIO_STREAM_JOIN_SUCCEED),
                        eq(SourceOriginForLogging.QR_CODE_SCAN_SETTINGS.ordinal()));
        verify(mHelper, never()).startMediaService(any(), anyInt(), anyString());
    }

    @Test
    public void testGetOnClickListener_startSubSettings() {
        when(mController.getFragment()).thenReturn(mFragment);
        when(mFragment.getMetricsCategory()).thenReturn(AUDIO_STREAM_MAIN);

        Preference.OnPreferenceClickListener listener = mInstance.getOnClickListener(mController);
        assertThat(listener).isNotNull();

        // mContext is not an Activity context, calling startActivity() from outside of an Activity
        // context requires the FLAG_ACTIVITY_NEW_TASK flag, create a mock to avoid this
        // AndroidRuntimeException.
        Context activityContext = mock(Context.class);
        when(mPreference.getContext()).thenReturn(activityContext);

        listener.onPreferenceClick(mPreference);

        ArgumentCaptor<Intent> argumentCaptor = ArgumentCaptor.forClass(Intent.class);
        verify(activityContext).startActivity(argumentCaptor.capture());

        Intent intent = argumentCaptor.getValue();
        assertThat(intent.getStringExtra(SettingsActivity.EXTRA_SHOW_FRAGMENT))
                .isEqualTo(AudioStreamDetailsFragment.class.getName());
        assertThat(intent.getIntExtra(SettingsActivity.EXTRA_SHOW_FRAGMENT_TITLE_RESID, 0))
                .isEqualTo(R.string.audio_streams_detail_page_title);
        assertThat(intent.getIntExtra(MetricsFeatureProvider.EXTRA_SOURCE_METRICS_CATEGORY, 0))
                .isEqualTo(AUDIO_STREAM_MAIN);

        Bundle bundle = intent.getBundleExtra(SettingsActivity.EXTRA_SHOW_FRAGMENT_ARGUMENTS);
        assertThat(bundle).isNotNull();
        assertThat(bundle.getString(AudioStreamDetailsFragment.BROADCAST_NAME_ARG))
                .isEqualTo(BROADCAST_TITLE);
        assertThat(bundle.getInt(AudioStreamDetailsFragment.BROADCAST_ID_ARG))
                .isEqualTo(BROADCAST_ID);
    }
}