def encode_ascii(control_file):
"""Force a control file to only contain ascii characters.
@param control_file: Control file to encode.
@returns the control file in an ascii encoding.
@raises error.ControlFileMalformed: if encoding fails.
"""
try:
return control_file.encode('ascii')
except UnicodeDecodeError as e:
raise error.ControlFileMalformed(str(e))
def _get_control_file_contents_by_name(build, ds, suite_name):
"""Return control file contents for |suite_name|.
Query the dev server at |ds| for the control file |suite_name|, included
in |build| for |board|.
@param build: unique name by which to refer to the image from now on.
@param ds: a dev_server.DevServer instance to fetch control file with.
@param suite_name: canonicalized suite name, e.g. test_suites/control.bvt.
@raises ControlFileNotFound if a unique suite control file doesn't exist.
@raises NoControlFileList if we can't list the control files at all.
@raises ControlFileEmpty if the control file exists on the server, but
can't be read.
@return the contents of the desired control file.
"""
getter = control_file_getter.DevServerGetter.create(build, ds)
timer = autotest_stats.Timer('control_files.parse.%s.%s' %
(ds.get_server_name(ds.url()).replace(
'.', '_'), suite_name.rsplit('.')[-1]))
# Get the control file for the suite.
try:
with timer:
control_file_in = getter.get_control_file_contents_by_name(
suite_name)
except error.CrosDynamicSuiteException as e:
raise type(e)("%s while testing %s." % (e, build))
if not control_file_in:
raise error.ControlFileEmpty("Fetching %s returned no data." %
suite_name)
# Force control files to only contain ascii characters.
try:
control_file_in.encode('ascii')
except UnicodeDecodeError as e:
raise error.ControlFileMalformed(str(e))
return control_file_in