Home | History | Annotate | Download | only in page_capture 
      1 // Copyright (c) 2012 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/base_switches.h"
      6 #include "base/command_line.h"
      7 #include "chrome/browser/extensions/api/page_capture/page_capture_api.h"
      8 #include "chrome/browser/extensions/extension_apitest.h"
      9 #include "chrome/common/chrome_switches.h"
     10 #include "chrome/test/base/ui_test_utils.h"
     11 #include "net/dns/mock_host_resolver.h"
     12 
     13 using extensions::PageCaptureSaveAsMHTMLFunction;
     14 
     15 class ExtensionPageCaptureApiTest : public ExtensionApiTest {
     16  public:
     17   virtual void SetUpCommandLine(CommandLine* command_line) OVERRIDE {
     18     ExtensionApiTest::SetUpCommandLine(command_line);
     19     command_line->AppendSwitchASCII(switches::kJavaScriptFlags, "--expose-gc");
     20   }
     21 };
     22 
     23 class PageCaptureSaveAsMHTMLDelegate
     24     : public PageCaptureSaveAsMHTMLFunction::TestDelegate {
     25  public:
     26   PageCaptureSaveAsMHTMLDelegate() {
     27     PageCaptureSaveAsMHTMLFunction::SetTestDelegate(this);
     28   }
     29 
     30   virtual ~PageCaptureSaveAsMHTMLDelegate() {
     31     PageCaptureSaveAsMHTMLFunction::SetTestDelegate(NULL);
     32   }
     33 
     34   virtual void OnTemporaryFileCreated(
     35       const base::FilePath& temp_file) OVERRIDE {
     36     temp_file_ = temp_file;
     37   }
     38 
     39   base::FilePath temp_file_;
     40 };
     41 
     42 IN_PROC_BROWSER_TEST_F(ExtensionPageCaptureApiTest, SaveAsMHTML) {
     43   host_resolver()->AddRule("www.a.com", "127.0.0.1");
     44   ASSERT_TRUE(StartEmbeddedTestServer());
     45   PageCaptureSaveAsMHTMLDelegate delegate;
     46   ASSERT_TRUE(RunExtensionTest("page_capture")) << message_;
     47   ASSERT_FALSE(delegate.temp_file_.empty());
     48   // Flush the message loops to make sure the delete happens.
     49   content::RunAllPendingInMessageLoop(content::BrowserThread::FILE);
     50   content::RunAllPendingInMessageLoop(content::BrowserThread::IO);
     51   ASSERT_FALSE(base::PathExists(delegate.temp_file_));
     52 }
     53