/*
 * 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;

import static com.android.settings.connecteddevice.audiosharing.AudioSharingUtils.MetricKey.METRIC_KEY_DEVICE_IS_PRIMARY;

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

import static org.mockito.ArgumentMatchers.anyInt;
import static org.mockito.ArgumentMatchers.eq;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.never;
import static org.mockito.Mockito.spy;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.verifyNoInteractions;
import static org.mockito.Mockito.when;
import static org.robolectric.Shadows.shadowOf;

import android.app.settings.SettingsEnums;
import android.bluetooth.BluetoothDevice;
import android.content.Context;
import android.media.AudioManager;
import android.os.Looper;
import android.platform.test.annotations.DisableFlags;
import android.platform.test.annotations.EnableFlags;
import android.platform.test.flag.junit.SetFlagsRule;
import android.provider.Settings;
import android.widget.SeekBar;

import androidx.test.core.app.ApplicationProvider;

import com.android.settings.R;
import com.android.settings.bluetooth.Utils;
import com.android.settings.testutils.FakeFeatureFactory;
import com.android.settings.testutils.shadow.ShadowBluetoothUtils;
import com.android.settingslib.bluetooth.BluetoothUtils;
import com.android.settingslib.bluetooth.CachedBluetoothDevice;
import com.android.settingslib.bluetooth.CachedBluetoothDeviceManager;
import com.android.settingslib.bluetooth.LeAudioProfile;
import com.android.settingslib.bluetooth.LocalBluetoothManager;
import com.android.settingslib.bluetooth.LocalBluetoothProfileManager;
import com.android.settingslib.bluetooth.VolumeControlProfile;
import com.android.settingslib.flags.Flags;

import com.google.common.collect.ImmutableList;

import org.junit.Before;
import org.junit.Rule;
import org.junit.Test;
import org.junit.runner.RunWith;
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 = {ShadowBluetoothUtils.class})
public class AudioSharingDeviceVolumeSliderPreferenceTest {
    private static final String TEST_DEVICE_NAME = "test";
    private static final String TEST_DEVICE_ADDRESS = "XX:XX:XX:XX:XX:11";
    private static final int TEST_DEVICE_GROUP_ID = 1;
    private static final int TEST_VOLUME_VALUE = 255;
    private static final int TEST_MAX_STREAM_VALUE = 10;
    private static final int TEST_MIN_STREAM_VALUE = 0;

    @Rule public final MockitoRule mMockitoRule = MockitoJUnit.rule();
    @Rule public final SetFlagsRule mSetFlagsRule = new SetFlagsRule();

    @Mock private LocalBluetoothManager mLocalBtManager;
    @Mock private LocalBluetoothProfileManager mLocalBtProfileManager;
    @Mock private LeAudioProfile mLeAudioProfile;
    @Mock private CachedBluetoothDeviceManager mDeviceManager;
    @Mock private VolumeControlProfile mVolumeControl;
    @Mock private CachedBluetoothDevice mCachedDevice;
    @Mock private BluetoothDevice mDevice;
    @Mock private AudioManager mAudioManager;
    @Mock private SeekBar mSeekBar;
    private Context mContext;
    private AudioSharingDeviceVolumeSliderPreference mPreference;
    private FakeFeatureFactory mFeatureFactory;

    @Before
    public void setup() {
        mContext = spy(ApplicationProvider.getApplicationContext());
        ShadowBluetoothUtils.sLocalBluetoothManager = mLocalBtManager;
        mLocalBtManager = Utils.getLocalBtManager(mContext);
        mFeatureFactory = FakeFeatureFactory.setupForTest();
        when(mLocalBtManager.getProfileManager()).thenReturn(mLocalBtProfileManager);
        when(mLocalBtProfileManager.getVolumeControlProfile()).thenReturn(mVolumeControl);
        when(mLocalBtProfileManager.getLeAudioProfile()).thenReturn(mLeAudioProfile);
        when(mLocalBtManager.getCachedDeviceManager()).thenReturn(mDeviceManager);
        when(mContext.getSystemService(AudioManager.class)).thenReturn(mAudioManager);
        when(mAudioManager.getStreamMaxVolume(AudioManager.STREAM_MUSIC))
                .thenReturn(TEST_MAX_STREAM_VALUE);
        when(mAudioManager.getStreamMinVolume(AudioManager.STREAM_MUSIC))
                .thenReturn(TEST_MIN_STREAM_VALUE);
        when(mDevice.getAddress()).thenReturn(TEST_DEVICE_ADDRESS);
        when(mCachedDevice.getName()).thenReturn(TEST_DEVICE_NAME);
        when(mCachedDevice.getDevice()).thenReturn(mDevice);
        when(mCachedDevice.getGroupId()).thenReturn(TEST_DEVICE_GROUP_ID);
        when(mDeviceManager.getCachedDevicesCopy()).thenReturn(ImmutableList.of(mCachedDevice));
        when(mSeekBar.getProgress()).thenReturn(TEST_VOLUME_VALUE);
        mPreference = new AudioSharingDeviceVolumeSliderPreference(mContext, mCachedDevice);
        mPreference.initialize();
    }

    @Test
    public void getCachedDevice_returnsDevice() {
        assertThat(mPreference.getCachedDevice()).isEqualTo(mCachedDevice);
    }

    @Test
    public void initialize_setupMaxMin() {
        mPreference = spy(mPreference);
        mPreference.initialize();
        shadowOf(Looper.getMainLooper()).idle();

        assertThat(mPreference.getMax())
                .isEqualTo(AudioSharingDeviceVolumeSliderPreference.MAX_VOLUME);
        assertThat(mPreference.getMin())
                .isEqualTo(AudioSharingDeviceVolumeSliderPreference.MIN_VOLUME);
        assertThat(mPreference.getTitle().toString()).isEqualTo(TEST_DEVICE_NAME);
        verify(mPreference)
                .setSliderContentDescription(
                        mContext.getString(
                                R.string.audio_sharing_device_volume_description,
                                TEST_DEVICE_NAME));
    }

    @Test
    public void callChangeListener_notFallbackDevice_setDeviceVolume() {
        mPreference.callChangeListener(TEST_VOLUME_VALUE);

        verify(mVolumeControl).setDeviceVolume(mDevice, TEST_VOLUME_VALUE, /* isGroupOp= */ true);
        verifyNoInteractions(mAudioManager);
        verify(mFeatureFactory.metricsFeatureProvider)
                .action(
                        SettingsEnums.PAGE_UNKNOWN,
                        SettingsEnums.ACTION_AUDIO_SHARING_CHANGE_MEDIA_DEVICE_VOLUME,
                        SettingsEnums.PAGE_UNKNOWN,
                        String.valueOf(METRIC_KEY_DEVICE_IS_PRIMARY.getId()),
                        /* isPrimary */ 0);
    }

    @Test
    public void setValue_notFallbackDevice_doNothing() {
        mPreference.setValue(TEST_VOLUME_VALUE);

        verifyNoInteractions(mVolumeControl);
        verifyNoInteractions(mAudioManager);
        verify(mFeatureFactory.metricsFeatureProvider, never())
                .action(
                        eq(SettingsEnums.PAGE_UNKNOWN),
                        eq(SettingsEnums.ACTION_AUDIO_SHARING_CHANGE_MEDIA_DEVICE_VOLUME),
                        eq(SettingsEnums.PAGE_UNKNOWN),
                        eq(String.valueOf(METRIC_KEY_DEVICE_IS_PRIMARY.getId())),
                        anyInt());
    }

    @Test
    @EnableFlags(Flags.FLAG_ADOPT_PRIMARY_GROUP_MANAGEMENT_API_V2)
    public void callChangeListener_adoptApi_fallbackDevice_setDeviceVolume() {
        when(mLeAudioProfile.getBroadcastToUnicastFallbackGroup()).thenReturn(TEST_DEVICE_GROUP_ID);
        mPreference.callChangeListener(TEST_VOLUME_VALUE);

        verifyNoInteractions(mVolumeControl);
        verify(mAudioManager)
                .setStreamVolume(AudioManager.STREAM_MUSIC, TEST_MAX_STREAM_VALUE, /* flags= */ 0);
        verify(mFeatureFactory.metricsFeatureProvider)
                .action(
                        SettingsEnums.PAGE_UNKNOWN,
                        SettingsEnums.ACTION_AUDIO_SHARING_CHANGE_MEDIA_DEVICE_VOLUME,
                        SettingsEnums.PAGE_UNKNOWN,
                        String.valueOf(METRIC_KEY_DEVICE_IS_PRIMARY.getId()),
                        /* isPrimary */ 1);
    }

    @Test
    @EnableFlags(Flags.FLAG_ADOPT_PRIMARY_GROUP_MANAGEMENT_API_V2)
    public void setValue_adoptApi_fallbackDevice_doNothing() {
        when(mLeAudioProfile.getBroadcastToUnicastFallbackGroup()).thenReturn(TEST_DEVICE_GROUP_ID);
        mPreference.setValue(TEST_VOLUME_VALUE);

        verifyNoInteractions(mVolumeControl);
        verifyNoInteractions(mAudioManager);
        verify(mFeatureFactory.metricsFeatureProvider, never())
                .action(
                        eq(SettingsEnums.PAGE_UNKNOWN),
                        eq(SettingsEnums.ACTION_AUDIO_SHARING_CHANGE_MEDIA_DEVICE_VOLUME),
                        eq(SettingsEnums.PAGE_UNKNOWN),
                        eq(String.valueOf(METRIC_KEY_DEVICE_IS_PRIMARY.getId())),
                        anyInt());
    }

    @Test
    @DisableFlags(Flags.FLAG_ADOPT_PRIMARY_GROUP_MANAGEMENT_API_V2)
    public void callChangeListener_fallbackDevice_setDeviceVolume() {
        Settings.Secure.putInt(
                mContext.getContentResolver(),
                BluetoothUtils.getPrimaryGroupIdUriForBroadcast(),
                TEST_DEVICE_GROUP_ID);
        mPreference.callChangeListener(TEST_VOLUME_VALUE);

        verifyNoInteractions(mVolumeControl);
        verify(mAudioManager)
                .setStreamVolume(AudioManager.STREAM_MUSIC, TEST_MAX_STREAM_VALUE, /* flags= */ 0);
        verify(mFeatureFactory.metricsFeatureProvider)
                .action(
                        SettingsEnums.PAGE_UNKNOWN,
                        SettingsEnums.ACTION_AUDIO_SHARING_CHANGE_MEDIA_DEVICE_VOLUME,
                        SettingsEnums.PAGE_UNKNOWN,
                        String.valueOf(METRIC_KEY_DEVICE_IS_PRIMARY.getId()),
                        /* isPrimary */ 1);
    }

    @Test
    @DisableFlags(Flags.FLAG_ADOPT_PRIMARY_GROUP_MANAGEMENT_API_V2)
    public void setValue_fallbackDevice_doNothing() {
        Settings.Secure.putInt(
                mContext.getContentResolver(),
                BluetoothUtils.getPrimaryGroupIdUriForBroadcast(),
                TEST_DEVICE_GROUP_ID);
        mPreference.setValue(TEST_VOLUME_VALUE);

        verifyNoInteractions(mVolumeControl);
        verifyNoInteractions(mAudioManager);
        verify(mFeatureFactory.metricsFeatureProvider, never())
                .action(
                        eq(SettingsEnums.PAGE_UNKNOWN),
                        eq(SettingsEnums.ACTION_AUDIO_SHARING_CHANGE_MEDIA_DEVICE_VOLUME),
                        eq(SettingsEnums.PAGE_UNKNOWN),
                        eq(String.valueOf(METRIC_KEY_DEVICE_IS_PRIMARY.getId())),
                        anyInt());
    }

    @Test
    public void equals_returnsTrue() {
        AudioSharingDeviceVolumeSliderPreference preference =
                new AudioSharingDeviceVolumeSliderPreference(mContext, mCachedDevice);
        assertThat(mPreference.equals(preference)).isTrue();
    }

    @Test
    public void equals_returnsFalse() {
        CachedBluetoothDevice cachedDevice = mock(CachedBluetoothDevice.class);
        AudioSharingDeviceVolumeSliderPreference preference =
                new AudioSharingDeviceVolumeSliderPreference(mContext, cachedDevice);
        assertThat(mPreference.equals(preference)).isFalse();
    }

    @Test
    public void toString_correctValue() {
        mPreference.setTitle(TEST_DEVICE_NAME);
        StringBuilder builder = new StringBuilder("Preference{");
        builder.append("preference=").append(TEST_DEVICE_NAME);
        when(mDevice.getAnonymizedAddress()).thenReturn(TEST_DEVICE_ADDRESS);
        builder.append(", device=").append(TEST_DEVICE_ADDRESS);
        builder.append("}");
        assertThat(mPreference.toString()).isEqualTo(builder.toString());
    }

    @Test
    public void onPreferenceAttributesChanged_nameChanged_updatePreference() {
        mPreference = spy(mPreference);
        when(mCachedDevice.getName()).thenReturn("new");
        mPreference.onPreferenceAttributesChanged();
        shadowOf(Looper.getMainLooper()).idle();

        assertThat(mPreference.getTitle().toString()).isEqualTo("new");
        verify(mPreference)
                .setSliderContentDescription(
                        mContext.getString(
                                R.string.audio_sharing_device_volume_description, "new"));
    }
}