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. pdf_test_server()32 net::test_server::EmbeddedTestServer* pdf_test_server() { 33 return &pdf_test_server_; 34 } 35 load_stop_notification_count()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_ 72