| OLD | NEW |
| 1 // Copyright 2014 the V8 project authors. All rights reserved. | 1 // Copyright 2014 the V8 project authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "include/libplatform/libplatform.h" | 5 #include "include/libplatform/libplatform.h" |
| 6 #include "include/v8.h" | 6 #include "include/v8.h" |
| 7 #include "src/base/compiler-specific.h" | 7 #include "src/base/compiler-specific.h" |
| 8 #include "testing/gmock/include/gmock/gmock.h" | 8 #include "testing/gmock/include/gmock/gmock.h" |
| 9 | 9 |
| 10 #ifdef V8_USE_EXTERNAL_STARTUP_DATA | 10 #ifdef V8_USE_EXTERNAL_STARTUP_DATA |
| 11 #include "src/startup-data-util.h" | 11 #include "src/startup-data-util.h" |
| 12 #endif // V8_USE_EXTERNAL_STARTUP_DATA | 12 #endif // V8_USE_EXTERNAL_STARTUP_DATA |
| 13 | 13 |
| 14 namespace { | 14 namespace { |
| 15 | 15 |
| 16 class ArrayBufferAllocator : public v8::ArrayBuffer::Allocator { |
| 17 public: |
| 18 virtual void* Allocate(size_t length) { |
| 19 void* data = AllocateUninitialized(length); |
| 20 return data == NULL ? data : memset(data, 0, length); |
| 21 } |
| 22 virtual void* AllocateUninitialized(size_t length) { return malloc(length); } |
| 23 virtual void Free(void* data, size_t) { free(data); } |
| 24 }; |
| 25 |
| 16 class DefaultPlatformEnvironment final : public ::testing::Environment { | 26 class DefaultPlatformEnvironment final : public ::testing::Environment { |
| 17 public: | 27 public: |
| 18 DefaultPlatformEnvironment() : platform_(NULL) {} | 28 DefaultPlatformEnvironment() : platform_(NULL) {} |
| 19 ~DefaultPlatformEnvironment() {} | 29 ~DefaultPlatformEnvironment() {} |
| 20 | 30 |
| 21 virtual void SetUp() override { | 31 virtual void SetUp() override { |
| 22 EXPECT_EQ(NULL, platform_); | 32 EXPECT_EQ(NULL, platform_); |
| 23 platform_ = v8::platform::CreateDefaultPlatform(); | 33 platform_ = v8::platform::CreateDefaultPlatform(); |
| 24 ASSERT_TRUE(platform_ != NULL); | 34 ASSERT_TRUE(platform_ != NULL); |
| 25 v8::V8::InitializePlatform(platform_); | 35 v8::V8::InitializePlatform(platform_); |
| 36 v8::V8::SetArrayBufferAllocator(&array_buffer_allocator_); |
| 26 ASSERT_TRUE(v8::V8::Initialize()); | 37 ASSERT_TRUE(v8::V8::Initialize()); |
| 27 } | 38 } |
| 28 | 39 |
| 29 virtual void TearDown() override { | 40 virtual void TearDown() override { |
| 30 ASSERT_TRUE(platform_ != NULL); | 41 ASSERT_TRUE(platform_ != NULL); |
| 31 v8::V8::Dispose(); | 42 v8::V8::Dispose(); |
| 32 v8::V8::ShutdownPlatform(); | 43 v8::V8::ShutdownPlatform(); |
| 33 delete platform_; | 44 delete platform_; |
| 34 platform_ = NULL; | 45 platform_ = NULL; |
| 35 } | 46 } |
| 36 | 47 |
| 37 private: | 48 private: |
| 38 v8::Platform* platform_; | 49 v8::Platform* platform_; |
| 50 ArrayBufferAllocator array_buffer_allocator_; |
| 39 }; | 51 }; |
| 40 | 52 |
| 41 } // namespace | 53 } // namespace |
| 42 | 54 |
| 43 | 55 |
| 44 int main(int argc, char** argv) { | 56 int main(int argc, char** argv) { |
| 45 testing::InitGoogleMock(&argc, argv); | 57 testing::InitGoogleMock(&argc, argv); |
| 46 testing::AddGlobalTestEnvironment(new DefaultPlatformEnvironment); | 58 testing::AddGlobalTestEnvironment(new DefaultPlatformEnvironment); |
| 47 v8::V8::SetFlagsFromCommandLine(&argc, argv, true); | 59 v8::V8::SetFlagsFromCommandLine(&argc, argv, true); |
| 48 #ifdef V8_USE_EXTERNAL_STARTUP_DATA | 60 #ifdef V8_USE_EXTERNAL_STARTUP_DATA |
| 49 v8::StartupDataHandler startup_data(argv[0], NULL, NULL); | 61 v8::StartupDataHandler startup_data(argv[0], NULL, NULL); |
| 50 #endif | 62 #endif |
| 51 return RUN_ALL_TESTS(); | 63 return RUN_ALL_TESTS(); |
| 52 } | 64 } |
| OLD | NEW |