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" | |
11 #include "testing/gtest/include/gtest/gtest.h" | 7 #include "testing/gtest/include/gtest/gtest.h" |
12 | 8 |
13 namespace chrome_pdf { | 9 namespace chrome_pdf { |
14 namespace { | |
15 typedef ChunkStream<10> TestChunkStream; | |
16 | 10 |
17 std::unique_ptr<TestChunkStream::ChunkData> CreateChunkData() { | 11 TEST(ChunkStreamTest, Simple) { |
18 return base::MakeUnique<TestChunkStream::ChunkData>(); | 12 ChunkStream stream; |
19 } | 13 stream.Preallocate(1000); |
| 14 EXPECT_FALSE(stream.IsRangeAvailable(100, 500)); |
20 } | 15 } |
21 | 16 |
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 } | |
99 } // namespace chrome_pdf | 17 } // namespace chrome_pdf |
OLD | NEW |