OLD | NEW |
1 # Copyright 2013 The Chromium Authors. All rights reserved. | 1 # Copyright 2013 The Chromium Authors. All rights reserved. |
2 # Use of this source code is governed by a BSD-style license that can be | 2 # Use of this source code is governed by a BSD-style license that can be |
3 # found in the LICENSE file. | 3 # found in the LICENSE file. |
4 | 4 |
5 """Generates test runner factory and tests for performance tests.""" | 5 """Generates test runner factory and tests for performance tests.""" |
6 | 6 |
7 import json | 7 import json |
8 import fnmatch | 8 import fnmatch |
9 import logging | 9 import logging |
10 import os | 10 import os |
11 import psutil | 11 import psutil |
12 import signal | 12 import signal |
| 13 import shutil |
13 import time | 14 import time |
14 | 15 |
15 from pylib import android_commands | 16 from pylib import android_commands |
16 from pylib import cmd_helper | 17 from pylib import cmd_helper |
17 from pylib import constants | 18 from pylib import constants |
18 from pylib import forwarder | 19 from pylib import forwarder |
19 from pylib import ports | 20 from pylib import ports |
20 | 21 |
21 import test_runner | 22 import test_runner |
22 | 23 |
(...skipping 28 matching lines...) Expand all Loading... |
51 | 52 |
52 def Setup(test_options): | 53 def Setup(test_options): |
53 """Create and return the test runner factory and tests. | 54 """Create and return the test runner factory and tests. |
54 | 55 |
55 Args: | 56 Args: |
56 test_options: A PerformanceOptions object. | 57 test_options: A PerformanceOptions object. |
57 | 58 |
58 Returns: | 59 Returns: |
59 A tuple of (TestRunnerFactory, tests). | 60 A tuple of (TestRunnerFactory, tests). |
60 """ | 61 """ |
61 if not os.path.exists(constants.PERF_OUTPUT_DIR): | 62 if os.path.exists(constants.PERF_OUTPUT_DIR): |
62 os.makedirs(constants.PERF_OUTPUT_DIR) | 63 shutil.rmtree(constants.PERF_OUTPUT_DIR) |
| 64 os.makedirs(constants.PERF_OUTPUT_DIR) |
63 | 65 |
64 # Before running the tests, kill any leftover server. | 66 # Before running the tests, kill any leftover server. |
65 _KillPendingServers() | 67 _KillPendingServers() |
66 | 68 |
67 with file(test_options.steps, 'r') as f: | 69 with file(test_options.steps, 'r') as f: |
68 tests = json.load(f) | 70 tests = json.load(f) |
69 | 71 |
70 # The list is necessary to keep the steps order, but internally | 72 # The list is necessary to keep the steps order, but internally |
71 # the format is squashed from a list of lists into a single dict: | 73 # the format is squashed from a list of lists into a single dict: |
72 # [["A", "cmd"], ["B", "cmd"]] into {"A": "cmd", "B": "cmd"} | 74 # [["A", "cmd"], ["B", "cmd"]] into {"A": "cmd", "B": "cmd"} |
73 sorted_test_names = [i[0] for i in tests] | 75 sorted_test_names = [i[0] for i in tests] |
74 tests_dict = dict(tests) | 76 tests_dict = dict(tests) |
75 | 77 |
76 if test_options.test_filter: | 78 if test_options.test_filter: |
77 sorted_test_names = fnmatch.filter(sorted_test_names, | 79 sorted_test_names = fnmatch.filter(sorted_test_names, |
78 test_options.test_filter) | 80 test_options.test_filter) |
79 tests_dict = dict((k, v) for k, v in tests_dict.iteritems() | 81 tests_dict = dict((k, v) for k, v in tests_dict.iteritems() |
80 if k in sorted_test_names) | 82 if k in sorted_test_names) |
81 | 83 |
82 flaky_steps = [] | 84 flaky_steps = [] |
83 if test_options.flaky_steps: | 85 if test_options.flaky_steps: |
84 with file(test_options.flaky_steps, 'r') as f: | 86 with file(test_options.flaky_steps, 'r') as f: |
85 flaky_steps = json.load(f) | 87 flaky_steps = json.load(f) |
86 | 88 |
87 def TestRunnerFactory(device, shard_index): | 89 def TestRunnerFactory(device, shard_index): |
88 return test_runner.TestRunner( | 90 return test_runner.TestRunner( |
89 test_options, device, tests_dict, flaky_steps) | 91 test_options, device, tests_dict, flaky_steps) |
90 | 92 |
91 return (TestRunnerFactory, sorted_test_names) | 93 return (TestRunnerFactory, sorted_test_names) |
OLD | NEW |