Workarounding fix for save complete page problems.

content/browser/download/save_package_browsertest.cc

 // Copyright (c) 2011 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.
 
 #include "base/files/scoped_temp_dir.h"
 #include "content/browser/download/save_package.h"
+#include "content/public/browser/browser_context.h"
+#include "content/public/browser/download_manager.h"
+#include "content/public/browser/web_contents.h"
 #include "content/public/test/content_browser_test.h"
 #include "content/public/test/content_browser_test_utils.h"
+#include "content/public/test/test_utils.h"
 #include "content/shell/browser/shell.h"
 
 namespace content {
 
 const char kTestFile[] = "files/simple_page.html";
+const char kIframeDocumentWrite[] = "files/iframe_doc_write.htm";
 
 class SavePackageBrowserTest : public ContentBrowserTest {
  protected:
   virtual void SetUp() OVERRIDE {
     ASSERT_TRUE(save_dir_.CreateUniqueTempDir());
     ContentBrowserTest::SetUp();
   }
 
   // Returns full paths of destination file and directory.
   void GetDestinationPaths(const std::string& prefix,
                            base::FilePath* full_file_name,
                            base::FilePath* dir) {
     *full_file_name = save_dir_.path().AppendASCII(prefix + ".htm");
     *dir = save_dir_.path().AppendASCII(prefix + "_files");
   }
 
   // Temporary directory we will save pages to.
   base::ScopedTempDir save_dir_;
 };
 
+class SavePackageFinishedObserver : public DownloadManager::Observer {
+ public:
+  SavePackageFinishedObserver(content::DownloadManager* manager,
+                              const base::Closure& callback)
+                              : download_manager_(manager),
+                                callback_(callback),
+                                successfully_finished_(false) {
+    download_manager_->AddObserver(this);
+  }
+
+  virtual ~SavePackageFinishedObserver() {
+    if (download_manager_)
+      download_manager_->RemoveObserver(this);
+  }
+
+  // DownloadManager::Observer:
+  virtual void OnSavePackageFinished(
+      bool success,
+      content::DownloadManager* manager,
+      content::DownloadItem* item) OVERRIDE{
+    successfully_finished_ = success;
+    callback_.Run();
+  }
+
+  virtual void ManagerGoingDown(content::DownloadManager* manager) OVERRIDE{
+    download_manager_->RemoveObserver(this);
+    download_manager_ = NULL;
+  }
+
+  bool successfully_finished() const { return successfully_finished_; }
+
+private:
+  content::DownloadManager* download_manager_;
+  base::Closure callback_;
+
+  bool successfully_finished_;
+
+  DISALLOW_COPY_AND_ASSIGN(SavePackageFinishedObserver);
+};
+
 // Create a SavePackage and delete it without calling Init.
 // SavePackage dtor has various asserts/checks that should not fire.
 IN_PROC_BROWSER_TEST_F(SavePackageBrowserTest, ImplicitCancel) {
   ASSERT_TRUE(test_server()->Start());
   GURL url = test_server()->GetURL(kTestFile);
   NavigateToURL(shell(), url);
   base::FilePath full_file_name, dir;
   GetDestinationPaths("a", &full_file_name, &dir);
   scoped_refptr<SavePackage> save_package(new SavePackage(
       shell()->web_contents(), SAVE_PAGE_TYPE_AS_ONLY_HTML, full_file_name,
       dir));
   ASSERT_TRUE(test_server()->Stop());
 }
 
 // Create a SavePackage, call Cancel, then delete it.
 // SavePackage dtor has various asserts/checks that should not fire.
 IN_PROC_BROWSER_TEST_F(SavePackageBrowserTest, ExplicitCancel) {
   ASSERT_TRUE(test_server()->Start());
   GURL url = test_server()->GetURL(kTestFile);
   NavigateToURL(shell(), url);
   base::FilePath full_file_name, dir;
   GetDestinationPaths("a", &full_file_name, &dir);
   scoped_refptr<SavePackage> save_package(new SavePackage(
       shell()->web_contents(), SAVE_PAGE_TYPE_AS_ONLY_HTML, full_file_name,
       dir));
   save_package->Cancel(true);
   ASSERT_TRUE(test_server()->Stop());
 }
 
+// We should not cancel saving a main document html file in case of an
+// having the same url. That is possible in the used TC.
+IN_PROC_BROWSER_TEST_F(SavePackageBrowserTest, DuplicateURLForFrames) {
+  ASSERT_TRUE(test_server()->Start());
+  GURL url = test_server()->GetURL(kIframeDocumentWrite);
+  NavigateToURL(shell(), url);
+
+  scoped_refptr<content::MessageLoopRunner> loop_runner(
+      new MessageLoopRunner);
+  SavePackageFinishedObserver observer(
+      BrowserContext::GetDownloadManager(
+          shell()->web_contents()->GetBrowserContext()),
+      loop_runner->QuitClosure());
+
+  base::FilePath full_file_name, dir;
+  GetDestinationPaths("a", &full_file_name, &dir);
+  ASSERT_TRUE(shell()->web_contents()->SavePage(
+      full_file_name,
+      dir,
+      SAVE_PAGE_TYPE_AS_COMPLETE_HTML));
+
+  loop_runner->Run();
+
+  EXPECT_TRUE(observer.successfully_finished());
+}
+
 }  // namespace content