# Copyright (c) 2014 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

import dbus
import dbus.types
import time

from autotest_lib.client.cros.cellular import mm1_constants
from autotest_lib.client.cros.cellular.pseudomodem import modem_3gpp
from autotest_lib.client.cros.cellular.pseudomodem import sim
from autotest_lib.client.cros.cellular.pseudomodem import utils as pm_utils

I_ACTIVATION_TEST = 'Interface.LTEActivationTest'

class TestModem(modem_3gpp.Modem3gpp):
    """
    Base class for the custom 3GPP fake modems that are defined in this test.
    This modem boots up as unprovisioned & becomes activated only if it has
    been explicitly activated by calling CompleteCellularActivation

    """
    def __init__(self,
                 state_machine_factory=None,
                 bus=None,
                 device='pseudomodem0',
                 index=0,
                 roaming_networks=None,
                 config=None):
        super(TestModem, self).__init__(state_machine_factory,
                                        bus=bus,
                                        device=device,
                                        roaming_networks=roaming_networks,
                                        config=config)
        # Update the registered susbscription state as unprovisioned
        # for this activation test
        self._cached_registered_subscription_state = (
                mm1_constants.MM_MODEM_3GPP_SUBSCRIPTION_STATE_UNPROVISIONED)


    def _InitializeProperties(self):
        props = modem_3gpp.Modem3gpp._InitializeProperties(self)
        modem_props = props[mm1_constants.I_MODEM]
        modem_props['OwnNumbers'] = ['0000000000']
        modem_props['AccessTechnologies'] = dbus.types.UInt32(
            mm1_constants.MM_MODEM_ACCESS_TECHNOLOGY_LTE)
        modem_props['ModemCapabilities'] = dbus.types.UInt32(
            mm1_constants.MM_MODEM_CAPABILITY_LTE)
        modem_props['CurrentCapabilities'] = dbus.types.UInt32(
            mm1_constants.MM_MODEM_CAPABILITY_LTE)

        # For the purposes of this test, introduce a property to help
        # verify that a reset has taken place. Expose this under a test
        # specific interface.
        if hasattr(self, '_properties'):
            reset_called = \
                self._properties[I_ACTIVATION_TEST]['ResetCalled']
        else:
            reset_called = False
        props[I_ACTIVATION_TEST] = {
            'ResetCalled' : dbus.types.Boolean(reset_called)
        }
        return props


    @pm_utils.log_dbus_method()
    def Reset(self):
        self.Set(
            I_ACTIVATION_TEST, 'ResetCalled', dbus.types.Boolean(True))
        modem_3gpp.Modem3gpp.Reset(self)


class ResetRequiredForActivationModem(TestModem):
    """
    Fake modem boots up as unprovisioned & becomes activated only if it has
    been explicitly activated by calling CompleteCellularActivation
    and has been reset at least once after initiating activatation
    """
    def RegisterWithNetwork(
            self, operator_id='', return_cb=None, raise_cb=None):
        if self.Get(I_ACTIVATION_TEST, 'ResetCalled'):
            modem_3gpp.Modem3gpp.RegisterWithNetwork(
                    self, operator_id, return_cb, raise_cb)


class TestSIM(sim.SIM):
    """ SIM instantiated with the default test network, for ease of use. """
    def __init__(self):
        # Shill's activating ICCID store tracks which SIM identifiers are in
        # the process of activation. If we use the same SIM identifier for
        # every test pass, then a failed test may leave a stale entry in the
        # activating ICCD store which will erroneously mark the SIM as pending
        # activation. So, to avoid this, try to use a unique SIM identifier
        # each time.
        sim_identifier = int(time.time())
        sim.SIM.__init__(
                self,
                sim.SIM.Carrier('test'),
                mm1_constants.MM_MODEM_ACCESS_TECHNOLOGY_LTE,
                msin=str(sim_identifier))