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


import static org.mockito.ArgumentMatchers.any;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.times;
import static org.mockito.Mockito.when;

import android.annotation.NonNull;
import android.annotation.Nullable;
import android.annotation.RequiresPermission;
import android.app.ActivityManager;
import android.content.BroadcastReceiver;
import android.content.ComponentName;
import android.content.ContentResolver;
import android.content.Context;
import android.content.ContextWrapper;
import android.content.Intent;
import android.content.IntentFilter;
import android.content.ServiceConnection;
import android.content.pm.PackageManager;
import android.nfc.Constants;
import android.nfc.cardemulation.HostNfcFService;
import android.nfc.cardemulation.NfcFServiceInfo;
import android.os.Build;
import android.os.Handler;
import android.os.IBinder;
import android.os.Message;
import android.os.Messenger;
import android.os.UserHandle;
import android.provider.Settings;
import android.util.Log;

import androidx.test.ext.junit.runners.AndroidJUnit4;
import androidx.test.platform.app.InstrumentationRegistry;

import com.android.dx.mockito.inline.extended.ExtendedMockito;
import com.android.nfc.cardemulation.AidRoutingManager;
import com.android.nfc.cardemulation.CardEmulationManager;
import com.android.nfc.cardemulation.HostNfcFEmulationManager;
import com.android.nfc.cardemulation.RegisteredAidCache;
import com.android.nfc.cardemulation.RegisteredT3tIdentifiersCache;
import com.android.nfc.cardemulation.RoutingOptionManager;

import org.junit.After;
import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.MockitoSession;
import org.mockito.quality.Strictness;

import com.android.nfc.flags.Flags;

@RunWith(AndroidJUnit4.class)
public class HostNfcFEmulationManagerTest {

    private static final String TAG = HostNfcFEmulationManagerTest.class.getSimpleName();
    private boolean mNfcSupported;
    private MockitoSession mStaticMockSession;
    private HostNfcFEmulationManager mHostNfcFEmulationManager;
    private ComponentName componentName;

    @Before
    public void setUp() throws Exception {
        mStaticMockSession = ExtendedMockito.mockitoSession()
                .mockStatic(Flags.class)
                .mockStatic(NfcStatsLog.class)
                .mockStatic(Message.class)
                .strictness(Strictness.LENIENT)
                .startMocking();

        Context context = InstrumentationRegistry.getInstrumentation().getTargetContext();
        PackageManager pm = context.getPackageManager();
        if (!pm.hasSystemFeature(PackageManager.FEATURE_NFC_ANY)) {
            mNfcSupported = false;
            return;
        }
        mNfcSupported = true;

        Context mockContext = new ContextWrapper(context) {

            public Context createContextAsUser(@NonNull UserHandle user,
                    @CreatePackageOptions int flags) {
                if (Build.IS_ENG) {
                    throw new IllegalStateException("createContextAsUser not overridden!");
                }
                return this;
            }

            public Intent registerReceiverForAllUsers(@Nullable BroadcastReceiver receiver,
                    @NonNull IntentFilter filter, @Nullable String broadcastPermission,
                    @Nullable Handler scheduler) {
                return mock(Intent.class);
            }

            public ContentResolver getContentResolver() {
                return mock(ContentResolver.class);
            }

            public boolean bindServiceAsUser(
                    @NonNull @RequiresPermission Intent service, @NonNull ServiceConnection conn,
                    int flags,
                    @NonNull UserHandle user) {
                return true;
            }

            public void unbindService(@NonNull ServiceConnection conn) {

            }


        };

        when(Flags.statsdCeEventsFlag()).thenReturn(false);
        RegisteredT3tIdentifiersCache t3tIdentifiersCache = mock(
                RegisteredT3tIdentifiersCache.class);
        NfcFServiceInfo nfcFServiceInfo = mock(NfcFServiceInfo.class);
        componentName = mock(ComponentName.class);
        when(nfcFServiceInfo.getComponent()).thenReturn(componentName);
        when(t3tIdentifiersCache.resolveNfcid2("6D2E616E64726F69")).thenReturn(nfcFServiceInfo);
        Message message = mock(Message.class);
        when(Message.obtain(null, HostNfcFService.MSG_COMMAND_PACKET)).thenReturn(message);
        when(Message.obtain(null, HostNfcFService.MSG_DEACTIVATED)).thenReturn(message);
        when(Message.obtain()).thenReturn(message);

        InstrumentationRegistry.getInstrumentation().runOnMainSync(
                () -> mHostNfcFEmulationManager =
                        new HostNfcFEmulationManager(mockContext, t3tIdentifiersCache));
        Assert.assertNotNull(mHostNfcFEmulationManager);
    }

    @After
    public void tearDown() throws Exception {
        mStaticMockSession.finishMocking();
    }

    @Test
    public void testOnEnabledForegroundNfcFServiceChanged() {
        if (!mNfcSupported) return;

        String packageName = mHostNfcFEmulationManager.getEnabledFgServiceName();
        Assert.assertNull(packageName);
        when(componentName.getPackageName()).thenReturn("com.android.nfc");
        mHostNfcFEmulationManager.onEnabledForegroundNfcFServiceChanged(0, componentName);
        packageName = mHostNfcFEmulationManager.getEnabledFgServiceName();
        Assert.assertNotNull(packageName);
        Assert.assertEquals("com.android.nfc", packageName);

    }

    @Test
    public void testOnHostEmulationData() {
        if (!mNfcSupported) return;

        testOnEnabledForegroundNfcFServiceChanged();
        mHostNfcFEmulationManager.onHostEmulationData("com.android.nfc".getBytes());
        ExtendedMockito.verify(() -> NfcStatsLog.write(NfcStatsLog.NFC_CARDEMULATION_OCCURRED,
                NfcStatsLog.NFC_CARDEMULATION_OCCURRED__CATEGORY__HCE_PAYMENT,
                "HCEF",
                0));
    }

    @Test
    public void testOnNfcDisabled() {
        if (!mNfcSupported) return;

        testOnHostEmulationData();
        ServiceConnection serviceConnection = mHostNfcFEmulationManager.getServiceConnection();
        Message message = mock(Message.class);
        when(Message.obtain(null, HostNfcFService.MSG_COMMAND_PACKET)).thenReturn(message);
        serviceConnection.onServiceConnected(mock(ComponentName.class), mock(IBinder.class));
        int userId = mHostNfcFEmulationManager.getServiceUserId();
        Assert.assertEquals(0, userId);
        mHostNfcFEmulationManager.onNfcDisabled();
        userId = mHostNfcFEmulationManager.getServiceUserId();
        Assert.assertEquals(-1, userId);
        ExtendedMockito.verify(() -> Message.obtain(null, HostNfcFService.MSG_COMMAND_PACKET));
    }


    @Test
    public void testOnUserSwitched() {
        if (!mNfcSupported) return;

        testOnHostEmulationData();
        ServiceConnection serviceConnection = mHostNfcFEmulationManager.getServiceConnection();
        Message message = mock(Message.class);
        when(Message.obtain(null, HostNfcFService.MSG_COMMAND_PACKET)).thenReturn(message);
        serviceConnection.onServiceConnected(mock(ComponentName.class), mock(IBinder.class));
        int userId = mHostNfcFEmulationManager.getServiceUserId();
        Assert.assertEquals(0, userId);
        mHostNfcFEmulationManager.onUserSwitched();
        boolean isUserSwitched = mHostNfcFEmulationManager.isUserSwitched();
        Assert.assertTrue(isUserSwitched);
        userId = mHostNfcFEmulationManager.getServiceUserId();
        Assert.assertEquals(-1, userId);
        ExtendedMockito.verify(() -> Message.obtain(null, HostNfcFService.MSG_COMMAND_PACKET));
    }

    @Test
    public void testOnHostEmulationDeactivated() {
        if (!mNfcSupported) return;

        testOnHostEmulationData();
        ServiceConnection serviceConnection = mHostNfcFEmulationManager.getServiceConnection();
        Message message = mock(Message.class);
        when(Message.obtain(null, HostNfcFService.MSG_COMMAND_PACKET)).thenReturn(message);
        serviceConnection.onServiceConnected(mock(ComponentName.class), mock(IBinder.class));
        ComponentName serviceName = mHostNfcFEmulationManager.getServiceName();
        Assert.assertNotNull(serviceName);
        mHostNfcFEmulationManager.onHostEmulationDeactivated();
        serviceName = mHostNfcFEmulationManager.getServiceName();
        Assert.assertNull(serviceName);
    }
}