# 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 gobject

from autotest_lib.client.cros.cellular.pseudomodem import modem_cdma
from autotest_lib.client.cros.cellular.pseudomodem import pm_errors
from autotest_lib.client.cros.cellular.pseudomodem import utils as pm_utils

I_ACTIVATION_TEST = 'Interface.CDMAActivationTest'

class UnactivatedCdmaModem(modem_cdma.ModemCdma):
    """ A |ModemCDMA| subclass that starts off unactivated. """
    def __init__(self):
        super(UnactivatedCdmaModem, self).__init__(
                home_network=modem_cdma.ModemCdma.CdmaNetwork(activated=False))


class ActivationRetryModem(modem_cdma.ModemCdma):
    """
    TestModem to test that shill retries OTASP activation until it succeeds.

    """
    def __init__(self, num_activate_retries):
        # This assignment is needed before the call to super.__init__(...)
        self.activate_count = 0
        super(ActivationRetryModem, self).__init__(
                home_network=modem_cdma.ModemCdma.CdmaNetwork(activated=False))
        self._num_activate_retries = num_activate_retries


    def _InitializeProperties(self):
        props = super(ActivationRetryModem, self)._InitializeProperties()

        # For the purposes of this test, introduce a property that
        # stores how many times "Activate" has been called on this
        # modem.
        props[I_ACTIVATION_TEST] = {
            'ActivateCount' : dbus.types.UInt32(self.activate_count)
        }
        return props


    def _IncrementActivateCount(self):
        self.activate_count += 1
        self.Set(I_ACTIVATION_TEST,
                 'ActivateCount',
                 self.activate_count)


    @pm_utils.log_dbus_method(return_cb_arg='return_cb',
                              raise_cb_arg='raise_cb')
    def Activate(self, carrier, return_cb, raise_cb):
        """
        Activation will only succeed on the NUM_ACTIVATE_RETRIESth try.

        """
        self._IncrementActivateCount()
        if (self.activate_count == self._num_activate_retries):
            super(ActivationRetryModem, self).Activate(
                    carrier, return_cb, raise_cb)
        else:
            def _raise_activation_error():
                raise_cb(pm_errors.MMCdmaActivationError(
                         pm_errors.MMCdmaActivationError.START_FAILED))
            gobject.idle_add(_raise_activation_error)