1// Copyright 2018 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#include <string>
6
7#include "core/fxcrt/fx_memory.h"
8#include "testing/gmock/include/gmock/gmock.h"
9#include "testing/gtest/include/gtest/gtest.h"
10#include "testing/test_support.h"
11
12#ifdef PDF_ENABLE_V8
13#include "v8/include/v8-platform.h"
14#include "v8/include/v8.h"
15#endif  // PDF_ENABLE_v8
16
17namespace {
18
19const char* g_exe_path = nullptr;
20
21#ifdef PDF_ENABLE_V8
22#ifdef V8_USE_EXTERNAL_STARTUP_DATA
23v8::StartupData* g_v8_natives = nullptr;
24v8::StartupData* g_v8_snapshot = nullptr;
25#endif  // V8_USE_EXTERNAL_STARTUP_DATA
26#endif  // PDF_ENABLE_V8
27
28// The loading time of the CFGAS_FontMgr is linear in the number of times it is
29// loaded. So, if a test suite has a lot of tests that need a font manager they
30// can end up executing very, very slowly.
31class Environment : public testing::Environment {
32 public:
33  void SetUp() override {
34#ifdef PDF_ENABLE_V8
35#ifdef V8_USE_EXTERNAL_STARTUP_DATA
36    if (g_v8_natives && g_v8_snapshot) {
37      InitializeV8ForPDFium(g_exe_path, std::string(), nullptr, nullptr,
38                            &platform_);
39    } else {
40      g_v8_natives = new v8::StartupData;
41      g_v8_snapshot = new v8::StartupData;
42      InitializeV8ForPDFium(g_exe_path, std::string(), g_v8_natives,
43                            g_v8_snapshot, &platform_);
44    }
45#else
46    InitializeV8ForPDFium(g_exe_path, &platform_);
47#endif  // V8_USE_EXTERNAL_STARTUP_DATA
48#endif  // FPDF_ENABLE_V8
49  }
50
51  void TearDown() override {
52#ifdef PDF_ENABLE_V8
53    v8::V8::ShutdownPlatform();
54    delete platform_;
55#endif  // PDF_ENABLE_V8
56  }
57
58 private:
59#ifdef PDF_ENABLE_V8
60  v8::Platform* platform_;
61#endif  // PDF_ENABLE_V8
62};
63
64Environment* env_ = nullptr;
65
66}  // namespace
67
68// Can't use gtest-provided main since we need to stash the path to the
69// executable in order to find the external V8 binary data files.
70int main(int argc, char** argv) {
71  g_exe_path = argv[0];
72
73  FXMEM_InitializePartitionAlloc();
74
75  env_ = new Environment();
76  // The env will be deleted by gtest.
77  AddGlobalTestEnvironment(env_);
78
79  testing::InitGoogleTest(&argc, argv);
80  testing::InitGoogleMock(&argc, argv);
81
82  int ret_val = RUN_ALL_TESTS();
83
84#ifdef PDF_ENABLE_V8
85#ifdef V8_USE_EXTERNAL_STARTUP_DATA
86  if (g_v8_natives)
87    free(const_cast<char*>(g_v8_natives->data));
88  if (g_v8_snapshot)
89    free(const_cast<char*>(g_v8_snapshot->data));
90#endif  // V8_USE_EXTERNAL_STARTUP_DATA
91#endif  // PDF_ENABLE_V8
92
93  return ret_val;
94}
95