Home | History | Annotate | Download | only in download 
      1 // Copyright (c) 2011 The Chromium Authors. All rights reserved.
      2 // Use of this source code is governed by a BSD-style license that can be
      3 // found in the LICENSE file.
      4 
      5 #include "base/files/scoped_temp_dir.h"
      6 #include "content/browser/download/save_package.h"
      7 #include "content/public/test/content_browser_test.h"
      8 #include "content/public/test/content_browser_test_utils.h"
      9 #include "content/shell/browser/shell.h"
     10 
     11 namespace content {
     12 
     13 const char kTestFile[] = "files/simple_page.html";
     14 
     15 class SavePackageBrowserTest : public ContentBrowserTest {
     16  protected:
     17   virtual void SetUp() OVERRIDE {
     18     ASSERT_TRUE(save_dir_.CreateUniqueTempDir());
     19     ContentBrowserTest::SetUp();
     20   }
     21 
     22   // Returns full paths of destination file and directory.
     23   void GetDestinationPaths(const std::string& prefix,
     24                            base::FilePath* full_file_name,
     25                            base::FilePath* dir) {
     26     *full_file_name = save_dir_.path().AppendASCII(prefix + ".htm");
     27     *dir = save_dir_.path().AppendASCII(prefix + "_files");
     28   }
     29 
     30   // Temporary directory we will save pages to.
     31   base::ScopedTempDir save_dir_;
     32 };
     33 
     34 // Create a SavePackage and delete it without calling Init.
     35 // SavePackage dtor has various asserts/checks that should not fire.
     36 IN_PROC_BROWSER_TEST_F(SavePackageBrowserTest, ImplicitCancel) {
     37   ASSERT_TRUE(test_server()->Start());
     38   GURL url = test_server()->GetURL(kTestFile);
     39   NavigateToURL(shell(), url);
     40   base::FilePath full_file_name, dir;
     41   GetDestinationPaths("a", &full_file_name, &dir);
     42   scoped_refptr<SavePackage> save_package(new SavePackage(
     43       shell()->web_contents(), SAVE_PAGE_TYPE_AS_ONLY_HTML, full_file_name,
     44       dir));
     45   ASSERT_TRUE(test_server()->Stop());
     46 }
     47 
     48 // Create a SavePackage, call Cancel, then delete it.
     49 // SavePackage dtor has various asserts/checks that should not fire.
     50 IN_PROC_BROWSER_TEST_F(SavePackageBrowserTest, ExplicitCancel) {
     51   ASSERT_TRUE(test_server()->Start());
     52   GURL url = test_server()->GetURL(kTestFile);
     53   NavigateToURL(shell(), url);
     54   base::FilePath full_file_name, dir;
     55   GetDestinationPaths("a", &full_file_name, &dir);
     56   scoped_refptr<SavePackage> save_package(new SavePackage(
     57       shell()->web_contents(), SAVE_PAGE_TYPE_AS_ONLY_HTML, full_file_name,
     58       dir));
     59   save_package->Cancel(true);
     60   ASSERT_TRUE(test_server()->Stop());
     61 }
     62 
     63 }  // namespace content
     64