1# Copyright (c) 2014 The Chromium OS Authors. All rights reserved.
2# Use of this source code is governed by a BSD-style license that can be
3# found in the LICENSE file.
4
5import dbus
6import dbus.types
7import gobject
8
9from autotest_lib.client.cros.cellular.pseudomodem import modem_cdma
10from autotest_lib.client.cros.cellular.pseudomodem import pm_errors
11from autotest_lib.client.cros.cellular.pseudomodem import utils as pm_utils
12
13I_ACTIVATION_TEST = 'Interface.CDMAActivationTest'
14
15class UnactivatedCdmaModem(modem_cdma.ModemCdma):
16    """ A |ModemCDMA| subclass that starts off unactivated. """
17    def __init__(self):
18        super(UnactivatedCdmaModem, self).__init__(
19                home_network=modem_cdma.ModemCdma.CdmaNetwork(activated=False))
20
21
22class ActivationRetryModem(modem_cdma.ModemCdma):
23    """
24    TestModem to test that shill retries OTASP activation until it succeeds.
25
26    """
27    def __init__(self, num_activate_retries):
28        # This assignment is needed before the call to super.__init__(...)
29        self.activate_count = 0
30        super(ActivationRetryModem, self).__init__(
31                home_network=modem_cdma.ModemCdma.CdmaNetwork(activated=False))
32        self._num_activate_retries = num_activate_retries
33
34
35    def _InitializeProperties(self):
36        props = super(ActivationRetryModem, self)._InitializeProperties()
37
38        # For the purposes of this test, introduce a property that
39        # stores how many times "Activate" has been called on this
40        # modem.
41        props[I_ACTIVATION_TEST] = {
42            'ActivateCount' : dbus.types.UInt32(self.activate_count)
43        }
44        return props
45
46
47    def _IncrementActivateCount(self):
48        self.activate_count += 1
49        self.Set(I_ACTIVATION_TEST,
50                 'ActivateCount',
51                 self.activate_count)
52
53
54    @pm_utils.log_dbus_method(return_cb_arg='return_cb',
55                              raise_cb_arg='raise_cb')
56    def Activate(self, carrier, return_cb, raise_cb):
57        """
58        Activation will only succeed on the NUM_ACTIVATE_RETRIESth try.
59
60        """
61        self._IncrementActivateCount()
62        if (self.activate_count == self._num_activate_retries):
63            super(ActivationRetryModem, self).Activate(
64                    carrier, return_cb, raise_cb)
65        else:
66            def _raise_activation_error():
67                raise_cb(pm_errors.MMCdmaActivationError(
68                         pm_errors.MMCdmaActivationError.START_FAILED))
69            gobject.idle_add(_raise_activation_error)
70