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 5 import dbus 6 import dbus.types 7 import gobject 8 9 from autotest_lib.client.cros.cellular.pseudomodem import modem_cdma 10 from autotest_lib.client.cros.cellular.pseudomodem import pm_errors 11 from autotest_lib.client.cros.cellular.pseudomodem import utils as pm_utils 12 13 I_ACTIVATION_TEST = 'Interface.CDMAActivationTest' 14 15 class 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 22 class 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