| OLD | NEW |
| 1 #!/usr/bin/python | 1 #!/usr/bin/python |
| 2 | 2 |
| 3 # Copyright (c) 2011 The Chromium OS Authors. All rights reserved. | 3 # Copyright (c) 2011 The Chromium OS Authors. All rights reserved. |
| 4 # Use of this source code is governed by a BSD-style license that can be | 4 # Use of this source code is governed by a BSD-style license that can be |
| 5 # found in the LICENSE file. | 5 # found in the LICENSE file. |
| 6 | 6 |
| 7 """This module runs a suite of Auto Update tests. | 7 """This module runs a suite of Auto Update tests. |
| 8 | 8 |
| 9 The tests can be run on either a virtual machine or actual device depending | 9 The tests can be run on either a virtual machine or actual device depending |
| 10 on parameters given. Specific tests can be run by invoking --test_prefix. | 10 on parameters given. Specific tests can be run by invoking --test_prefix. |
| 11 Verbose is useful for many of the tests if you want to see individual commands | 11 Verbose is useful for many of the tests if you want to see individual commands |
| 12 being run during the update process. | 12 being run during the update process. |
| 13 """ | 13 """ |
| 14 | 14 |
| 15 import optparse | 15 import optparse |
| 16 import os | 16 import os |
| 17 import re | 17 import re |
| 18 import sys | 18 import sys |
| 19 import unittest | 19 import unittest |
| 20 | 20 |
| 21 sys.path.append(os.path.join(os.path.dirname(__file__), '../lib')) | 21 # TODO(sosa): Migrate to chromite cros_build_lib. |
| 22 import constants |
| 23 sys.path.append(constants.CROSUTILS_LIB_DIR) |
| 22 import cros_build_lib as cros_lib | 24 import cros_build_lib as cros_lib |
| 23 | 25 |
| 24 import au_test | 26 import au_test |
| 25 import au_worker | 27 import au_worker |
| 26 import dummy_au_worker | 28 import dummy_au_worker |
| 27 import dev_server_wrapper | 29 import dev_server_wrapper |
| 28 import parallel_test_job | 30 import parallel_test_job |
| 29 import public_key_manager | 31 import public_key_manager |
| 30 import update_exception | 32 import update_exception |
| 31 | 33 |
| (...skipping 237 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 269 if options.public_key: | 271 if options.public_key: |
| 270 cros_lib.Info('Cleaning up. Removing keys added as part of testing.') | 272 cros_lib.Info('Cleaning up. Removing keys added as part of testing.') |
| 271 target_directory = os.path.dirname(options.target_image) | 273 target_directory = os.path.dirname(options.target_image) |
| 272 for key_manager in au_test.AUTest.public_key_managers: | 274 for key_manager in au_test.AUTest.public_key_managers: |
| 273 if key_manager.image_path.startswith(target_directory): | 275 if key_manager.image_path.startswith(target_directory): |
| 274 key_manager.RemoveKeyFromImage() | 276 key_manager.RemoveKeyFromImage() |
| 275 | 277 |
| 276 | 278 |
| 277 if __name__ == '__main__': | 279 if __name__ == '__main__': |
| 278 main() | 280 main() |
| OLD | NEW |