Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(771)

Unified Diff: blimp/net/compressed_packet_unittest.cc

Issue 2632803002: Remove all blimp network code. (Closed)
Patch Set: merge from origin/master for good measure Created 3 years, 11 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
« no previous file with comments | « blimp/net/compressed_packet_reader.cc ('k') | blimp/net/compressed_packet_writer.h » ('j') | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
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
« no previous file with comments | « blimp/net/compressed_packet_reader.cc ('k') | blimp/net/compressed_packet_writer.h » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698