Move pyauto extensions.py expected files to data dir.

chrome/test/functional/extensions.py

 #!/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'
-and that there is a file of newline-separated urls to visit called 'urls.txt'
-in the same directory as the script.
+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."""
-  # TODO: provide a way in pyauto to pass args to a test and take these as args
-  extensions_dir_ = 'extensions'  # The directory of extensions
-  urls_file_ = 'urls.txt'         # The file which holds a list of urls to visit
 
   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()
@@ skipping 51 matching lines @@
           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"""
-    self.assertTrue(os.path.exists(self.extensions_dir_),
-             'The dir "%s" must exist' % os.path.abspath(self.extensions_dir_))
-    self.assertTrue(os.path.exists(self.urls_file_),
-             'The file "%s" must exist' % os.path.abspath(self.urls_file_))
+    # 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(self.urls_file_).readlines()[:num_urls_to_visit]]
+                open(urls_file).readlines()[:num_urls_to_visit]]
 
-    failed_extensions = glob.glob(os.path.join(self.extensions_dir_, '*.crx'))
+    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()