1 # Copyright 2018 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 logging 6 import shutil 7 8 from autotest_lib.client.bin import test, utils 9 from autotest_lib.client.common_lib import error 10 from autotest_lib.client.cros.update_engine import update_engine_util 11 12 class UpdateEngineTest(test.test, update_engine_util.UpdateEngineUtil): 13 """Base class for update engine client tests.""" 14 15 _NETWORK_INTERFACES = ['eth0', 'eth1', 'eth2'] 16 17 18 def initialize(self): 19 """Initialize for this test.""" 20 self._create_update_engine_variables() 21 self._internet_was_disabled = False 22 23 24 def cleanup(self): 25 """Cleanup for this test.""" 26 # Make sure to grab the update engine log for every test run. 27 shutil.copy(self._UPDATE_ENGINE_LOG, self.resultsdir) 28 29 # Ensure ethernet adapters are back on 30 self._enable_internet() 31 32 33 def _enable_internet(self, ping_server='google.com'): 34 """ 35 Re-enables the internet connection. 36 37 @param ping_server: The server to ping to check we are online. 38 39 """ 40 if not self._internet_was_disabled: 41 return 42 43 self._internet_was_disabled = False 44 logging.debug('Before reconnect: %s', utils.run('ifconfig')) 45 for eth in self._NETWORK_INTERFACES: 46 utils.run('ifconfig %s up' % eth, ignore_status=True) 47 utils.start_service('recover_duts', ignore_status=True) 48 49 # Print ifconfig to help debug DUTs that stay offline. 50 logging.debug('After reconnect: %s', utils.run('ifconfig')) 51 52 # We can't return right after reconnecting the network or the server 53 # test may not receive the message. So we wait a bit longer for the 54 # DUT to be reconnected. 55 utils.poll_for_condition(lambda: utils.ping(ping_server, 56 tries=3, timeout=10) == 0, 57 timeout=120, 58 sleep_interval=1, 59 exception=error.TestFail( 60 'Ping failed after reconnecting network')) 61 62 63 def _disable_internet(self, ping_server='google.com'): 64 """Disable the internet connection""" 65 self._internet_was_disabled = True 66 try: 67 logging.debug('Before disconnect: %s', utils.run('ifconfig')) 68 # DUTs in the lab have a service called recover_duts that is used to 69 # check that the DUT is online and if it is not it will bring it 70 # back online. We will need to stop this service for the length 71 # of this test. 72 utils.stop_service('recover_duts', ignore_status=True) 73 for eth in self._NETWORK_INTERFACES: 74 result = utils.run('ifconfig %s down' % eth, ignore_status=True) 75 logging.debug(result) 76 77 # Print ifconfig to help debug DUTs that stay online. 78 logging.debug('After disconnect: %s', utils.run('ifconfig')) 79 80 # Make sure we are offline 81 utils.poll_for_condition(lambda: utils.ping(ping_server, 82 deadline=5, 83 timeout=5) != 0, 84 timeout=60, 85 sleep_interval=1, 86 desc='Ping failure while offline.') 87 except (error.CmdError, utils.TimeoutError): 88 logging.exception('Failed to disconnect one or more interfaces.') 89 logging.debug(utils.run('ifconfig', ignore_status=True)) 90 raise error.TestFail('Disabling the internet connection failed.') 91 92
