sascha/workload-automation
1
0
Fork 0
mirror of https://github.com/ARM-software/workload-automation.git synced 2025-09-02 03:12:34 +01:00
Code Issues Releases Wiki Activity

Add network check methods to Device and Workload classes

Browse Source 
- Device subclasses should provide their own implementation.
    Default behaviour is to raise a `NotImplementedError`
  - Workload subclasses can set `requires_network` to `True` and
    network connectivity check will be performed during `setup()`
This commit is contained in:
muendelezaji
2016-08-31 16:00:50 +01:00
parent b5c0bdb0eb
commit 392a3f1600
4 changed files with 35 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
wlauto/common/linux/device.py
View File

@@ -39,6 +39,8 @@ FstabEntry = namedtuple('FstabEntry', ['device', 'mount_point', 'fs_type', 'opti
PsEntry = namedtuple('PsEntry', 'user pid ppid vsize rss wchan pc state name')
LsmodEntry = namedtuple('LsmodEntry', ['name', 'size', 'use_count', 'used_by'])
GOOGLE_DNS_SERVER_ADDRESS = '8.8.8.8'
class BaseLinuxDevice(Device): # pylint: disable=abstract-method
@@ -318,25 +320,28 @@ class BaseLinuxDevice(Device): # pylint: disable=abstract-method
please see: https://pythonhosted.org/wlauto/writing_extensions.html""")
def is_network_connected(self, ip_address='8.8.8.8'):
def is_network_connected(self):
"""
Checks for internet connectivity on the device
by pinging IP address provided.
Checks for internet connectivity on the device by pinging IP address provided.
:param ip_address: the IP address to ping
(default - google-public-dns-a.google.com)
:param ip_address: IP address to ping. Default is Google's public DNS server (8.8.8.8)
:returns: ``True`` if internet is available, ``False`` otherwise.
"""
self.logger.debug('Checking for internet connectivity.')
output = self.execute('ping -q -w 1 -c 1 {}'.format(ip_address),
self.logger.debug('Checking for internet connectivity...')
return self._ping_server(GOOGLE_DNS_SERVER_ADDRESS)
def _ping_server(self, ip_address, timeout=1, packet_count=1):
output = self.execute('ping -q -c {} -w {} {}'.format(packet_count, timeout, ip_address),
check_exit_code=False)
if 'network is unreachable' not in output.lower():
self.logger.debug('Found IP address {}'.format(ip_address))
return True
else:
if 'network is unreachable' in output.lower():
self.logger.debug('Cannot find IP address {}'.format(ip_address))
return False
else:
self.logger.debug('Found IP address {}'.format(ip_address))
return True
def get_binary_path(self, name, search_system_binaries=True):
"""