Fix leaks in the embedder tests themselves.
[pdfium.git] / testing / embedder_test.h
1 // Copyright (c) 2015 PDFium 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 TESTING_EMBEDDER_TEST_H_
6 #define TESTING_EMBEDDER_TEST_H_
7
8 #include <string>
9
10 #include "../core/include/fxcrt/fx_system.h"
11 #include "../public/fpdf_dataavail.h"
12 #include "../public/fpdf_ext.h"
13 #include "../public/fpdf_formfill.h"
14 #include "../public/fpdfview.h"
15 #include "testing/gtest/include/gtest/gtest.h"
16 #include "v8/include/v8.h"
17
18 class TestLoader;
19
20 // This class is used to load a PDF document, and then run programatic
21 // API tests against it.
22 class EmbedderTest : public ::testing::Test,
23                      public UNSUPPORT_INFO,
24                      public IPDF_JSPLATFORM,
25                      public FPDF_FORMFILLINFO {
26  public:
27   class Delegate {
28    public:
29     virtual ~Delegate() { }
30
31     // Equivalent to UNSUPPORT_INFO::FSDK_UnSupport_Handler().
32     virtual void UnsupportedHandler(int type) { }
33
34     // Equivalent to IPDF_JSPLATFORM::app_alert().
35     virtual int Alert(FPDF_WIDESTRING message, FPDF_WIDESTRING title,
36                       int type, int icon) {
37       return 0;
38     }
39   };
40
41   EmbedderTest();
42   virtual ~EmbedderTest();
43
44   void SetUp() override;
45   void TearDown() override;
46
47   void SetDelegate(Delegate* delegate) {
48     delegate_ = delegate ? delegate : default_delegate_;
49   }
50
51   FPDF_DOCUMENT document() { return document_; }
52   FPDF_FORMHANDLE form_handle() { return form_handle_; }
53
54   // Open the document specified by |filename|, and create its form fill
55   // environment, or return false on failure.
56   virtual bool OpenDocument(const std::string& filename);
57
58   // Perform JavaScript actions that are to run at document open time.
59   virtual void DoOpenActions();
60
61   // Determine the page numbers present in the document.
62   virtual int GetFirstPageNum();
63   virtual int GetPageCount();
64
65   // Load a specific page of the open document.
66   virtual FPDF_PAGE LoadPage(int page_number);
67
68   // Convert a loaded page into a bitmap.
69   virtual FPDF_BITMAP RenderPage(FPDF_PAGE page);
70
71   // Relese the resources obtained from LoadPage(). Further use of |page|
72   // is prohibited after this call is made.
73   virtual void UnloadPage(FPDF_PAGE page);
74
75  protected:
76   Delegate* delegate_;
77   Delegate* default_delegate_;
78   FPDF_DOCUMENT document_;
79   FPDF_FORMHANDLE form_handle_;
80   FPDF_AVAIL avail_;
81   FX_DOWNLOADHINTS hints_;
82   FPDF_FILEACCESS file_access_;
83   FX_FILEAVAIL file_avail_;
84   v8::StartupData natives_;
85   v8::StartupData snapshot_;
86   TestLoader* loader_;
87   size_t file_length_;
88   char* file_contents_;
89
90  private:
91   static void UnsupportedHandlerTrampoline(UNSUPPORT_INFO*, int type);
92   static int AlertTrampoline(IPDF_JSPLATFORM* plaform, FPDF_WIDESTRING message,
93                              FPDF_WIDESTRING title, int type, int icon);
94 };
95
96 #endif  // TESTING_EMBEDDER_TEST_H_
97