Index: net/server/http_connection_unittest.cc |
diff --git a/net/server/http_connection_unittest.cc b/net/server/http_connection_unittest.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..488fd6fb15b40c07457c60f89cbe6b2b96a0a142 |
--- /dev/null |
+++ b/net/server/http_connection_unittest.cc |
@@ -0,0 +1,331 @@ |
+// Copyright 2014 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 "net/server/http_connection.h" |
+ |
+#include <string> |
+ |
+#include "base/memory/ref_counted.h" |
+#include "base/strings/string_piece.h" |
+#include "testing/gtest/include/gtest/gtest.h" |
+ |
+namespace net { |
+namespace { |
+ |
+std::string GetTestString(int size) { |
+ std::string test_string; |
+ for (int i = 0; i < size; ++i) { |
+ test_string.push_back('A' + (i % 26)); |
+ } |
+ return test_string; |
+} |
+ |
+TEST(HttpConnectionTest, ReadIOBuffer_SetCapacity) { |
+ scoped_refptr<HttpConnection::ReadIOBuffer> buffer( |
+ new HttpConnection::ReadIOBuffer); |
+ EXPECT_EQ(HttpConnection::ReadIOBuffer::kInitialBufSize + 0, |
+ buffer->GetCapacity()); |
+ EXPECT_EQ(HttpConnection::ReadIOBuffer::kInitialBufSize + 0, |
+ buffer->RemainingCapacity()); |
+ EXPECT_EQ(0, buffer->GetSize()); |
+ |
+ const int kNewCapacity = HttpConnection::ReadIOBuffer::kInitialBufSize + 128; |
+ buffer->SetCapacity(kNewCapacity); |
+ EXPECT_EQ(kNewCapacity, buffer->GetCapacity()); |
+ EXPECT_EQ(kNewCapacity, buffer->RemainingCapacity()); |
+ EXPECT_EQ(0, buffer->GetSize()); |
+} |
+ |
+TEST(HttpConnectionTest, ReadIOBuffer_SetCapacity_WithData) { |
+ scoped_refptr<HttpConnection::ReadIOBuffer> buffer( |
+ new HttpConnection::ReadIOBuffer); |
+ EXPECT_EQ(HttpConnection::ReadIOBuffer::kInitialBufSize + 0, |
+ buffer->GetCapacity()); |
+ EXPECT_EQ(HttpConnection::ReadIOBuffer::kInitialBufSize + 0, |
+ buffer->RemainingCapacity()); |
+ |
+ // Write arbitrary data up to kInitialBufSize. |
+ const std::string kReadData( |
+ GetTestString(HttpConnection::ReadIOBuffer::kInitialBufSize)); |
+ memcpy(buffer->data(), kReadData.data(), kReadData.size()); |
+ buffer->DidRead(kReadData.size()); |
+ EXPECT_EQ(HttpConnection::ReadIOBuffer::kInitialBufSize + 0, |
+ buffer->GetCapacity()); |
+ EXPECT_EQ(HttpConnection::ReadIOBuffer::kInitialBufSize - |
+ static_cast<int>(kReadData.size()), |
+ buffer->RemainingCapacity()); |
+ EXPECT_EQ(static_cast<int>(kReadData.size()), buffer->GetSize()); |
+ EXPECT_EQ(kReadData, |
+ base::StringPiece(buffer->StartOfBuffer(), buffer->GetSize())); |
+ |
+ // Check if read data in the buffer is same after SetCapacity(). |
+ const int kNewCapacity = HttpConnection::ReadIOBuffer::kInitialBufSize + 128; |
+ buffer->SetCapacity(kNewCapacity); |
+ EXPECT_EQ(kNewCapacity, buffer->GetCapacity()); |
+ EXPECT_EQ(kNewCapacity - static_cast<int>(kReadData.size()), |
+ buffer->RemainingCapacity()); |
+ EXPECT_EQ(static_cast<int>(kReadData.size()), buffer->GetSize()); |
+ EXPECT_EQ(kReadData, |
+ base::StringPiece(buffer->StartOfBuffer(), buffer->GetSize())); |
+} |
+ |
+TEST(HttpConnectionTest, ReadIOBuffer_IncreaseCapacity) { |
+ scoped_refptr<HttpConnection::ReadIOBuffer> buffer( |
+ new HttpConnection::ReadIOBuffer); |
+ EXPECT_TRUE(buffer->IncreaseCapacity()); |
+ const int kExpectedInitialBufSize = |
+ HttpConnection::ReadIOBuffer::kInitialBufSize * |
+ HttpConnection::ReadIOBuffer::kCapacityIncreaseFactor; |
+ EXPECT_EQ(kExpectedInitialBufSize, buffer->GetCapacity()); |
+ EXPECT_EQ(kExpectedInitialBufSize, buffer->RemainingCapacity()); |
+ EXPECT_EQ(0, buffer->GetSize()); |
+ |
+ // Increase capacity until it fails. |
+ while (buffer->IncreaseCapacity()); |
+ EXPECT_FALSE(buffer->IncreaseCapacity()); |
+ EXPECT_EQ(HttpConnection::ReadIOBuffer::kDefaultMaxBufferSize + 0, |
+ buffer->max_buffer_size()); |
+ EXPECT_EQ(HttpConnection::ReadIOBuffer::kDefaultMaxBufferSize + 0, |
+ buffer->GetCapacity()); |
+ |
+ // Enlarge capacity limit. |
+ buffer->set_max_buffer_size(buffer->max_buffer_size() * 2); |
+ EXPECT_TRUE(buffer->IncreaseCapacity()); |
+ EXPECT_EQ(HttpConnection::ReadIOBuffer::kDefaultMaxBufferSize * |
+ HttpConnection::ReadIOBuffer::kCapacityIncreaseFactor, |
+ buffer->GetCapacity()); |
+ |
+ // Shrink capacity limit. It doesn't change capacity itself. |
+ buffer->set_max_buffer_size( |
+ HttpConnection::ReadIOBuffer::kDefaultMaxBufferSize / 2); |
+ EXPECT_FALSE(buffer->IncreaseCapacity()); |
+ EXPECT_EQ(HttpConnection::ReadIOBuffer::kDefaultMaxBufferSize * |
+ HttpConnection::ReadIOBuffer::kCapacityIncreaseFactor, |
+ buffer->GetCapacity()); |
+} |
+ |
+TEST(HttpConnectionTest, ReadIOBuffer_IncreaseCapacity_WithData) { |
+ scoped_refptr<HttpConnection::ReadIOBuffer> buffer( |
+ new HttpConnection::ReadIOBuffer); |
+ EXPECT_TRUE(buffer->IncreaseCapacity()); |
+ const int kExpectedInitialBufSize = |
+ HttpConnection::ReadIOBuffer::kInitialBufSize * |
+ HttpConnection::ReadIOBuffer::kCapacityIncreaseFactor; |
+ EXPECT_EQ(kExpectedInitialBufSize, buffer->GetCapacity()); |
+ EXPECT_EQ(kExpectedInitialBufSize, buffer->RemainingCapacity()); |
+ EXPECT_EQ(0, buffer->GetSize()); |
+ |
+ // Write arbitrary data up to kExpectedInitialBufSize. |
+ std::string kReadData(GetTestString(kExpectedInitialBufSize)); |
+ memcpy(buffer->data(), kReadData.data(), kReadData.size()); |
+ buffer->DidRead(kReadData.size()); |
+ EXPECT_EQ(kExpectedInitialBufSize, buffer->GetCapacity()); |
+ EXPECT_EQ(kExpectedInitialBufSize - static_cast<int>(kReadData.size()), |
+ buffer->RemainingCapacity()); |
+ EXPECT_EQ(static_cast<int>(kReadData.size()), buffer->GetSize()); |
+ EXPECT_EQ(kReadData, |
+ base::StringPiece(buffer->StartOfBuffer(), buffer->GetSize())); |
+ |
+ // Increase capacity until it fails and check if read data in the buffer is |
+ // same. |
+ while (buffer->IncreaseCapacity()); |
+ EXPECT_FALSE(buffer->IncreaseCapacity()); |
+ EXPECT_EQ(HttpConnection::ReadIOBuffer::kDefaultMaxBufferSize + 0, |
+ buffer->max_buffer_size()); |
+ EXPECT_EQ(HttpConnection::ReadIOBuffer::kDefaultMaxBufferSize + 0, |
+ buffer->GetCapacity()); |
+ EXPECT_EQ(HttpConnection::ReadIOBuffer::kDefaultMaxBufferSize - |
+ static_cast<int>(kReadData.size()), |
+ buffer->RemainingCapacity()); |
+ EXPECT_EQ(static_cast<int>(kReadData.size()), buffer->GetSize()); |
+ EXPECT_EQ(kReadData, |
+ base::StringPiece(buffer->StartOfBuffer(), buffer->GetSize())); |
+} |
+ |
+TEST(HttpConnectionTest, ReadIOBuffer_DidRead_DidConsume) { |
+ scoped_refptr<HttpConnection::ReadIOBuffer> buffer( |
+ new HttpConnection::ReadIOBuffer); |
+ const char* start_of_buffer = buffer->StartOfBuffer(); |
+ EXPECT_EQ(start_of_buffer, buffer->data()); |
+ |
+ // Read data. |
+ const int kReadLength = 128; |
+ const std::string kReadData(GetTestString(kReadLength)); |
+ memcpy(buffer->data(), kReadData.data(), kReadLength); |
+ buffer->DidRead(kReadLength); |
+ // No change in total capacity. |
+ EXPECT_EQ(HttpConnection::ReadIOBuffer::kInitialBufSize + 0, |
+ buffer->GetCapacity()); |
+ // Change in unused capacity because of read data. |
+ EXPECT_EQ(HttpConnection::ReadIOBuffer::kInitialBufSize - kReadLength, |
+ buffer->RemainingCapacity()); |
+ EXPECT_EQ(kReadLength, buffer->GetSize()); |
+ // No change in start pointers of read data. |
+ EXPECT_EQ(start_of_buffer, buffer->StartOfBuffer()); |
+ // Change in start pointer of unused buffer. |
+ EXPECT_EQ(start_of_buffer + kReadLength, buffer->data()); |
+ // Test read data. |
+ EXPECT_EQ(kReadData, std::string(buffer->StartOfBuffer(), buffer->GetSize())); |
+ |
+ // Consume data partially. |
+ const int kConsumedLength = 32; |
+ ASSERT_LT(kConsumedLength, kReadLength); |
+ buffer->DidConsume(kConsumedLength); |
+ // Capacity reduced because read data was too small comparing to capacity. |
+ EXPECT_EQ(HttpConnection::ReadIOBuffer::kInitialBufSize / |
+ HttpConnection::ReadIOBuffer::kCapacityIncreaseFactor, |
+ buffer->GetCapacity()); |
+ // Change in unused capacity because of read data. |
+ EXPECT_EQ(HttpConnection::ReadIOBuffer::kInitialBufSize / |
+ HttpConnection::ReadIOBuffer::kCapacityIncreaseFactor - |
+ kReadLength + kConsumedLength, |
+ buffer->RemainingCapacity()); |
+ // Change in read size. |
+ EXPECT_EQ(kReadLength - kConsumedLength, buffer->GetSize()); |
+ // Start data could be changed even when capacity is reduced. |
+ start_of_buffer = buffer->StartOfBuffer(); |
+ // Change in start pointer of unused buffer. |
+ EXPECT_EQ(start_of_buffer + kReadLength - kConsumedLength, buffer->data()); |
+ // Change in read data. |
+ EXPECT_EQ(kReadData.substr(kConsumedLength), |
+ std::string(buffer->StartOfBuffer(), buffer->GetSize())); |
+ |
+ // Read more data. |
+ const int kReadLength2 = 64; |
+ buffer->DidRead(kReadLength2); |
+ // No change in total capacity. |
+ EXPECT_EQ(HttpConnection::ReadIOBuffer::kInitialBufSize / |
+ HttpConnection::ReadIOBuffer::kCapacityIncreaseFactor, |
+ buffer->GetCapacity()); |
+ // Change in unused capacity because of read data. |
+ EXPECT_EQ(HttpConnection::ReadIOBuffer::kInitialBufSize / |
+ HttpConnection::ReadIOBuffer::kCapacityIncreaseFactor - |
+ kReadLength + kConsumedLength - kReadLength2, |
+ buffer->RemainingCapacity()); |
+ // Change in read size |
+ EXPECT_EQ(kReadLength - kConsumedLength + kReadLength2, buffer->GetSize()); |
+ // No change in start pointer of read part. |
+ EXPECT_EQ(start_of_buffer, buffer->StartOfBuffer()); |
+ // Change in start pointer of unused buffer. |
+ EXPECT_EQ(start_of_buffer + kReadLength - kConsumedLength + kReadLength2, |
+ buffer->data()); |
+ |
+ // Consume data fully. |
+ buffer->DidConsume(kReadLength - kConsumedLength + kReadLength2); |
+ // Capacity reduced again because read data was too small. |
+ EXPECT_EQ(HttpConnection::ReadIOBuffer::kInitialBufSize / |
+ HttpConnection::ReadIOBuffer::kCapacityIncreaseFactor / |
+ HttpConnection::ReadIOBuffer::kCapacityIncreaseFactor, |
+ buffer->GetCapacity()); |
+ EXPECT_EQ(HttpConnection::ReadIOBuffer::kInitialBufSize / |
+ HttpConnection::ReadIOBuffer::kCapacityIncreaseFactor / |
+ HttpConnection::ReadIOBuffer::kCapacityIncreaseFactor, |
+ buffer->RemainingCapacity()); |
+ // All reverts to initial because no data is left. |
+ EXPECT_EQ(0, buffer->GetSize()); |
+ // Start data could be changed even when capacity is reduced. |
+ start_of_buffer = buffer->StartOfBuffer(); |
+ EXPECT_EQ(start_of_buffer, buffer->data()); |
+} |
+ |
+TEST(HttpConnectionTest, QueuedWriteIOBuffer_Append_DidConsume) { |
+ scoped_refptr<HttpConnection::QueuedWriteIOBuffer> buffer( |
+ new HttpConnection::QueuedWriteIOBuffer()); |
+ EXPECT_TRUE(buffer->IsEmpty()); |
+ EXPECT_EQ(0, buffer->GetSizeToWrite()); |
+ EXPECT_EQ(0, buffer->total_size()); |
+ |
+ const std::string kData("data to write"); |
+ EXPECT_TRUE(buffer->Append(kData)); |
+ EXPECT_FALSE(buffer->IsEmpty()); |
+ EXPECT_EQ(static_cast<int>(kData.size()), buffer->GetSizeToWrite()); |
+ EXPECT_EQ(static_cast<int>(kData.size()), buffer->total_size()); |
+ // First data to write is same to kData. |
+ EXPECT_EQ(kData, base::StringPiece(buffer->data(), buffer->GetSizeToWrite())); |
+ |
+ const std::string kData2("more data to write"); |
+ EXPECT_TRUE(buffer->Append(kData2)); |
+ EXPECT_FALSE(buffer->IsEmpty()); |
+ // No change in size to write. |
+ EXPECT_EQ(static_cast<int>(kData.size()), buffer->GetSizeToWrite()); |
+ // Change in total size. |
+ EXPECT_EQ(static_cast<int>(kData.size() + kData2.size()), |
+ buffer->total_size()); |
+ // First data to write has not been changed. Same to kData. |
+ EXPECT_EQ(kData, base::StringPiece(buffer->data(), buffer->GetSizeToWrite())); |
+ |
+ // Consume data partially. |
+ const int kConsumedLength = kData.length() - 1; |
+ buffer->DidConsume(kConsumedLength); |
+ EXPECT_FALSE(buffer->IsEmpty()); |
+ // Change in size to write. |
+ EXPECT_EQ(static_cast<int>(kData.size()) - kConsumedLength, |
+ buffer->GetSizeToWrite()); |
+ // Change in total size. |
+ EXPECT_EQ(static_cast<int>(kData.size() + kData2.size()) - kConsumedLength, |
+ buffer->total_size()); |
+ // First data to write has shrinked. |
+ EXPECT_EQ(kData.substr(kConsumedLength), |
+ base::StringPiece(buffer->data(), buffer->GetSizeToWrite())); |
+ |
+ // Consume first data fully. |
+ buffer->DidConsume(kData.size() - kConsumedLength); |
+ EXPECT_FALSE(buffer->IsEmpty()); |
+ // Now, size to write is size of data added second. |
+ EXPECT_EQ(static_cast<int>(kData2.size()), buffer->GetSizeToWrite()); |
+ // Change in total size. |
+ EXPECT_EQ(static_cast<int>(kData2.size()), buffer->total_size()); |
+ // First data to write has changed to kData2. |
+ EXPECT_EQ(kData2, |
+ base::StringPiece(buffer->data(), buffer->GetSizeToWrite())); |
+ |
+ // Consume second data fully. |
+ buffer->DidConsume(kData2.size()); |
+ EXPECT_TRUE(buffer->IsEmpty()); |
+ EXPECT_EQ(0, buffer->GetSizeToWrite()); |
+ EXPECT_EQ(0, buffer->total_size()); |
+} |
+ |
+TEST(HttpConnectionTest, QueuedWriteIOBuffer_TotalSizeLimit) { |
+ scoped_refptr<HttpConnection::QueuedWriteIOBuffer> buffer( |
+ new HttpConnection::QueuedWriteIOBuffer()); |
+ EXPECT_EQ(HttpConnection::QueuedWriteIOBuffer::kDefaultMaxBufferSize + 0, |
+ buffer->max_buffer_size()); |
+ |
+ // Set total size limit very small. |
+ buffer->set_max_buffer_size(10); |
+ |
+ const int kDataLength = 4; |
+ const std::string kData(kDataLength, 'd'); |
+ EXPECT_TRUE(buffer->Append(kData)); |
+ EXPECT_EQ(kDataLength, buffer->total_size()); |
+ EXPECT_TRUE(buffer->Append(kData)); |
+ EXPECT_EQ(kDataLength * 2, buffer->total_size()); |
+ |
+ // Cannot append more data because it exceeds the limit. |
+ EXPECT_FALSE(buffer->Append(kData)); |
+ EXPECT_EQ(kDataLength * 2, buffer->total_size()); |
+ |
+ // Consume data partially. |
+ const int kConsumedLength = 2; |
+ buffer->DidConsume(kConsumedLength); |
+ EXPECT_EQ(kDataLength * 2 - kConsumedLength, buffer->total_size()); |
+ |
+ // Can add more data. |
+ EXPECT_TRUE(buffer->Append(kData)); |
+ EXPECT_EQ(kDataLength * 3 - kConsumedLength, buffer->total_size()); |
+ |
+ // Cannot append more data because it exceeds the limit. |
+ EXPECT_FALSE(buffer->Append(kData)); |
+ EXPECT_EQ(kDataLength * 3 - kConsumedLength, buffer->total_size()); |
+ |
+ // Enlarge limit. |
+ buffer->set_max_buffer_size(20); |
+ // Can add more data. |
+ EXPECT_TRUE(buffer->Append(kData)); |
+ EXPECT_EQ(kDataLength * 4 - kConsumedLength, buffer->total_size()); |
+} |
+ |
+} // namespace |
+} // namespace net |