// Copyright 2015 PDFium Authors. All rights reserved. // Use of this source code is governed by a BSD-style license that can be // found in the LICENSE file. #ifndef TESTING_EMBEDDER_TEST_H_ #define TESTING_EMBEDDER_TEST_H_ #include <map> #include <memory> #include <string> #include "public/fpdf_dataavail.h" #include "public/fpdf_ext.h" #include "public/fpdf_formfill.h" #include "public/fpdfview.h" #include "testing/gtest/include/gtest/gtest.h" #include "testing/test_support.h" #ifdef PDF_ENABLE_V8 #include "v8/include/v8.h" #endif // PDF_ENABLE_v8 class TestLoader; // This class is used to load a PDF document, and then run programatic // API tests against it. class EmbedderTest : public ::testing::Test, public UNSUPPORT_INFO, public IPDF_JSPLATFORM, public FPDF_FORMFILLINFO { public: class Delegate { public: virtual ~Delegate() {} // Equivalent to UNSUPPORT_INFO::FSDK_UnSupport_Handler(). virtual void UnsupportedHandler(int type) {} // Equivalent to IPDF_JSPLATFORM::app_alert(). virtual int Alert(FPDF_WIDESTRING message, FPDF_WIDESTRING title, int type, int icon) { return 0; } // Equivalent to FPDF_FORMFILLINFO::FFI_SetTimer(). virtual int SetTimer(int msecs, TimerCallback fn) { return 0; } // Equivalent to FPDF_FORMFILLINFO::FFI_KillTimer(). virtual void KillTimer(int id) {} // Equivalent to FPDF_FORMFILLINFO::FFI_GetPage(). virtual FPDF_PAGE GetPage(FPDF_FORMFILLINFO* info, FPDF_DOCUMENT document, int page_index); }; EmbedderTest(); virtual ~EmbedderTest(); void SetUp() override; void TearDown() override; #ifdef PDF_ENABLE_V8 // Call before SetUp to pass shared isolate, otherwise PDFium creates one. void SetExternalIsolate(void* isolate) { external_isolate_ = static_cast<v8::Isolate*>(isolate); } #endif // PDF_ENABLE_V8 void SetDelegate(Delegate* delegate) { delegate_ = delegate ? delegate : default_delegate_.get(); } FPDF_DOCUMENT document() { return document_; } FPDF_FORMHANDLE form_handle() { return form_handle_; } // Create an empty document, and its form fill environment. Returns true // on success or false on failure. virtual bool CreateEmptyDocument(); // Open the document specified by |filename|, and create its form fill // environment, or return false on failure. // The filename is relative to the test data directory where we store all the // test files. virtual bool OpenDocument(const std::string& filename, const char* password = nullptr, bool must_linearize = false); // Perform JavaScript actions that are to run at document open time. virtual void DoOpenActions(); // Determine the page numbers present in the document. virtual int GetFirstPageNum(); virtual int GetPageCount(); // Load a specific page of the open document. virtual FPDF_PAGE LoadPage(int page_number); // Convert a loaded page into a bitmap. virtual FPDF_BITMAP RenderPage(FPDF_PAGE page); // Relese the resources obtained from LoadPage(). Further use of |page| // is prohibited after this call is made. virtual void UnloadPage(FPDF_PAGE page); protected: void SetupFormFillEnvironment(); // Return the hash of |bitmap|. static std::string HashBitmap(FPDF_BITMAP bitmap, int expected_width, int expected_height); // Check |bitmap| to make sure it has the right dimensions and content. static void CompareBitmap(FPDF_BITMAP bitmap, int expected_width, int expected_height, const char* expected_md5sum); Delegate* delegate_; std::unique_ptr<Delegate> default_delegate_; FPDF_DOCUMENT document_; FPDF_FORMHANDLE form_handle_; FPDF_AVAIL avail_; FX_DOWNLOADHINTS hints_; FPDF_FILEACCESS file_access_; FX_FILEAVAIL file_avail_; #ifdef PDF_ENABLE_V8 v8::Platform* platform_; #endif // PDF_ENABLE_V8 void* external_isolate_; TestLoader* loader_; size_t file_length_; std::unique_ptr<char, pdfium::FreeDeleter> file_contents_; std::map<int, FPDF_PAGE> page_map_; std::map<FPDF_PAGE, int> page_reverse_map_; private: static void UnsupportedHandlerTrampoline(UNSUPPORT_INFO*, int type); static int AlertTrampoline(IPDF_JSPLATFORM* plaform, FPDF_WIDESTRING message, FPDF_WIDESTRING title, int type, int icon); static int SetTimerTrampoline(FPDF_FORMFILLINFO* info, int msecs, TimerCallback fn); static void KillTimerTrampoline(FPDF_FORMFILLINFO* info, int id); static FPDF_PAGE GetPageTrampoline(FPDF_FORMFILLINFO* info, FPDF_DOCUMENT document, int page_index); }; #endif // TESTING_EMBEDDER_TEST_H_