src/chrome/browser/ui/pdf/pdf_browsertest_base.h

   1 // Copyright 2014 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 #ifndef CHROME_BROWSER_UI_PDF_PDF_BROWSERTEST_BASE_H_
   6 #define CHROME_BROWSER_UI_PDF_PDF_BROWSERTEST_BASE_H_
   7
   8 #include <string>
   9
  10 #include "base/files/file_path.h"
  11 #include "chrome/test/base/in_process_browser_test.h"
  12 #include "content/public/browser/notification_observer.h"
  13 #include "net/test/embedded_test_server/embedded_test_server.h"
  14
  15 namespace base {
  16 class CommandLine;
  17 }
  18
  19 class SkBitmap;
  20
  21 // This class is in its own separate file because it is used in both
  22 // browser_tests and interactive_ui_tests.
  23 class PDFBrowserTest : public InProcessBrowserTest,
  24                        public testing::WithParamInterface<int>,
  25                        public content::NotificationObserver {
  26  public:
  27   PDFBrowserTest();
  28   virtual ~PDFBrowserTest();
  29
  30  protected:
  31   // Use our own TestServer so that we can serve files from the pdf directory.
  32   net::test_server::EmbeddedTestServer* pdf_test_server() {
  33     return &pdf_test_server_;
  34   }
  35
  36   int load_stop_notification_count() const {
  37     return load_stop_notification_count_;
  38   }
  39
  40   void Load();
  41   void WaitForResponse();
  42   bool VerifySnapshot(const std::string& expected_filename);
  43
  44  private:
  45   void CopyFromBackingStoreCallback(bool success, const SkBitmap& bitmap);
  46
  47   // content::NotificationObserver
  48   virtual void Observe(int type,
  49                        const content::NotificationSource& source,
  50                        const content::NotificationDetails& details) OVERRIDE;
  51
  52   // InProcessBrowserTest
  53   virtual void SetUpCommandLine(base::CommandLine* command_line) OVERRIDE;
  54
  55   // True if the snapshot differed from the expected value.
  56   bool snapshot_different_;
  57   // Internal variable used to synchronize to the renderer.
  58   int next_dummy_search_value_;
  59   // The filename of the bitmap to compare the snapshot to.
  60   std::string expected_filename_;
  61   // If the snapshot is different, holds the location where it's saved.
  62   base::FilePath snapshot_filename_;
  63   // How many times we've seen chrome::LOAD_STOP.
  64   int load_stop_notification_count_;
  65
  66   net::test_server::EmbeddedTestServer pdf_test_server_;
  67
  68   DISALLOW_COPY_AND_ASSIGN(PDFBrowserTest);
  69 };
  70
  71 #endif  // CHROME_BROWSER_UI_PDF_PDF_BROWSERTEST_BASE_H_