1 #!/usr/bin/python 2 # 3 # Copyright 2010 Google Inc. All Rights Reserved. 4 """ 5 This allows a site to customize the test creation attributes. 6 7 """ 8 9 10 import common, re 11 from autotest_lib.frontend.afe import models 12 13 14 def _set_attributes_custom(test, data): 15 # We set the test name to the dirname of the control file. 16 test_new_name = test.path.split('/') 17 if test_new_name[-1] == 'control' or test_new_name[-1] == 'control.srv': 18 test.name = test_new_name[-2] 19 else: 20 control_name = "%s:%s" 21 control_name %= (test_new_name[-2], 22 test_new_name[-1]) 23 test.name = re.sub('control.*\.', '', control_name) 24 25 # We set verify to always False (0). 26 test.run_verify = 0 27 28 if hasattr(data, 'test_parameters'): 29 for para_name in data.test_parameters: 30 test_parameter = models.TestParameter.objects.get_or_create( 31 test=test, name=para_name)[0] 32 test_parameter.save() 33