/*
 * Copyright (C) 2016 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.internal.telephony;

import static com.android.internal.telephony.TelephonyTestUtils.waitForMs;

import static org.junit.Assert.assertTrue;
import static org.mockito.Mockito.doReturn;
import static org.mockito.Mockito.eq;
import static org.mockito.Mockito.times;
import static org.mockito.Mockito.verify;

import android.content.IntentFilter;
import android.content.pm.ServiceInfo;
import android.os.RemoteException;
import android.telephony.AccessNetworkConstants;
import android.telephony.INetworkService;
import android.telephony.INetworkServiceCallback;
import android.telephony.NetworkRegistrationState;
import android.telephony.NetworkService;
import android.telephony.NetworkServiceCallback;
import android.telephony.ServiceState;
import android.telephony.SubscriptionManager;
import android.test.suitebuilder.annotation.MediumTest;

import com.android.internal.R;

import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import org.mockito.Mock;

public class CellularNetworkServiceTest extends TelephonyTest {
    CellularNetworkService mCellularNetworkService;

    @Mock
    private INetworkServiceCallback mCallback;

    private void addNetworkService() {
        mCellularNetworkService = new CellularNetworkService();
        ServiceInfo serviceInfo =  new ServiceInfo();
        serviceInfo.packageName = "com.android.phone";
        serviceInfo.permission = "android.permission.BIND_NETWORK_SERVICE";
        IntentFilter filter = new IntentFilter();
        mContextFixture.addService(
                NetworkService.NETWORK_SERVICE_INTERFACE,
                null,
                "com.android.phone",
                mCellularNetworkService.mBinder,
                serviceInfo,
                filter);
    }
    INetworkService.Stub mBinder;

    @Before
    public void setUp() throws Exception {

        logd("CellularNetworkServiceTest +Setup!");
        super.setUp("CellularNetworkServiceTest");

        mContextFixture.putResource(R.string.config_wwan_network_service_package,
                "com.android.phone");
        addNetworkService();
        mBinder = mCellularNetworkService.mBinder;
        mBinder.createNetworkServiceProvider(0);

        int dds = SubscriptionManager.getDefaultDataSubscriptionId();
        doReturn(dds).when(mPhone).getSubId();

        logd("CellularNetworkServiceTest -Setup!");
    }

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

    @Test
    @MediumTest
    public void testGetNetworkRegistrationState() {
        int voiceRegState = NetworkRegistrationState.REG_STATE_HOME;
        int dataRegState = NetworkRegistrationState.REG_STATE_HOME;
        int voiceRadioTech = ServiceState.RIL_RADIO_TECHNOLOGY_HSPA;
        int dataRadioTech = ServiceState.RIL_RADIO_TECHNOLOGY_HSPA;
        int domain = NetworkRegistrationState.DOMAIN_CS;

        boolean cssSupported = true;
        int roamingIndicator = 1;
        int systemIsInPrl = 2;
        int defaultRoamingIndicator = 3;
        int reasonForDenial = 0;
        int maxDataCalls = 4;
        int[] availableServices = new int[] {
                NetworkRegistrationState.SERVICE_TYPE_VOICE,
                NetworkRegistrationState.SERVICE_TYPE_SMS,
                NetworkRegistrationState.SERVICE_TYPE_VIDEO
        };

        mSimulatedCommands.setVoiceRegState(voiceRegState);
        mSimulatedCommands.setVoiceRadioTech(voiceRadioTech);
        mSimulatedCommands.setDataRegState(dataRegState);
        mSimulatedCommands.setDataRadioTech(dataRadioTech);
        mSimulatedCommands.mCssSupported = cssSupported;
        mSimulatedCommands.mRoamingIndicator = roamingIndicator;
        mSimulatedCommands.mSystemIsInPrl = systemIsInPrl;
        mSimulatedCommands.mDefaultRoamingIndicator = defaultRoamingIndicator;
        mSimulatedCommands.mReasonForDenial = reasonForDenial;
        mSimulatedCommands.mMaxDataCalls = maxDataCalls;

        mSimulatedCommands.notifyNetworkStateChanged();

        try {
            mBinder.getNetworkRegistrationState(0, domain, mCallback);
        } catch (RemoteException e) {
            assertTrue(false);
        }

        waitForMs(1000);

        NetworkRegistrationState expectedState = new NetworkRegistrationState(
                AccessNetworkConstants.TransportType.WWAN, domain, voiceRegState,
                ServiceState.rilRadioTechnologyToNetworkType(voiceRadioTech), reasonForDenial,
                false, availableServices, null, cssSupported,
                roamingIndicator, systemIsInPrl, defaultRoamingIndicator);

        try {
            verify(mCallback, times(1)).onGetNetworkRegistrationStateComplete(
                    eq(NetworkServiceCallback.RESULT_SUCCESS), eq(expectedState));
        } catch (RemoteException e) {
            assertTrue(false);
        }

        domain = NetworkRegistrationState.DOMAIN_PS;
        availableServices = new int[] {NetworkRegistrationState.SERVICE_TYPE_DATA};
        try {
            mBinder.getNetworkRegistrationState(0, domain, mCallback);
        } catch (RemoteException e) {
            assertTrue(false);
        }

        waitForMs(1000);

        expectedState = new NetworkRegistrationState(
                AccessNetworkConstants.TransportType.WWAN, domain, voiceRegState,
                ServiceState.rilRadioTechnologyToNetworkType(voiceRadioTech), reasonForDenial,
                false, availableServices, null, maxDataCalls);

        try {
            verify(mCallback, times(1)).onGetNetworkRegistrationStateComplete(
                    eq(NetworkServiceCallback.RESULT_SUCCESS), eq(expectedState));
        } catch (RemoteException e) {
            assertTrue(false);
        }
    }
}