| // Copyright 2018 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. |
| |
| #include <memory> |
| #include <string> |
| |
| #include "core/fxcrt/fx_memory.h" |
| #include "testing/gmock/include/gmock/gmock.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| #include "testing/test_support.h" |
| |
| #ifdef PDF_ENABLE_V8 |
| #include "v8/include/v8-platform.h" |
| #include "v8/include/v8.h" |
| #endif // PDF_ENABLE_v8 |
| |
| namespace { |
| |
| const char* g_exe_path = nullptr; |
| |
| #ifdef PDF_ENABLE_V8 |
| #ifdef V8_USE_EXTERNAL_STARTUP_DATA |
| v8::StartupData* g_v8_natives = nullptr; |
| v8::StartupData* g_v8_snapshot = nullptr; |
| #endif // V8_USE_EXTERNAL_STARTUP_DATA |
| #endif // PDF_ENABLE_V8 |
| |
| // The loading time of the CFGAS_FontMgr is linear in the number of times it is |
| // loaded. So, if a test suite has a lot of tests that need a font manager they |
| // can end up executing very, very slowly. |
| class Environment final : public testing::Environment { |
| public: |
| void SetUp() override { |
| #ifdef PDF_ENABLE_V8 |
| #ifdef V8_USE_EXTERNAL_STARTUP_DATA |
| if (g_v8_natives && g_v8_snapshot) { |
| platform_ = InitializeV8ForPDFiumWithStartupData( |
| g_exe_path, std::string(), nullptr, nullptr); |
| } else { |
| g_v8_natives = new v8::StartupData; |
| g_v8_snapshot = new v8::StartupData; |
| platform_ = InitializeV8ForPDFiumWithStartupData( |
| g_exe_path, std::string(), g_v8_natives, g_v8_snapshot); |
| } |
| #else |
| platform_ = InitializeV8ForPDFium(g_exe_path); |
| #endif // V8_USE_EXTERNAL_STARTUP_DATA |
| #endif // FPDF_ENABLE_V8 |
| } |
| |
| void TearDown() override { |
| #ifdef PDF_ENABLE_V8 |
| v8::V8::ShutdownPlatform(); |
| #endif // PDF_ENABLE_V8 |
| } |
| |
| private: |
| #ifdef PDF_ENABLE_V8 |
| std::unique_ptr<v8::Platform> platform_; |
| #endif // PDF_ENABLE_V8 |
| }; |
| |
| Environment* env_ = nullptr; |
| |
| } // namespace |
| |
| // Can't use gtest-provided main since we need to stash the path to the |
| // executable in order to find the external V8 binary data files. |
| int main(int argc, char** argv) { |
| g_exe_path = argv[0]; |
| |
| FXMEM_InitializePartitionAlloc(); |
| |
| env_ = new Environment(); |
| // The env will be deleted by gtest. |
| AddGlobalTestEnvironment(env_); |
| |
| testing::InitGoogleTest(&argc, argv); |
| testing::InitGoogleMock(&argc, argv); |
| |
| int ret_val = RUN_ALL_TESTS(); |
| |
| #ifdef PDF_ENABLE_V8 |
| #ifdef V8_USE_EXTERNAL_STARTUP_DATA |
| if (g_v8_natives) |
| free(const_cast<char*>(g_v8_natives->data)); |
| if (g_v8_snapshot) |
| free(const_cast<char*>(g_v8_snapshot->data)); |
| #endif // V8_USE_EXTERNAL_STARTUP_DATA |
| #endif // PDF_ENABLE_V8 |
| |
| return ret_val; |
| } |