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 |