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