Hooking MHTML generation to the browser.

chrome/browser/download/mhtml_generation_browsertest.cc

Index: chrome/browser/download/mhtml_generation_browsertest.cc
diff --git a/chrome/browser/download/mhtml_generation_browsertest.cc b/chrome/browser/download/mhtml_generation_browsertest.cc
new file mode 100644
index 0000000000000000000000000000000000000000..8413964ed4ff4ec29fba2fffac1b1e6edcca9aee
--- /dev/null
+++ b/chrome/browser/download/mhtml_generation_browsertest.cc
@@ -0,0 +1,75 @@
+// 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/file_path.h"
+#include "base/scoped_temp_dir.h"
+#include "chrome/browser/download/mhtml_generation_manager.h"
+#include "chrome/browser/ui/browser.h"
+#include "chrome/browser/ui/tab_contents/tab_contents_wrapper.h"
+#include "chrome/test/in_process_browser_test.h"
+#include "chrome/test/testing_browser_process.h"
+#include "chrome/test/ui_test_utils.h"
+#include "content/browser/tab_contents/tab_contents.h"
+#include "net/test/test_server.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace {
+
+class MHTMLGenerationTest : public InProcessBrowserTest {
+ public:
+  MHTMLGenerationTest() {}
+
+  FilePath delete_on_teardown_;

[Paweł Hajdan Jr., 2011/06/11 18:04:45]

This is now unused, please remove it.

[Jay Civelli, 2011/06/13 22:45:41]

Done.

+
+ protected:
+  virtual void SetUp() {
+    ASSERT_TRUE(temp_dir_.CreateUniqueTempDir());
+    InProcessBrowserTest::SetUp();
+  }
+
+  virtual void TearDown() {
+    InProcessBrowserTest::TearDown();
+
+    if (!delete_on_teardown_.empty())
+      file_util::Delete(delete_on_teardown_, false);
+  }
+
+  ScopedTempDir temp_dir_;
+};
+
+// Tests that generating a MHTML does create contents.
+// Note that the actual content of the file is not tested, the purpose of this
+// test is to ensure we were successfull in creating the MHTML data from the
+// renderer.
+IN_PROC_BROWSER_TEST_F(MHTMLGenerationTest, GenerateMHTML) {
+  ASSERT_TRUE(test_server()->Start());
+
+  FilePath path(temp_dir_.path());
+  path = path.Append(FILE_PATH_LITERAL("test.mht"));
+
+  ui_test_utils::NavigateToURL(browser(),
+      test_server()->GetURL("files/google/google.html"));
+
+  TabContents* tab = browser()->GetSelectedTabContentsWrapper()->tab_contents();
+  MHTMLGenerationManager* mhtml_generation_manager =
+      g_browser_process->mhtml_generation_manager();
+
+  Source<RenderViewHost> source(tab->render_view_host());
+  ui_test_utils::WindowedNotificationObserverWithDetails<
+      MHTMLGenerationManager::NotificationDetails> signal(
+          NotificationType::MHTML_GENERATED, source);
+  mhtml_generation_manager->GenerateMHTML(tab, path);
+  signal.Wait();
+
+  MHTMLGenerationManager::NotificationDetails details;
+  ASSERT_TRUE(signal.GetDetailsFor(source.map_key(), &details));
+  ASSERT_TRUE(details.success);
+
+  // Make sure the generated file has some contents.
+  int64 file_size;
+  ASSERT_TRUE(file_util::GetFileSize(path, &file_size));
+  EXPECT_GT(file_size, 100);
+}
+
+}  // namespace