OLD | NEW |
| (Empty) |
1 # Copyright (c) 2011 The Chromium OS Authors. All rights reserved. | |
2 # Use of this source code is governed by a BSD-style license that can be | |
3 # found in the LICENSE file. | |
4 | |
5 """Module containing class that implements an au_worker for a test device.""" | |
6 | |
7 import unittest | |
8 | |
9 import cros_build_lib as cros_lib | |
10 | |
11 import au_worker | |
12 | |
13 class RealAUWorker(au_worker.AUWorker): | |
14 """Test harness for updating real images.""" | |
15 | |
16 def __init__(self, options, test_results_root): | |
17 """Processes non-vm-specific options.""" | |
18 au_worker.AUWorker.__init__(self, options, test_results_root) | |
19 self.remote = options.remote | |
20 if not self.remote: cros_lib.Die('We require a remote address for tests.') | |
21 | |
22 def PrepareBase(self, image_path): | |
23 """Auto-update to base image to prepare for test.""" | |
24 self.PrepareRealBase(image_path) | |
25 | |
26 def UpdateImage(self, image_path, src_image_path='', stateful_change='old', | |
27 proxy_port=None, private_key_path=None): | |
28 """Updates a remote image using image_to_live.sh.""" | |
29 stateful_change_flag = self.GetStatefulChangeFlag(stateful_change) | |
30 cmd = ['%s/image_to_live.sh' % self.crosutils, | |
31 '--remote=%s' % self.remote, | |
32 stateful_change_flag, | |
33 '--verify', | |
34 ] | |
35 self.AppendUpdateFlags(cmd, image_path, src_image_path, proxy_port, | |
36 private_key_path) | |
37 self.RunUpdateCmd(cmd) | |
38 | |
39 def UpdateUsingPayload(self, update_path, stateful_change='old', | |
40 proxy_port=None): | |
41 """Updates a remote image using image_to_live.sh.""" | |
42 stateful_change_flag = self.GetStatefulChangeFlag(stateful_change) | |
43 cmd = ['%s/image_to_live.sh' % self.crosutils, | |
44 '--payload=%s' % update_path, | |
45 '--remote=%s' % self.remote, | |
46 stateful_change_flag, | |
47 '--verify', | |
48 ] | |
49 if proxy_port: cmd.append('--proxy_port=%s' % proxy_port) | |
50 self.RunUpdateCmd(cmd) | |
51 | |
52 def VerifyImage(self, unittest, percent_required_to_pass=100): | |
53 """Verifies an image using run_remote_tests.sh with verification suite.""" | |
54 test_directory = self.GetNextResultsPath('verify') | |
55 output = cros_lib.RunCommand( | |
56 ['%s/run_remote_tests.sh' % self.crosutils, | |
57 '--remote=%s' % self.remote, | |
58 '--results_dir_root=%s' % test_directory, | |
59 self.verify_suite, | |
60 ], error_ok=True, enter_chroot=False, redirect_stdout=True) | |
61 return self.AssertEnoughTestsPassed(unittest, output, | |
62 percent_required_to_pass) | |
63 | |
OLD | NEW |