Index: pdf/chunk_stream_unittest.cc |
diff --git a/pdf/chunk_stream_unittest.cc b/pdf/chunk_stream_unittest.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..418291180678a9b0335a52c3965a2eada63a45ed |
--- /dev/null |
+++ b/pdf/chunk_stream_unittest.cc |
@@ -0,0 +1,98 @@ |
+// Copyright (c) 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 "testing/gtest/include/gtest/gtest.h" |
+ |
+namespace chrome_pdf { |
+namespace { |
+typedef ChunkStream<10> TestChunkStream; |
+std::unique_ptr<TestChunkStream::ChunkData> CreateChunkData() { |
+ return std::unique_ptr<TestChunkStream::ChunkData>( |
Lei Zhang
2016/10/21 09:33:09
nit: base::MakeUnique ?
snake
2016/10/21 15:13:14
Done.
|
+ new TestChunkStream::ChunkData()); |
+} |
+} |
+ |
+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); |
+ const 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 |