| OLD | NEW |
| 1 #!/usr/bin/python | 1 #!/usr/bin/python |
| 2 # | 2 # |
| 3 # Copyright 2010 Google Inc. All Rights Reserved. | 3 # Copyright 2010 Google Inc. All Rights Reserved. |
| 4 """ | 4 """ |
| 5 This allows a site to customize the test creation attributes. | 5 This allows a site to customize the test creation attributes. |
| 6 | 6 |
| 7 """ | 7 """ |
| 8 | 8 |
| 9 | 9 |
| 10 import common, re | 10 import common, re |
| 11 from autotest_lib.frontend.afe import models |
| 11 | 12 |
| 12 | 13 |
| 13 def _set_attributes_custom(test, data): | 14 def _set_attributes_custom(test, data): |
| 14 # We set the test name to the dirname of the control file. | 15 # We set the test name to the dirname of the control file. |
| 15 test_new_name = test.path.split('/') | 16 test_new_name = test.path.split('/') |
| 16 if test_new_name[-1] == 'control' or test_new_name[-1] == 'control.srv': | 17 if test_new_name[-1] == 'control' or test_new_name[-1] == 'control.srv': |
| 17 test.name = test_new_name[-2] | 18 test.name = test_new_name[-2] |
| 18 else: | 19 else: |
| 19 control_name = "%s:%s" | 20 control_name = "%s:%s" |
| 20 control_name %= (test_new_name[-2], | 21 control_name %= (test_new_name[-2], |
| 21 test_new_name[-1]) | 22 test_new_name[-1]) |
| 22 test.name = re.sub('control.*\.', '', control_name) | 23 test.name = re.sub('control.*\.', '', control_name) |
| 23 | 24 |
| 24 # We set verify to always False (0). | 25 # We set verify to always False (0). |
| 25 test.run_verify = 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() |
| OLD | NEW |