Index: blimp/net/compressed_packet_unittest.cc |
diff --git a/blimp/net/compressed_packet_unittest.cc b/blimp/net/compressed_packet_unittest.cc |
deleted file mode 100644 |
index 0742fda8d74659fb6abb0ee72bb9efe1d4fc54e0..0000000000000000000000000000000000000000 |
--- a/blimp/net/compressed_packet_unittest.cc |
+++ /dev/null |
@@ -1,227 +0,0 @@ |
-// 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 "base/memory/ptr_util.h" |
-#include "base/rand_util.h" |
-#include "base/sys_byteorder.h" |
-#include "blimp/net/common.h" |
-#include "blimp/net/compressed_packet_reader.h" |
-#include "blimp/net/compressed_packet_writer.h" |
-#include "blimp/net/test_common.h" |
-#include "net/base/test_completion_callback.h" |
-#include "testing/gmock/include/gmock/gmock.h" |
-#include "testing/gtest/include/gtest/gtest.h" |
- |
-using testing::_; |
-using testing::DoAll; |
-using testing::InvokeArgument; |
-using testing::SaveArg; |
- |
-namespace blimp { |
-namespace { |
- |
-// Calls the CompletionCallback (indicated by the 0-based index cb_idx) |
-// with the value |result|. |
-ACTION_TEMPLATE(InvokeCompletionCallback, |
- HAS_1_TEMPLATE_PARAMS(int, cb_idx), |
- AND_1_VALUE_PARAMS(result)) { |
- testing::get<cb_idx>(args).Run(result); |
-} |
- |
-// Copies a DrainableIOBuffer to a GrowableIOBuffer. |
-// |dest_buf_idx|: The 0-based index of a GrowableIOBuffer. |
-// |src_buf|: A DrainableIOBuffer. |
-ACTION_TEMPLATE(CopyBuffer, |
- HAS_1_TEMPLATE_PARAMS(int, dest_buf_idx), |
- AND_1_VALUE_PARAMS(src_buf)) { |
- const auto& dest_buf = testing::get<dest_buf_idx>(args); |
- ASSERT_TRUE(dest_buf); |
- ASSERT_TRUE(src_buf); |
- ASSERT_EQ(0, dest_buf->offset()); |
- src_buf->SetOffset(0); |
- dest_buf->SetCapacity(src_buf->BytesRemaining()); |
- memcpy(dest_buf->data(), src_buf->data(), src_buf->BytesRemaining()); |
-} |
- |
-class CompressedPacketTest : public testing::Test { |
- public: |
- CompressedPacketTest() |
- : mock_reader_(new MockPacketReader), |
- mock_writer_(new MockPacketWriter), |
- compressed_reader_( |
- new CompressedPacketReader(base::WrapUnique(mock_reader_))), |
- compressed_writer_( |
- new CompressedPacketWriter(base::WrapUnique(mock_writer_))) {} |
- ~CompressedPacketTest() override {} |
- |
- protected: |
- // Returns the compressed result of |content|. |
- std::string Compress(const std::string& content) { |
- scoped_refptr<net::StringIOBuffer> content_str_buf( |
- new net::StringIOBuffer(content)); |
- scoped_refptr<net::DrainableIOBuffer> content_buf( |
- new net::DrainableIOBuffer(content_str_buf.get(), |
- content_str_buf->size())); |
- scoped_refptr<net::DrainableIOBuffer> compressed_buf; |
- EXPECT_CALL(*mock_writer_, WritePacket(_, _)) |
- .WillOnce(DoAll(SaveArg<0>(&compressed_buf), |
- InvokeCompletionCallback<1>(net::OK))); |
- net::TestCompletionCallback completion_cb_1; |
- compressed_writer_->WritePacket(content_buf, completion_cb_1.callback()); |
- EXPECT_EQ(net::OK, completion_cb_1.WaitForResult()); |
- return std::string(compressed_buf->data(), |
- compressed_buf->BytesRemaining()); |
- } |
- |
- // Returns the decompressed result of |compressed|. |
- std::string Decompress(const std::string& compressed) { |
- scoped_refptr<net::StringIOBuffer> compressed_str_buf( |
- new net::StringIOBuffer(compressed)); |
- scoped_refptr<net::DrainableIOBuffer> compressed_buf( |
- new net::DrainableIOBuffer(compressed_str_buf.get(), |
- compressed_str_buf->size())); |
- scoped_refptr<net::GrowableIOBuffer> decompressed_buf( |
- new net::GrowableIOBuffer); |
- EXPECT_CALL(*mock_reader_, ReadPacket(_, _)) |
- .WillOnce(DoAll( |
- CopyBuffer<0>(compressed_buf), |
- InvokeCompletionCallback<1>(compressed_buf->BytesRemaining()))); |
- net::TestCompletionCallback completion_cb_2; |
- compressed_reader_->ReadPacket(decompressed_buf, |
- completion_cb_2.callback()); |
- int size = completion_cb_2.WaitForResult(); |
- return std::string(decompressed_buf->data(), size); |
- } |
- |
- bool CheckRoundTrip(const std::string& content) { |
- return Decompress(Compress(content)) == content; |
- } |
- |
- MockPacketReader* mock_reader_; |
- MockPacketWriter* mock_writer_; |
- std::unique_ptr<CompressedPacketReader> compressed_reader_; |
- std::unique_ptr<CompressedPacketWriter> compressed_writer_; |
- testing::InSequence s; |
-}; |
- |
-TEST_F(CompressedPacketTest, Empty) { |
- EXPECT_TRUE(CheckRoundTrip("1234")); |
- EXPECT_TRUE(CheckRoundTrip("")); |
- EXPECT_TRUE(CheckRoundTrip("1234")); |
-} |
- |
-TEST_F(CompressedPacketTest, Simple) { |
- std::string source = "AAAAAAAAAAAAAAAAAAAAAAAAA"; |
- std::string compressed = Compress(source); |
- std::string decompressed = Decompress(compressed); |
- EXPECT_GT(source.size(), compressed.size()); |
- EXPECT_EQ(source, decompressed); |
-} |
- |
-TEST_F(CompressedPacketTest, DisjointSequences) { |
- EXPECT_TRUE(CheckRoundTrip("1234")); |
- EXPECT_TRUE(CheckRoundTrip("5678")); |
- EXPECT_TRUE(CheckRoundTrip("ABCD")); |
-} |
- |
-TEST_F(CompressedPacketTest, AdditiveSequences) { |
- EXPECT_TRUE(CheckRoundTrip("12")); |
- EXPECT_TRUE(CheckRoundTrip("123")); |
- EXPECT_TRUE(CheckRoundTrip("1234")); |
- EXPECT_TRUE(CheckRoundTrip("12345")); |
- EXPECT_TRUE(CheckRoundTrip("123456")); |
-} |
- |
-TEST_F(CompressedPacketTest, ReversedSequences) { |
- EXPECT_TRUE(CheckRoundTrip("123456")); |
- EXPECT_TRUE(CheckRoundTrip("12345")); |
- EXPECT_TRUE(CheckRoundTrip("1234")); |
- EXPECT_TRUE(CheckRoundTrip("123")); |
- EXPECT_TRUE(CheckRoundTrip("12")); |
-} |
- |
-TEST_F(CompressedPacketTest, CompressionStateRetainedAcrossCalls) { |
- // Ensure that a character sequence is encoded in a more compact manner |
- // if it is compressed more than once. |
- int size_1 = Compress("1234").size(); |
- int size_2 = Compress("1234").size(); |
- EXPECT_GT(size_1, size_2); |
-} |
- |
-TEST_F(CompressedPacketTest, LargeInput) { |
- std::string big_str(kMaxPacketPayloadSizeBytes, 'A'); // 3MB of A's. |
- EXPECT_TRUE(CheckRoundTrip(big_str)); |
-} |
- |
-TEST_F(CompressedPacketTest, LowCompressionRatio) { |
- // This size (2338) was found "in the wild" to repro an issue with output |
- // buffer overflows. |
- const int data_size = 2338; |
- |
- EXPECT_TRUE(CheckRoundTrip(base::RandBytesAsString(data_size))); |
- EXPECT_TRUE(CheckRoundTrip(base::RandBytesAsString(data_size))); |
-} |
- |
-TEST_F(CompressedPacketTest, DecompressIllegallyLargePayload) { |
- // We can't use the compressor to compress an illegally sized payload, however |
- // we can concatenate the output of smaller payloads to form an uber-payload. |
- std::string huge_block = |
- Compress(std::string(kMaxPacketPayloadSizeBytes, 'A')) + |
- Compress("1337 payl0ad 0verfl0w 'spl0it"); |
- |
- scoped_refptr<net::StringIOBuffer> compressed_str_buf( |
- new net::StringIOBuffer(huge_block)); |
- scoped_refptr<net::DrainableIOBuffer> compressed_buf( |
- new net::DrainableIOBuffer(compressed_str_buf.get(), |
- compressed_str_buf->size())); |
- scoped_refptr<net::GrowableIOBuffer> decompressed_buf( |
- new net::GrowableIOBuffer); |
- EXPECT_CALL(*mock_reader_, ReadPacket(_, _)) |
- .WillOnce( |
- DoAll(CopyBuffer<0>(compressed_buf), |
- InvokeCompletionCallback<1>(compressed_buf->BytesRemaining()))); |
- net::TestCompletionCallback completion_cb_2; |
- compressed_reader_->ReadPacket(decompressed_buf, completion_cb_2.callback()); |
- EXPECT_EQ(net::ERR_FILE_TOO_BIG, completion_cb_2.WaitForResult()); |
-} |
- |
-TEST_F(CompressedPacketTest, CompressIllegallyLargePayload) { |
- scoped_refptr<net::IOBuffer> big_buf( |
- new net::IOBuffer(kMaxPacketPayloadSizeBytes + 1)); |
- scoped_refptr<net::DrainableIOBuffer> content_buf(new net::DrainableIOBuffer( |
- big_buf.get(), kMaxPacketPayloadSizeBytes + 1)); |
- net::TestCompletionCallback completion_cb_1; |
- compressed_writer_->WritePacket(content_buf, completion_cb_1.callback()); |
- EXPECT_EQ(net::ERR_FILE_TOO_BIG, completion_cb_1.WaitForResult()); |
-} |
- |
-TEST_F(CompressedPacketTest, PayloadSmallerThanDeclared) { |
- scoped_refptr<net::StringIOBuffer> content_str_buf( |
- new net::StringIOBuffer("Just a small payload")); |
- scoped_refptr<net::DrainableIOBuffer> content_buf(new net::DrainableIOBuffer( |
- content_str_buf.get(), content_str_buf->size())); |
- scoped_refptr<net::DrainableIOBuffer> compressed_buf; |
- EXPECT_CALL(*mock_writer_, WritePacket(_, _)) |
- .WillOnce(DoAll(SaveArg<0>(&compressed_buf), |
- InvokeCompletionCallback<1>(net::OK))); |
- net::TestCompletionCallback completion_cb_1; |
- compressed_writer_->WritePacket(content_buf, completion_cb_1.callback()); |
- EXPECT_EQ(net::OK, completion_cb_1.WaitForResult()); |
- |
- // Increase the payload size header significantly. |
- *(reinterpret_cast<uint32_t*>(compressed_buf->data())) = |
- base::HostToNet32(kMaxPacketPayloadSizeBytes - 1); |
- |
- EXPECT_CALL(*mock_reader_, ReadPacket(_, _)) |
- .WillOnce( |
- DoAll(CopyBuffer<0>(compressed_buf), |
- InvokeCompletionCallback<1>(compressed_buf->BytesRemaining()))); |
- net::TestCompletionCallback completion_cb_2; |
- compressed_reader_->ReadPacket(make_scoped_refptr(new net::GrowableIOBuffer), |
- completion_cb_2.callback()); |
- EXPECT_EQ(net::ERR_UNEXPECTED, completion_cb_2.WaitForResult()); |
-} |
- |
-} // namespace |
-} // namespace blimp |