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 |