Implement WebSocketDeflater.

net/websockets/websocket_deflater_test.cc

+// Copyright (c) 2013 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/websockets/websocket_deflater.h"
+
+#include <string>
+
+#include "base/memory/ref_counted.h"
+#include "net/base/io_buffer.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace net {
+
+namespace {
+
+std::string ToString(IOBufferWithSize* buffer) {
+  return std::string(buffer->data(), buffer->size());
+}
+
+TEST(WebSocketDeflaterTest, Construct) {
+  WebSocketDeflater deflater(WebSocketDeflater::TAKE_OVER_CONTEXT);
+  deflater.Initialize(8);
+  ASSERT_EQ(0u, deflater.CurrentOutputSize());
+  ASSERT_TRUE(deflater.Finish());
+  scoped_refptr<IOBufferWithSize> actual =
+      deflater.GetOutput(deflater.CurrentOutputSize());
+  EXPECT_EQ(std::string("\x02\00", 2), ToString(actual.get()));
+  ASSERT_EQ(0u, deflater.CurrentOutputSize());
+}
+
+TEST(WebSocketDeflaterTest, DeflateHelloTakeOverContext) {
+  WebSocketDeflater deflater(WebSocketDeflater::TAKE_OVER_CONTEXT);
+  deflater.Initialize(15);
+  scoped_refptr<IOBufferWithSize> actual1, actual2;
+
+  ASSERT_TRUE(deflater.AddBytes("Hello", 5));
+  ASSERT_TRUE(deflater.Finish());
+  actual1 = deflater.GetOutput(deflater.CurrentOutputSize());
+  EXPECT_EQ(std::string("\xf2\x48\xcd\xc9\xc9\x07\x00", 7),
+            ToString(actual1.get()));
+
+  ASSERT_TRUE(deflater.AddBytes("Hello", 5));
+  ASSERT_TRUE(deflater.Finish());
+  actual2 = deflater.GetOutput(deflater.CurrentOutputSize());
+  EXPECT_EQ(std::string("\xf2\x00\x11\x00\x00", 5), ToString(actual2.get()));
+}
+
+TEST(WebSocketDeflaterTest, DeflateHelloDoNotTakeOverContext) {
+  WebSocketDeflater deflater(WebSocketDeflater::DO_NOT_TAKE_OVER_CONTEXT);
+  deflater.Initialize(15);
+  scoped_refptr<IOBufferWithSize> actual1, actual2;
+
+  ASSERT_TRUE(deflater.AddBytes("Hello", 5));
+  ASSERT_TRUE(deflater.Finish());
+  actual1 = deflater.GetOutput(deflater.CurrentOutputSize());
+  EXPECT_EQ(std::string("\xf2\x48\xcd\xc9\xc9\x07\x00", 7),
+            ToString(actual1.get()));
+
+  ASSERT_TRUE(deflater.AddBytes("Hello", 5));
+  ASSERT_TRUE(deflater.Finish());
+  actual2 = deflater.GetOutput(deflater.CurrentOutputSize());
+  EXPECT_EQ(std::string("\xf2\x48\xcd\xc9\xc9\x07\x00", 7),
+            ToString(actual2.get()));
+}
+
+TEST(WebSocketDeflaterTest, MultipleAddBytesCalls) {
+  WebSocketDeflater deflater(WebSocketDeflater::DO_NOT_TAKE_OVER_CONTEXT);
+  deflater.Initialize(15);
+  std::string input(32, 'a');
+  scoped_refptr<IOBufferWithSize> actual;
+
+  for (size_t i = 0; i < input.size(); ++i) {
+    ASSERT_TRUE(deflater.AddBytes(&input[i], 1));
+  }
+  ASSERT_TRUE(deflater.Finish());
+  actual = deflater.GetOutput(deflater.CurrentOutputSize());
+  EXPECT_EQ(std::string("\x4a\x4c\xc4\x0f\x00\x00", 6), ToString(actual.get()));
+}
+
+TEST(WebSocketDeflaterTest, GetMultipleDeflatedOutput) {
+  WebSocketDeflater deflater(WebSocketDeflater::TAKE_OVER_CONTEXT);
+  deflater.Initialize(15);
+  scoped_refptr<IOBufferWithSize> actual;
+
+  ASSERT_TRUE(deflater.AddBytes("Hello", 5));
+  ASSERT_TRUE(deflater.Finish());
+  deflater.PushSyncMark();
+  ASSERT_TRUE(deflater.Finish());
+  deflater.PushSyncMark();
+  ASSERT_TRUE(deflater.AddBytes("Hello", 5));
+  ASSERT_TRUE(deflater.Finish());
+
+  actual = deflater.GetOutput(deflater.CurrentOutputSize());
+  EXPECT_EQ(std::string("\xf2\x48\xcd\xc9\xc9\x07\x00\x00\x00\xff\xff"
+                        "\x02\x00\x00\x00\xff\xff"
+                        "\xf2\x00\x11\x00\x00", 22),
+            ToString(actual.get()));
+  ASSERT_EQ(0u, deflater.CurrentOutputSize());
+}
+
+TEST(WebSocketDeflaterTest, WindowBits8) {
+  WebSocketDeflater deflater(WebSocketDeflater::DO_NOT_TAKE_OVER_CONTEXT);
+  deflater.Initialize(8);
+  // Set the head and tail of |input| so that back-reference
+  // can be used if the window size is sufficiently-large.
+  const std::string word = "Chromium";
+  std::string input = word + std::string(256, 'a') + word;
+  scoped_refptr<IOBufferWithSize> actual;
+
+  ASSERT_TRUE(deflater.AddBytes(input.data(), input.size()));
+  ASSERT_TRUE(deflater.Finish());
+  actual = deflater.GetOutput(deflater.CurrentOutputSize());
+  EXPECT_EQ(std::string("r\xce(\xca\xcf\xcd,\xcdM\x1c\xe1\xc0\x39\xa3"
+                        "(?7\xb3\x34\x17\x00", 21),
+            ToString(actual.get()));
+}
+
+TEST(WebSocketDeflaterTest, WindowBits10) {
+  WebSocketDeflater deflater(WebSocketDeflater::DO_NOT_TAKE_OVER_CONTEXT);
+  deflater.Initialize(10);
+  // Set the head and tail of |input| so that back-reference
+  // can be used if the window size is sufficiently-large.
+  const std::string word = "Chromium";
+  std::string input = word + std::string(256, 'a') + word;
+  scoped_refptr<IOBufferWithSize> actual;
+
+  ASSERT_TRUE(deflater.AddBytes(input.data(), input.size()));
+  ASSERT_TRUE(deflater.Finish());
+  actual = deflater.GetOutput(deflater.CurrentOutputSize());
+  EXPECT_EQ(
+      std::string("r\xce(\xca\xcf\xcd,\xcdM\x1c\xe1\xc0\x19\x1a\x0e\0\0", 17),
+      ToString(actual.get()));
+}
+
+}  // namespace
+
+}  // namespace net