| OLD | NEW |
| 1 #!/usr/bin/python | 1 #!/usr/bin/python |
| 2 # Copyright (c) 2010 The Chromium Authors. All rights reserved. | 2 # Copyright (c) 2010 The Chromium Authors. All rights reserved. |
| 3 # Use of this source code is governed by a BSD-style license that can be | 3 # Use of this source code is governed by a BSD-style license that can be |
| 4 # found in the LICENSE file. | 4 # found in the LICENSE file. |
| 5 | 5 |
| 6 import commands | 6 import commands |
| 7 import filecmp | 7 import filecmp |
| 8 import logging | 8 import logging |
| 9 import os | 9 import os |
| 10 import shutil | 10 import shutil |
| (...skipping 134 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 145 downloaded_pkg = os.path.join(self.GetDownloadDirectory().value(), | 145 downloaded_pkg = os.path.join(self.GetDownloadDirectory().value(), |
| 146 'a_zip_file.zip') | 146 'a_zip_file.zip') |
| 147 self._ClearLocalDownloadState(downloaded_pkg) | 147 self._ClearLocalDownloadState(downloaded_pkg) |
| 148 | 148 |
| 149 self.RunCommand(pyauto.IDC_NEW_INCOGNITO_WINDOW) # open incognito window | 149 self.RunCommand(pyauto.IDC_NEW_INCOGNITO_WINDOW) # open incognito window |
| 150 # Downloads from incognito window do not figure in GetDownloadsInfo() | 150 # Downloads from incognito window do not figure in GetDownloadsInfo() |
| 151 # since the download manager's list doesn't contain it. | 151 # since the download manager's list doesn't contain it. |
| 152 # Using WaitUntil is the only resort. | 152 # Using WaitUntil is the only resort. |
| 153 self.NavigateToURL(file_url, 1, 0) | 153 self.NavigateToURL(file_url, 1, 0) |
| 154 self.assertTrue(self.WaitUntil(lambda: os.path.exists(downloaded_pkg))) | 154 self.assertTrue(self.WaitUntil(lambda: os.path.exists(downloaded_pkg))) |
| 155 self.assertTrue(self._EqualFileContents(file_path, downloaded_pkg)) | 155 self.assertTrue(self._EqualFileContents(file_path, downloaded_pkg), |
| 156 msg='%s (size %d) and %s (size %d) do not match' % ( |
| 157 file_path, os.path.getsize(file_path), |
| 158 downloaded_pkg, os.path.getsize(downloaded_pkg))) |
| 156 self.assertTrue(self.IsDownloadShelfVisible(1)) | 159 self.assertTrue(self.IsDownloadShelfVisible(1)) |
| 157 | 160 |
| 158 def testSaveDangerousFile(self): | 161 def testSaveDangerousFile(self): |
| 159 """Verify that we can download and save a dangerous file.""" | 162 """Verify that we can download and save a dangerous file.""" |
| 160 file_path = self._GetDangerousDownload() | 163 file_path = self._GetDangerousDownload() |
| 161 downloaded_pkg = os.path.join(self.GetDownloadDirectory().value(), | 164 downloaded_pkg = os.path.join(self.GetDownloadDirectory().value(), |
| 162 os.path.basename(file_path)) | 165 os.path.basename(file_path)) |
| 163 self._ClearLocalDownloadState(downloaded_pkg) | 166 self._ClearLocalDownloadState(downloaded_pkg) |
| 164 self._TriggerUnsafeDownload(os.path.basename(file_path)) | 167 self._TriggerUnsafeDownload(os.path.basename(file_path)) |
| 165 self.PerformActionOnDownload(self._GetDownloadId(), | 168 self.PerformActionOnDownload(self._GetDownloadId(), |
| (...skipping 429 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 595 | 598 |
| 596 # Verify download in Incognito Window. | 599 # Verify download in Incognito Window. |
| 597 # WaitForAllDownloadsToComplete does not wait for incognito downloads | 600 # WaitForAllDownloadsToComplete does not wait for incognito downloads |
| 598 self.assertTrue(self.WaitUntil( | 601 self.assertTrue(self.WaitUntil( |
| 599 lambda: os.path.exists(downloaded_pkg_incog))) | 602 lambda: os.path.exists(downloaded_pkg_incog))) |
| 600 self.assertTrue(self._EqualFileContents(file_path, downloaded_pkg_incog)) | 603 self.assertTrue(self._EqualFileContents(file_path, downloaded_pkg_incog)) |
| 601 | 604 |
| 602 | 605 |
| 603 if __name__ == '__main__': | 606 if __name__ == '__main__': |
| 604 pyauto_functional.Main() | 607 pyauto_functional.Main() |
| OLD | NEW |