| OLD | NEW | 
|---|
|  | (Empty) | 
| 1 #!/usr/bin/env python |  | 
| 2 # Copyright (c) 2012 The Chromium Authors. All rights reserved. |  | 
| 3 # Use of this source code is governed by a BSD-style license that can be |  | 
| 4 # found in the LICENSE file. |  | 
| 5 |  | 
| 6 import os |  | 
| 7 |  | 
| 8 import pyauto_functional  # must be imported before pyauto |  | 
| 9 |  | 
| 10 import chromeos.file_browser |  | 
| 11 import pyauto |  | 
| 12 import test_utils |  | 
| 13 |  | 
| 14 |  | 
| 15 class DocViewingTest(pyauto.PyUITest): |  | 
| 16   """Basic tests for ChromeOS document viewing. |  | 
| 17 |  | 
| 18   Requires ChromeOS to be logged in. |  | 
| 19   """ |  | 
| 20 |  | 
| 21   def setUp(self): |  | 
| 22     pyauto.PyUITest.setUp(self) |  | 
| 23     extension_path = '/opt/google/chrome/extensions'\ |  | 
| 24         '/gbkeegbaiigmenfmjfclcdgdpimamgkj.crx' |  | 
| 25     # If crx file with doesn't exist, component extensions should be used. |  | 
| 26     if os.path.exists(extension_path): |  | 
| 27       ext_id = self.InstallExtension(extension_path, from_webstore=True) |  | 
| 28       self.assertTrue(ext_id, msg='Failed to install extension %s' % |  | 
| 29                       extension_path) |  | 
| 30 |  | 
| 31   def _GetFullPageFileBrowser(self): |  | 
| 32     """Display the full page file browser. |  | 
| 33 |  | 
| 34     Returns: |  | 
| 35       ChromeosFileBrowser object. |  | 
| 36     """ |  | 
| 37     self.NavigateToURL('chrome://files/#/Downloads') |  | 
| 38     executor = pyauto.PyUITest.JavascriptExecutorInTab(self) |  | 
| 39     file_browser = chromeos.file_browser.FileBrowser(self, executor) |  | 
| 40     if file_browser.WaitUntilInitialized(): |  | 
| 41       return file_browser |  | 
| 42     else: |  | 
| 43       return None |  | 
| 44 |  | 
| 45   def testOpenOfficeFiles(self): |  | 
| 46     """Test we can open office files from the file manager.""" |  | 
| 47     path = os.path.abspath(os.path.join(self.DataDir(), |  | 
| 48                                         'pyauto_private', 'office')) |  | 
| 49     # Copy sample files to Downloads directory. |  | 
| 50     for (path, dirs, private_office_files) in os.walk(path): |  | 
| 51       # Open sample files: .ppt, .pptx, .doc, .docx, xls, xlsx. |  | 
| 52       for fname in private_office_files: |  | 
| 53         test_utils.CopyFileFromDataDirToDownloadDir(self, os.path.join(path, |  | 
| 54                                                                        fname)) |  | 
| 55         file_browser = self._GetFullPageFileBrowser() |  | 
| 56         self.assertTrue(file_browser, msg='File browser failed to initialize.') |  | 
| 57 |  | 
| 58         def _SelectFile(): |  | 
| 59           try: |  | 
| 60             file_browser.Select(fname) |  | 
| 61             return True |  | 
| 62           except AssertionError: |  | 
| 63             return False |  | 
| 64 |  | 
| 65         self.assertTrue(self.WaitUntil(_SelectFile), |  | 
| 66                         msg='"%s" does not exist.' % fname) |  | 
| 67         file_browser.ExecuteDefaultTask() |  | 
| 68         self.assertTrue(self.WaitUntil(self.GetActiveTabTitle, |  | 
| 69                         expect_retval=fname), |  | 
| 70                         msg='"%s" does not open.' % fname) |  | 
| 71         # Close the document viewing tab after use. |  | 
| 72         self.CloseTab(tab_index=1) |  | 
| 73 |  | 
| 74 |  | 
| 75 if __name__ == '__main__': |  | 
| 76   pyauto_functional.Main() |  | 
| OLD | NEW | 
|---|