OLD | NEW |
1 // Copyright 2016 The Chromium Authors. All rights reserved. | 1 // Copyright 2016 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 "pdf/chunk_stream.h" | 5 #include "pdf/chunk_stream.h" |
6 | 6 |
| 7 #include <array> |
| 8 #include <memory> |
| 9 |
| 10 #include "base/memory/ptr_util.h" |
7 #include "testing/gtest/include/gtest/gtest.h" | 11 #include "testing/gtest/include/gtest/gtest.h" |
8 | 12 |
9 namespace chrome_pdf { | 13 namespace chrome_pdf { |
| 14 namespace { |
| 15 typedef ChunkStream<10> TestChunkStream; |
10 | 16 |
11 TEST(ChunkStreamTest, Simple) { | 17 std::unique_ptr<TestChunkStream::ChunkData> CreateChunkData() { |
12 ChunkStream stream; | 18 return base::MakeUnique<TestChunkStream::ChunkData>(); |
13 stream.Preallocate(1000); | 19 } |
14 EXPECT_FALSE(stream.IsRangeAvailable(100, 500)); | |
15 } | 20 } |
16 | 21 |
| 22 TEST(ChunkStreamTest, InRow) { |
| 23 TestChunkStream stream; |
| 24 EXPECT_FALSE(stream.IsComplete()); |
| 25 EXPECT_FALSE(stream.IsRangeAvailable(gfx::Range(0, 10))); |
| 26 stream.SetChunkData(0, CreateChunkData()); |
| 27 EXPECT_TRUE(stream.IsRangeAvailable(gfx::Range(0, 10))); |
| 28 EXPECT_FALSE(stream.IsRangeAvailable(gfx::Range(0, 20))); |
| 29 stream.SetChunkData(1, CreateChunkData()); |
| 30 EXPECT_TRUE(stream.IsRangeAvailable(gfx::Range(0, 20))); |
| 31 EXPECT_FALSE(stream.IsRangeAvailable(gfx::Range(0, 30))); |
| 32 stream.SetChunkData(2, CreateChunkData()); |
| 33 EXPECT_TRUE(stream.IsRangeAvailable(gfx::Range(0, 30))); |
| 34 stream.set_eof_pos(25); |
| 35 EXPECT_FALSE(stream.IsRangeAvailable(gfx::Range(0, 30))); |
| 36 EXPECT_TRUE(stream.IsRangeAvailable(gfx::Range(0, 25))); |
| 37 EXPECT_TRUE(stream.IsComplete()); |
| 38 } |
| 39 |
| 40 TEST(ChunkStreamTest, InBackRow) { |
| 41 TestChunkStream stream; |
| 42 stream.set_eof_pos(25); |
| 43 EXPECT_FALSE(stream.IsComplete()); |
| 44 EXPECT_FALSE(stream.IsRangeAvailable(gfx::Range(20, 25))); |
| 45 stream.SetChunkData(2, CreateChunkData()); |
| 46 EXPECT_TRUE(stream.IsRangeAvailable(gfx::Range(20, 25))); |
| 47 EXPECT_FALSE(stream.IsRangeAvailable(gfx::Range(10, 20))); |
| 48 stream.SetChunkData(1, CreateChunkData()); |
| 49 EXPECT_TRUE(stream.IsRangeAvailable(gfx::Range(10, 20))); |
| 50 EXPECT_FALSE(stream.IsRangeAvailable(gfx::Range(0, 10))); |
| 51 stream.SetChunkData(0, CreateChunkData()); |
| 52 EXPECT_TRUE(stream.IsRangeAvailable(gfx::Range(0, 10))); |
| 53 EXPECT_TRUE(stream.IsComplete()); |
| 54 } |
| 55 |
| 56 TEST(ChunkStreamTest, FillGap) { |
| 57 TestChunkStream stream; |
| 58 stream.set_eof_pos(25); |
| 59 EXPECT_FALSE(stream.IsComplete()); |
| 60 stream.SetChunkData(0, CreateChunkData()); |
| 61 stream.SetChunkData(2, CreateChunkData()); |
| 62 EXPECT_TRUE(stream.IsRangeAvailable(gfx::Range(0, 10))); |
| 63 EXPECT_TRUE(stream.IsRangeAvailable(gfx::Range(20, 25))); |
| 64 EXPECT_FALSE(stream.IsRangeAvailable(gfx::Range(0, 25))); |
| 65 stream.SetChunkData(1, CreateChunkData()); |
| 66 EXPECT_TRUE(stream.IsRangeAvailable(gfx::Range(0, 25))); |
| 67 EXPECT_TRUE(stream.IsComplete()); |
| 68 } |
| 69 |
| 70 TEST(ChunkStreamTest, Read) { |
| 71 TestChunkStream stream; |
| 72 stream.set_eof_pos(25); |
| 73 const unsigned char start_value = 33; |
| 74 unsigned char value = start_value; |
| 75 auto chunk_0 = CreateChunkData(); |
| 76 for (auto& it : *chunk_0) { |
| 77 it = ++value; |
| 78 } |
| 79 auto chunk_1 = CreateChunkData(); |
| 80 for (auto& it : *chunk_1) { |
| 81 it = ++value; |
| 82 } |
| 83 auto chunk_2 = CreateChunkData(); |
| 84 for (auto& it : *chunk_2) { |
| 85 it = ++value; |
| 86 } |
| 87 stream.SetChunkData(0, std::move(chunk_0)); |
| 88 stream.SetChunkData(2, std::move(chunk_2)); |
| 89 stream.SetChunkData(1, std::move(chunk_1)); |
| 90 |
| 91 std::array<unsigned char, 25> result_data; |
| 92 EXPECT_TRUE(stream.ReadData(gfx::Range(0, 25), result_data.data())); |
| 93 |
| 94 value = start_value; |
| 95 for (const auto& it : result_data) { |
| 96 EXPECT_EQ(++value, it); |
| 97 } |
| 98 } |
17 } // namespace chrome_pdf | 99 } // namespace chrome_pdf |
OLD | NEW |