| OLD | NEW |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 The Chromium 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 "base/files/memory_mapped_file.h" | 5 #include "base/files/memory_mapped_file.h" |
| 6 | 6 |
| 7 #include "base/files/file_path.h" | 7 #include "base/files/file_path.h" |
| 8 #include "base/files/file_util.h" | 8 #include "base/files/file_util.h" |
| 9 #include "testing/gtest/include/gtest/gtest.h" | 9 #include "testing/gtest/include/gtest/gtest.h" |
| 10 #include "testing/platform_test.h" | 10 #include "testing/platform_test.h" |
| 11 | 11 |
| 12 namespace base { | 12 namespace base { |
| 13 | 13 |
| 14 namespace { | 14 namespace { |
| 15 | 15 |
| 16 // Create a temporary buffer and fill it with a watermark sequence. | 16 // Create a temporary buffer and fill it with a watermark sequence. |
| 17 scoped_ptr<uint8[]> CreateTestBuffer(size_t size, size_t offset) { | 17 scoped_ptr<uint8_t[]> CreateTestBuffer(size_t size, size_t offset) { |
| 18 scoped_ptr<uint8[]> buf(new uint8[size]); | 18 scoped_ptr<uint8_t[]> buf(new uint8_t[size]); |
| 19 for (size_t i = 0; i < size; ++i) | 19 for (size_t i = 0; i < size; ++i) |
| 20 buf.get()[i] = static_cast<uint8>((offset + i) % 253); | 20 buf.get()[i] = static_cast<uint8_t>((offset + i) % 253); |
| 21 return buf; | 21 return buf; |
| 22 } | 22 } |
| 23 | 23 |
| 24 // Check that the watermark sequence is consistent with the |offset| provided. | 24 // Check that the watermark sequence is consistent with the |offset| provided. |
| 25 bool CheckBufferContents(const uint8* data, size_t size, size_t offset) { | 25 bool CheckBufferContents(const uint8_t* data, size_t size, size_t offset) { |
| 26 scoped_ptr<uint8[]> test_data(CreateTestBuffer(size, offset)); | 26 scoped_ptr<uint8_t[]> test_data(CreateTestBuffer(size, offset)); |
| 27 return memcmp(test_data.get(), data, size) == 0; | 27 return memcmp(test_data.get(), data, size) == 0; |
| 28 } | 28 } |
| 29 | 29 |
| 30 class MemoryMappedFileTest : public PlatformTest { | 30 class MemoryMappedFileTest : public PlatformTest { |
| 31 protected: | 31 protected: |
| 32 void SetUp() override { | 32 void SetUp() override { |
| 33 PlatformTest::SetUp(); | 33 PlatformTest::SetUp(); |
| 34 CreateTemporaryFile(&temp_file_path_); | 34 CreateTemporaryFile(&temp_file_path_); |
| 35 } | 35 } |
| 36 | 36 |
| 37 void TearDown() override { EXPECT_TRUE(DeleteFile(temp_file_path_, false)); } | 37 void TearDown() override { EXPECT_TRUE(DeleteFile(temp_file_path_, false)); } |
| 38 | 38 |
| 39 void CreateTemporaryTestFile(size_t size) { | 39 void CreateTemporaryTestFile(size_t size) { |
| 40 File file(temp_file_path_, | 40 File file(temp_file_path_, |
| 41 File::FLAG_CREATE_ALWAYS | File::FLAG_READ | File::FLAG_WRITE); | 41 File::FLAG_CREATE_ALWAYS | File::FLAG_READ | File::FLAG_WRITE); |
| 42 EXPECT_TRUE(file.IsValid()); | 42 EXPECT_TRUE(file.IsValid()); |
| 43 | 43 |
| 44 scoped_ptr<uint8[]> test_data(CreateTestBuffer(size, 0)); | 44 scoped_ptr<uint8_t[]> test_data(CreateTestBuffer(size, 0)); |
| 45 size_t bytes_written = | 45 size_t bytes_written = |
| 46 file.Write(0, reinterpret_cast<char*>(test_data.get()), size); | 46 file.Write(0, reinterpret_cast<char*>(test_data.get()), size); |
| 47 EXPECT_EQ(size, bytes_written); | 47 EXPECT_EQ(size, bytes_written); |
| 48 file.Close(); | 48 file.Close(); |
| 49 } | 49 } |
| 50 | 50 |
| 51 const FilePath temp_file_path() const { return temp_file_path_; } | 51 const FilePath temp_file_path() const { return temp_file_path_; } |
| 52 | 52 |
| 53 private: | 53 private: |
| 54 FilePath temp_file_path_; | 54 FilePath temp_file_path_; |
| (...skipping 106 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 161 map.Initialize(file.Pass(), region); | 161 map.Initialize(file.Pass(), region); |
| 162 ASSERT_EQ(kPartialSize, map.length()); | 162 ASSERT_EQ(kPartialSize, map.length()); |
| 163 ASSERT_TRUE(map.data() != NULL); | 163 ASSERT_TRUE(map.data() != NULL); |
| 164 EXPECT_TRUE(map.IsValid()); | 164 EXPECT_TRUE(map.IsValid()); |
| 165 ASSERT_TRUE(CheckBufferContents(map.data(), kPartialSize, kOffset)); | 165 ASSERT_TRUE(CheckBufferContents(map.data(), kPartialSize, kOffset)); |
| 166 } | 166 } |
| 167 | 167 |
| 168 } // namespace | 168 } // namespace |
| 169 | 169 |
| 170 } // namespace base | 170 } // namespace base |
| OLD | NEW |