forked from chromium/chromium
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathchunk_stream_unittest.cc
99 lines (88 loc) · 3.21 KB
/
chunk_stream_unittest.cc
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
// Copyright 2016 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "pdf/chunk_stream.h"
#include <array>
#include <memory>
#include <utility>
#include "testing/gtest/include/gtest/gtest.h"
namespace chrome_pdf {
namespace {
typedef ChunkStream<10> TestChunkStream;
std::unique_ptr<TestChunkStream::ChunkData> CreateChunkData() {
return std::make_unique<TestChunkStream::ChunkData>();
}
} // namespace
TEST(ChunkStreamTest, InRow) {
TestChunkStream stream;
EXPECT_FALSE(stream.IsComplete());
EXPECT_FALSE(stream.IsRangeAvailable(gfx::Range(0, 10)));
stream.SetChunkData(0, CreateChunkData());
EXPECT_TRUE(stream.IsRangeAvailable(gfx::Range(0, 10)));
EXPECT_FALSE(stream.IsRangeAvailable(gfx::Range(0, 20)));
stream.SetChunkData(1, CreateChunkData());
EXPECT_TRUE(stream.IsRangeAvailable(gfx::Range(0, 20)));
EXPECT_FALSE(stream.IsRangeAvailable(gfx::Range(0, 30)));
stream.SetChunkData(2, CreateChunkData());
EXPECT_TRUE(stream.IsRangeAvailable(gfx::Range(0, 30)));
stream.set_eof_pos(25);
EXPECT_FALSE(stream.IsRangeAvailable(gfx::Range(0, 30)));
EXPECT_TRUE(stream.IsRangeAvailable(gfx::Range(0, 25)));
EXPECT_TRUE(stream.IsComplete());
}
TEST(ChunkStreamTest, InBackRow) {
TestChunkStream stream;
stream.set_eof_pos(25);
EXPECT_FALSE(stream.IsComplete());
EXPECT_FALSE(stream.IsRangeAvailable(gfx::Range(20, 25)));
stream.SetChunkData(2, CreateChunkData());
EXPECT_TRUE(stream.IsRangeAvailable(gfx::Range(20, 25)));
EXPECT_FALSE(stream.IsRangeAvailable(gfx::Range(10, 20)));
stream.SetChunkData(1, CreateChunkData());
EXPECT_TRUE(stream.IsRangeAvailable(gfx::Range(10, 20)));
EXPECT_FALSE(stream.IsRangeAvailable(gfx::Range(0, 10)));
stream.SetChunkData(0, CreateChunkData());
EXPECT_TRUE(stream.IsRangeAvailable(gfx::Range(0, 10)));
EXPECT_TRUE(stream.IsComplete());
}
TEST(ChunkStreamTest, FillGap) {
TestChunkStream stream;
stream.set_eof_pos(25);
EXPECT_FALSE(stream.IsComplete());
stream.SetChunkData(0, CreateChunkData());
stream.SetChunkData(2, CreateChunkData());
EXPECT_TRUE(stream.IsRangeAvailable(gfx::Range(0, 10)));
EXPECT_TRUE(stream.IsRangeAvailable(gfx::Range(20, 25)));
EXPECT_FALSE(stream.IsRangeAvailable(gfx::Range(0, 25)));
stream.SetChunkData(1, CreateChunkData());
EXPECT_TRUE(stream.IsRangeAvailable(gfx::Range(0, 25)));
EXPECT_TRUE(stream.IsComplete());
}
TEST(ChunkStreamTest, Read) {
TestChunkStream stream;
stream.set_eof_pos(25);
constexpr unsigned char start_value = 33;
unsigned char value = start_value;
auto chunk_0 = CreateChunkData();
for (auto& it : *chunk_0) {
it = ++value;
}
auto chunk_1 = CreateChunkData();
for (auto& it : *chunk_1) {
it = ++value;
}
auto chunk_2 = CreateChunkData();
for (auto& it : *chunk_2) {
it = ++value;
}
stream.SetChunkData(0, std::move(chunk_0));
stream.SetChunkData(2, std::move(chunk_2));
stream.SetChunkData(1, std::move(chunk_1));
std::array<unsigned char, 25> result_data;
EXPECT_TRUE(stream.ReadData(gfx::Range(0, 25), result_data.data()));
value = start_value;
for (const auto& it : result_data) {
EXPECT_EQ(++value, it);
}
}
} // namespace chrome_pdf