/*
 * Copyright (C) 2015 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.contacts.preference;

import android.accounts.Account;
import android.content.Context;
import android.content.SharedPreferences;
import android.content.res.Resources;
import android.provider.ContactsContract.RawContacts.DefaultAccount.DefaultAccountAndState;
import android.test.InstrumentationTestCase;

import androidx.test.InstrumentationRegistry;
import androidx.test.filters.SmallTest;

import com.android.contacts.model.account.AccountWithDataSet;
import com.android.contacts.preference.ContactsPreferences.DefaultAccountReader;

import junit.framework.Assert;

import org.mockito.Mock;
import org.mockito.Mockito;
import org.mockito.MockitoAnnotations;

import java.util.Arrays;

@SmallTest
public class ContactsPreferencesTest extends InstrumentationTestCase {

    private static final String ACCOUNT_KEY = "ACCOUNT_KEY";

    @Mock private Context mContext;
    @Mock private Resources mResources;
    @Mock private SharedPreferences mSharedPreferences;
    @Mock private DefaultAccountReader mDefaultAccountReader;

    private ContactsPreferences mContactsPreferences;

    @Override
    public void setUp() throws Exception {
        super.setUp();
        System.setProperty(
                "dexmaker.dexcache",
                getInstrumentation().getTargetContext().getCacheDir().getPath());
        MockitoAnnotations.initMocks(this);

        Mockito.when(mContext.getResources()).thenReturn(mResources);
        Mockito.when(mResources.getString(Mockito.anyInt()))
                .thenReturn(ACCOUNT_KEY); // contact_editor_default_account_key

        Mockito.when(mContext.getSharedPreferences(Mockito.any(), Mockito.anyInt()))
                .thenReturn(mSharedPreferences);
        Mockito.when(mSharedPreferences.contains(ContactsPreferences.SORT_ORDER_KEY))
                .thenReturn(true);
        Mockito.when(mSharedPreferences.contains(ContactsPreferences.DISPLAY_ORDER_KEY))
                .thenReturn(true);
        Mockito.when(mSharedPreferences.contains(ContactsPreferences.PHONETIC_NAME_DISPLAY_KEY))
                .thenReturn(true);

        InstrumentationRegistry.getInstrumentation()
                .runOnMainSync(
                        new Runnable() {
                            @Override
                            public void run() {
                                mContactsPreferences = new ContactsPreferences(mContext);
                            }
                        });
    }

    public void testGetSortOrderDefault() {
        Mockito.when(mResources.getBoolean(Mockito.anyInt()))
                .thenReturn(
                        false, // R.bool.config_sort_order_user_changeable
                        true // R.bool.config_default_sort_order_primary
                        );
        Assert.assertEquals(
                ContactsPreferences.SORT_ORDER_PRIMARY, mContactsPreferences.getSortOrder());
    }

    public void testGetSortOrder() {
        Mockito.when(mResources.getBoolean(Mockito.anyInt()))
                .thenReturn(
                        true // R.bool.config_sort_order_user_changeable
                        );
        Mockito.when(
                        mSharedPreferences.getInt(
                                Mockito.eq(ContactsPreferences.SORT_ORDER_KEY), Mockito.anyInt()))
                .thenReturn(ContactsPreferences.SORT_ORDER_PRIMARY);
        Assert.assertEquals(
                ContactsPreferences.SORT_ORDER_PRIMARY, mContactsPreferences.getSortOrder());
    }

    public void testGetDisplayOrderDefault() {
        Mockito.when(mResources.getBoolean(Mockito.anyInt()))
                .thenReturn(
                        false, // R.bool.config_display_order_user_changeable
                        true // R.bool.config_default_display_order_primary
                        );
        Assert.assertEquals(
                ContactsPreferences.DISPLAY_ORDER_PRIMARY, mContactsPreferences.getDisplayOrder());
    }

    public void testGetDisplayOrder() {
        Mockito.when(mResources.getBoolean(Mockito.anyInt()))
                .thenReturn(
                        true // R.bool.config_display_order_user_changeable
                        );
        Mockito.when(
                        mSharedPreferences.getInt(
                                Mockito.eq(ContactsPreferences.DISPLAY_ORDER_KEY),
                                Mockito.anyInt()))
                .thenReturn(ContactsPreferences.DISPLAY_ORDER_PRIMARY);
        Assert.assertEquals(
                ContactsPreferences.DISPLAY_ORDER_PRIMARY, mContactsPreferences.getDisplayOrder());
    }

    public void testGetPhoneticNameDisplayDefault() {
        Mockito.when(mResources.getBoolean(Mockito.anyInt()))
                .thenReturn(
                        false, // R.bool.config_phonetic_name_display_user_changeable
                        true // R.bool.config_default_hide_phonetic_name_if_empty
                        );
        Assert.assertEquals(
                PhoneticNameDisplayPreference.HIDE_IF_EMPTY,
                mContactsPreferences.getPhoneticNameDisplayPreference());
    }

    public void testGetPhoneticNameDisplay() {
        Mockito.when(mResources.getBoolean(Mockito.anyInt()))
                .thenReturn(
                        true // R.bool.config_phonetic_name_display_user_changeable
                        );
        Mockito.when(
                        mSharedPreferences.getInt(
                                Mockito.eq(ContactsPreferences.PHONETIC_NAME_DISPLAY_KEY),
                                Mockito.anyInt()))
                .thenReturn(PhoneticNameDisplayPreference.HIDE_IF_EMPTY);
        Assert.assertEquals(
                PhoneticNameDisplayPreference.HIDE_IF_EMPTY,
                mContactsPreferences.getPhoneticNameDisplayPreference());
    }

    public void testRefreshPhoneticNameDisplay() throws InterruptedException {
        Mockito.when(mResources.getBoolean(Mockito.anyInt()))
                .thenReturn(
                        true // R.bool.config_phonetic_name_display_user_changeable
                        );
        Mockito.when(
                        mSharedPreferences.getInt(
                                Mockito.eq(ContactsPreferences.PHONETIC_NAME_DISPLAY_KEY),
                                Mockito.anyInt()))
                .thenReturn(
                        PhoneticNameDisplayPreference.HIDE_IF_EMPTY,
                        PhoneticNameDisplayPreference.SHOW_ALWAYS);

        Assert.assertEquals(
                PhoneticNameDisplayPreference.HIDE_IF_EMPTY,
                mContactsPreferences.getPhoneticNameDisplayPreference());
        mContactsPreferences.refreshValue(ContactsPreferences.PHONETIC_NAME_DISPLAY_KEY);

        Assert.assertEquals(
                PhoneticNameDisplayPreference.SHOW_ALWAYS,
                mContactsPreferences.getPhoneticNameDisplayPreference());
    }

    public void testRefreshSortOrder() throws InterruptedException {
        Mockito.when(mResources.getBoolean(Mockito.anyInt()))
                .thenReturn(
                        true // R.bool.config_sort_order_user_changeable
                        );
        Mockito.when(
                        mSharedPreferences.getInt(
                                Mockito.eq(ContactsPreferences.SORT_ORDER_KEY), Mockito.anyInt()))
                .thenReturn(
                        ContactsPreferences.SORT_ORDER_PRIMARY,
                        ContactsPreferences.SORT_ORDER_ALTERNATIVE);

        Assert.assertEquals(
                ContactsPreferences.SORT_ORDER_PRIMARY, mContactsPreferences.getSortOrder());
        mContactsPreferences.refreshValue(ContactsPreferences.SORT_ORDER_KEY);

        Assert.assertEquals(
                ContactsPreferences.SORT_ORDER_ALTERNATIVE, mContactsPreferences.getSortOrder());
    }

    public void testRefreshDisplayOrder() throws InterruptedException {
        Mockito.when(mResources.getBoolean(Mockito.anyInt()))
                .thenReturn(
                        true // R.bool.config_display_order_user_changeable
                        );
        Mockito.when(
                        mSharedPreferences.getInt(
                                Mockito.eq(ContactsPreferences.DISPLAY_ORDER_KEY),
                                Mockito.anyInt()))
                .thenReturn(
                        ContactsPreferences.DISPLAY_ORDER_PRIMARY,
                        ContactsPreferences.DISPLAY_ORDER_ALTERNATIVE);

        Assert.assertEquals(
                ContactsPreferences.DISPLAY_ORDER_PRIMARY, mContactsPreferences.getDisplayOrder());
        mContactsPreferences.refreshValue(ContactsPreferences.DISPLAY_ORDER_KEY);

        Assert.assertEquals(
                ContactsPreferences.DISPLAY_ORDER_ALTERNATIVE,
                mContactsPreferences.getDisplayOrder());
    }

    public void testRefreshDefaultAccount() throws InterruptedException {
        mContactsPreferences =
                new ContactsPreferences(
                        mContext, /* isDefaultAccountUserChangeable */ true, mDefaultAccountReader);

        Mockito.when(mDefaultAccountReader.getDefaultAccountAndState())
                .thenReturn(
                        DefaultAccountAndState.ofCloud(new Account("name1", "type1")),
                        DefaultAccountAndState.ofCloud(new Account("name2", "type2")));

        Assert.assertEquals(
                new AccountWithDataSet("name1", "type1", null),
                mContactsPreferences.getDefaultAccount());
        mContactsPreferences.refreshValue(ACCOUNT_KEY);

        Assert.assertEquals(
                new AccountWithDataSet("name2", "type2", null),
                mContactsPreferences.getDefaultAccount());
    }

    public void testShouldShowAccountChangedNotificationIfAccountNotSaved() {
        mContactsPreferences =
                new ContactsPreferences(
                        mContext, /* isDefaultAccountUserChangeable */ true, mDefaultAccountReader);
        Mockito.when(mDefaultAccountReader.getDefaultAccountAndState())
                .thenReturn(DefaultAccountAndState.ofNotSet());

        assertTrue(
                "Should prompt to change default if no default is saved",
                mContactsPreferences.shouldShowAccountChangedNotification(
                        Arrays.asList(
                                new AccountWithDataSet("name1", "type1", "dataset1"),
                                new AccountWithDataSet("name2", "type2", "dataset2"))));
    }

    public void testShouldShowAccountChangedNotification() {
        mContactsPreferences =
                new ContactsPreferences(
                        mContext, /* isDefaultAccountUserChangeable */ true, mDefaultAccountReader);
        Mockito.when(mDefaultAccountReader.getDefaultAccountAndState())
                .thenReturn(DefaultAccountAndState.ofCloud(new Account("name", "type")));

        assertFalse(
                "Should not prompt to change default if current default exists",
                mContactsPreferences.shouldShowAccountChangedNotification(
                        Arrays.asList(
                                new AccountWithDataSet("name", "type", null),
                                new AccountWithDataSet("name1", "type1", "dataset1"))));

        assertTrue(
                "Should prompt to change default if current default does not exist",
                mContactsPreferences.shouldShowAccountChangedNotification(
                        Arrays.asList(
                                new AccountWithDataSet("name1", "type1", "dataset1"),
                                new AccountWithDataSet("name2", "type2", "dataset2"))));
    }

    public void testShouldShowAccountChangedNotificationWhenThereIsOneAccount() {
        mContactsPreferences =
                new ContactsPreferences(
                        mContext, /* isDefaultAccountUserChangeable */ true, mDefaultAccountReader);
        Mockito.when(mDefaultAccountReader.getDefaultAccountAndState())
                .thenReturn(DefaultAccountAndState.ofNotSet());

        // Normally we would prompt because there is no default set but if there is just one
        // account we should just use it.
        assertFalse(
                "Should not prompt to change default if there is only one account available",
                mContactsPreferences.shouldShowAccountChangedNotification(
                        Arrays.asList(new AccountWithDataSet("name", "type", "dataset"))));
    }
}