1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
|
#!/usr/bin/python
# Copyright (c) 2010 The Chromium Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""
This module is a simple qa tool that installs extensions and tests whether the
browser crashes while visiting a list of urls.
Usage: python extensions.py -v
Note: This assumes that there is a directory of extensions called
'extensions-tool' and that there is a file of newline-separated urls to visit
called 'urls.txt' in the data directory.
"""
import glob
import logging
import os
import sys
import pyauto_functional # must be imported before pyauto
import pyauto
class ExtensionsTest(pyauto.PyUITest):
"""Test of extensions."""
def Debug(self):
"""Test method for experimentation.
This method is not run automatically.
"""
while True:
raw_input('Interact with the browser and hit <enter> to dump history.. ')
print '*' * 20
extensions = self.GetExtensionsInfo()
import pprint
pp = pprint.PrettyPrinter(indent=2)
pp.pprint(extensions)
def _GetInstalledExtensionIds(self):
return [extension['id'] for extension in self.GetExtensionsInfo()]
def _ReturnCrashingExtensions(self, extensions, group_size, top_urls):
"""Install the given extensions in groups of group_size and return the
group of extensions that crashes (if any).
Args:
extensions: A list of extensions to install.
group_size: The number of extensions to install at one time.
top_urls: The list of top urls to visit.
Returns:
The extensions in the crashing group or None if there is no crash.
"""
curr_extension = 0
num_extensions = len(extensions)
self.RestartBrowser()
orig_extension_ids = self._GetInstalledExtensionIds()
while curr_extension < num_extensions:
logging.debug('New group of %d extensions.' % group_size)
group_end = curr_extension + group_size
for extension in extensions[curr_extension:group_end]:
logging.debug('Installing extension: %s' % extension)
self.InstallExtension(pyauto.FilePath(extension), False)
for url in top_urls:
self.NavigateToURL(url)
def _LogAndReturnCrashing():
crashing_extensions = extensions[curr_extension:group_end]
logging.debug('Crashing extensions: %s' % crashing_extensions)
return crashing_extensions
# If the browser has crashed, return the extensions in the failing group.
try:
num_browser_windows = self.GetBrowserWindowCount()
except:
return _LogAndReturnCrashing()
else:
if not num_browser_windows:
return _LogAndReturnCrashing()
else:
# Uninstall all extensions that aren't installed by default.
new_extension_ids = [id for id in self._GetInstalledExtensionIds()
if id not in orig_extension_ids]
for extension_id in new_extension_ids:
self.UninstallExtensionById(extension_id)
curr_extension = group_end
# None of the extensions crashed.
return None
def testExtensionCrashes(self):
"""Add top extensions; confirm browser stays up when visiting top urls"""
# TODO: provide a way in pyauto to pass args to a test - take these as args
extensions_dir = os.path.join(self.DataDir(), 'extensions-tool')
urls_file = os.path.join(self.DataDir(), 'urls.txt')
assert(os.path.exists(extensions_dir),
'The dir "%s" must exist' % os.path.abspath(extensions_dir))
assert(os.path.exists(urls_file),
'The file "%s" must exist' % os.path.abspath(urls_file))
num_urls_to_visit = 100
extensions_group_size = 20
top_urls = [l.rstrip() for l in
open(urls_file).readlines()[:num_urls_to_visit]]
failed_extensions = glob.glob(os.path.join(extensions_dir, '*.crx'))
group_size = extensions_group_size
while(group_size and failed_extensions):
failed_extensions = self._ReturnCrashingExtensions(
failed_extensions, group_size, top_urls)
group_size = group_size // 2
self.assertFalse(failed_extensions,
'Extension(s) in failing group: %s' % failed_extensions)
if __name__ == '__main__':
pyauto_functional.Main()
|
