OLD | NEW |
| (Empty) |
1 #!/usr/bin/env python | |
2 # | |
3 # Copyright (c) 2013 The Chromium Authors. All rights reserved. | |
4 # Use of this source code is governed by a BSD-style license that can be | |
5 # found in the LICENSE file. | |
6 | |
7 """Runs semi-automated update testing on a non-rooted device.""" | |
8 import logging | |
9 import optparse | |
10 import os | |
11 import shutil | |
12 import sys | |
13 import time | |
14 | |
15 from pylib import android_commands | |
16 | |
17 | |
18 def _SaveAppData(adb, package_name, from_apk=None, data_dir=None): | |
19 def _BackupAppData(data_dir=None): | |
20 adb.Adb().SendCommand('backup %s' % package_name) | |
21 backup_file = os.path.join(os.getcwd(), 'backup.ab') | |
22 assert os.path.exists(backup_file), 'Backup failed.' | |
23 if data_dir: | |
24 if not os.path.isdir(data_dir): | |
25 os.makedirs(data_dir) | |
26 shutil.move(backup_file, data_dir) | |
27 backup_file = os.path.join(data_dir, 'backup.ab') | |
28 print 'Application data saved to %s' % backup_file | |
29 | |
30 if from_apk: | |
31 logging.info('Installing %s...', from_apk) | |
32 output = adb.Install(from_apk, reinstall=True) | |
33 if 'Success' not in output: | |
34 raise Exception('Unable to install %s. output: %s' % (from_apk, output)) | |
35 | |
36 raw_input('Set the application state. Once ready, press enter and ' | |
37 'select "Backup my data" on the device.') | |
38 _BackupAppData(data_dir) | |
39 | |
40 | |
41 def _VerifyAppUpdate(adb, to_apk, app_data, from_apk=None): | |
42 def _RestoreAppData(): | |
43 assert os.path.exists(app_data), 'Backup file does not exist!' | |
44 adb.Adb().SendCommand('restore %s' % app_data) | |
45 # It seems restore command is not synchronous. | |
46 time.sleep(15) | |
47 | |
48 if from_apk: | |
49 logging.info('Installing %s...', from_apk) | |
50 output = adb.Install(from_apk, reinstall=True) | |
51 if 'Success' not in output: | |
52 raise Exception('Unable to install %s. output: %s' % (from_apk, output)) | |
53 | |
54 logging.info('Restoring the application data...') | |
55 raw_input('Press enter and select "Restore my data" on the device.') | |
56 _RestoreAppData() | |
57 | |
58 logging.info('Verifying that %s cannot be installed side-by-side...', | |
59 to_apk) | |
60 output = adb.Install(to_apk) | |
61 if 'INSTALL_FAILED_ALREADY_EXISTS' not in output: | |
62 if 'Success' in output: | |
63 raise Exception('Package name has changed! output: %s' % output) | |
64 else: | |
65 raise Exception(output) | |
66 | |
67 logging.info('Verifying that %s can be overinstalled...', to_apk) | |
68 output = adb.Install(to_apk, reinstall=True) | |
69 if 'Success' not in output: | |
70 raise Exception('Unable to install %s.\n output: %s' % (to_apk, output)) | |
71 logging.info('Successfully updated to the new apk. Please verify that the ' | |
72 'the application data is preserved.') | |
73 | |
74 | |
75 def main(): | |
76 logger = logging.getLogger() | |
77 logger.setLevel(logging.DEBUG) | |
78 desc = ( | |
79 'Performs semi-automated application update verification testing. ' | |
80 'When given --save, it takes a snapshot of the application data ' | |
81 'on the device. (A dialog on the device will prompt the user to grant ' | |
82 'permission to backup the data.) Otherwise, it performs the update ' | |
83 'testing as follows: ' | |
84 '1. Installs the |from-apk| (optional). ' | |
85 '2. Restores the previously stored snapshot of application data ' | |
86 'given by |app-data| ' | |
87 '(A dialog on the device will prompt the user to grant permission to ' | |
88 'restore the data.) ' | |
89 '3. Verifies that |to-apk| cannot be installed side-by-side. ' | |
90 '4. Verifies that |to-apk| can replace |from-apk|.') | |
91 parser = optparse.OptionParser(description=desc) | |
92 parser.add_option('--package-name', help='Package name for the application.') | |
93 parser.add_option('--save', action='store_true', | |
94 help=('Save a snapshot of application data. ' | |
95 'This will be saved as backup.db in the ' | |
96 'current directory if |app-data| directory ' | |
97 'is not specifid.')) | |
98 parser.add_option('--from-apk', | |
99 help=('APK to update from. This is optional if you already ' | |
100 'have the app installed.')) | |
101 parser.add_option('--to-apk', help='APK to update to.') | |
102 parser.add_option('--app-data', | |
103 help=('Path to the application data to be restored or the ' | |
104 'directory where the data should be saved.')) | |
105 (options, args) = parser.parse_args() | |
106 | |
107 if args: | |
108 parser.print_help(sys.stderr) | |
109 parser.error('Unknown arguments: %s.' % args) | |
110 | |
111 if len(android_commands.GetAttachedDevices()) != 1: | |
112 parser.error('Exactly 1 device must be attached.') | |
113 adb = android_commands.AndroidCommands() | |
114 | |
115 if options.from_apk: | |
116 assert os.path.isfile(options.from_apk) | |
117 | |
118 if options.save: | |
119 if not options.package_name: | |
120 parser.print_help(sys.stderr) | |
121 parser.error('Missing --package-name.') | |
122 _SaveAppData(adb, options.package_name, from_apk=options.from_apk, | |
123 data_dir=options.app_data) | |
124 else: | |
125 if not options.to_apk or not options.app_data: | |
126 parser.print_help(sys.stderr) | |
127 parser.error('Missing --to-apk or --app-data.') | |
128 assert os.path.isfile(options.to_apk) | |
129 assert os.path.isfile(options.app_data) | |
130 _VerifyAppUpdate(adb, options.to_apk, options.app_data, | |
131 from_apk=options.from_apk) | |
132 | |
133 | |
134 if __name__ == '__main__': | |
135 main() | |
OLD | NEW |