blob: 8f5e97df1238a6e6f054ae05e41d145f606c2ac3 [file] [log] [blame]
# Copyright (c) 2011 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 optparse, pickle, re, subprocess
import cellular, labconfig_data
class LabConfigError(Exception):
def get_interface_ip(interface='eth0'):
"""Returns the IP address for an interface, or None if not found."""
# We'd like to use
# utils.system_output('ifconfig eth0 2>&1', retain_output=True)
# but that gives us a dependency on the rest of autotest, which
# means that running the unit test requires pythonpath manipulation
ifconfig = subprocess.Popen(['/sbin/ifconfig', interface],
match ='inet addr:([0-9.]+)', ifconfig)
if not match:
return None
class Configuration(object):
"""Configuration for a cellular test.
This includes things like the address of the cell emulator device
and details of the RF switching between the emulated basestation
and the DUT."""
def __init__(self, args):
# For server tests, this constructor runs as part of the
# server control file, on whatever machine the test was
# started on.
parser = optparse.OptionParser()
# Record our args so we can serialize ourself.
self.args = args
self.ip = None
parser.add_option('--cell', dest='cell', default=None,
help='Cellular test cell to use')
'--technology', dest='technology', default='all',
help='Radio access technologies to use (e.g. "WCDMA")')
(self.options, _) = parser.parse_args(args)
self.cell = self._get_cell(self.options.cell)
def _get_cell(self, name):
"""Extracts the named cell from labconfig_data.CELLS."""
if not name:
raise LabConfigError(
'Could not find --cell argument. ' +
'To specify a cell, pass --args=--cell=foo to run_remote_tests')
if name not in labconfig_data.CELLS:
raise LabConfigError(
'Could not find cell %s, valid cells are %s' % (
name, labconfig_data.CELLS.keys()))
return labconfig_data.CELLS[name]
def _get_dut(self, machine=None):
"""Returns the DUT record for machine from cell["duts"]
machine: name or IP of machine. None: for "the current machine".
Right now, we use the interface of eth0 to figure out which
machine we're running on. The important thing is that this
matches the IP address in the cell duts configuration. We'll
have to come up with a better way if this proves brittle."""
if self.ip and not machine:
machine = self.ip
if not machine:
machine = get_interface_ip('eth0')
self.ip = machine
if not machine:
machine = get_interface_ip('eth_test')
self.ip = machine
if not machine:
raise LabConfigError(
'Could not determine which machine we are. Cell = %s' %
for dut in self.cell["duts"]:
if machine == dut["address"] or machine == dut["name"]:
return dut
raise LabConfigError(
'Could not find DUT in lab configuration. Cell = %s' %
def get_technologies(self, machine=None):
"""Gets technologies to use for machine; defaults to all available."""
technologies_list =',')
if 'all' in technologies_list:
m = self._get_dut(machine)
technologies_list = m["technologies"]
enums = [getattr(cellular.Technology, t, None)
for t in technologies_list]
if None in enums:
raise LabConfigError(
'Could not understand a technology in %s' % technologies_list)
return enums
def get_rf_switch_port(self, machine=None):
"""Get the RF Switch Port for the specified machine."""
dut = self._get_dut(machine)
return dut['rf_switch_port']
def get_pickle(self):
return pickle.dumps(self)