// 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 <string>

#include "net/quic/quic_spdy_compressor.h"
#include "net/quic/quic_spdy_decompressor.h"
#include "net/quic/spdy_utils.h"
#include "net/quic/test_tools/quic_test_utils.h"
#include "testing/gtest/include/gtest/gtest.h"

using std::string;

namespace net {
namespace test {
namespace {

class QuicSpdyDecompressorTest : public ::testing::Test {
 protected:
  QuicSpdyDecompressor decompressor_;
  QuicSpdyCompressor compressor_;
  TestDecompressorVisitor visitor_;
};

TEST_F(QuicSpdyDecompressorTest, Decompress) {
  SpdyHeaderBlock headers;
  headers[":host"] = "www.google.com";
  headers[":path"] = "/index.hml";
  headers[":scheme"] = "https";

  EXPECT_EQ(1u, decompressor_.current_header_id());
  string compressed_headers = compressor_.CompressHeaders(headers).substr(4);
  EXPECT_EQ(compressed_headers.length(),
            decompressor_.DecompressData(compressed_headers, &visitor_));

  EXPECT_EQ(SpdyUtils::SerializeUncompressedHeaders(headers), visitor_.data());
  EXPECT_EQ(2u, decompressor_.current_header_id());
}

TEST_F(QuicSpdyDecompressorTest, DecompressPartial) {
  SpdyHeaderBlock headers;
  headers[":host"] = "www.google.com";
  headers[":path"] = "/index.hml";
  headers[":scheme"] = "https";
  string compressed_headers = compressor_.CompressHeaders(headers).substr(4);

  for (size_t i = 0; i < compressed_headers.length(); ++i) {
    QuicSpdyDecompressor decompressor;
    TestDecompressorVisitor visitor;

    EXPECT_EQ(1u, decompressor.current_header_id());

    string partial_compressed_headers = compressed_headers.substr(0, i);
    EXPECT_EQ(partial_compressed_headers.length(),
              decompressor.DecompressData(partial_compressed_headers,
                                          &visitor));
    EXPECT_EQ(1u, decompressor.current_header_id()) << "i: " << i;

    string remaining_compressed_headers =
        compressed_headers.substr(partial_compressed_headers.length());
    EXPECT_EQ(remaining_compressed_headers.length(),
              decompressor.DecompressData(remaining_compressed_headers,
                                          &visitor));
    EXPECT_EQ(SpdyUtils::SerializeUncompressedHeaders(headers), visitor.data());

    EXPECT_EQ(2u, decompressor.current_header_id());
  }
}

TEST_F(QuicSpdyDecompressorTest, DecompressAndIgnoreTrailingData) {
  SpdyHeaderBlock headers;
  headers[":host"] = "www.google.com";
  headers[":path"] = "/index.hml";
  headers[":scheme"] = "https";

  string compressed_headers = compressor_.CompressHeaders(headers).substr(4);
  EXPECT_EQ(compressed_headers.length(),
            decompressor_.DecompressData(compressed_headers + "abc123",
                                        &visitor_));

  EXPECT_EQ(SpdyUtils::SerializeUncompressedHeaders(headers), visitor_.data());
}

TEST_F(QuicSpdyDecompressorTest, DecompressError) {
  SpdyHeaderBlock headers;
  headers[":host"] = "www.google.com";
  headers[":path"] = "/index.hml";
  headers[":scheme"] = "https";

  EXPECT_EQ(1u, decompressor_.current_header_id());
  string compressed_headers = compressor_.CompressHeaders(headers).substr(4);
  compressed_headers[compressed_headers.length() - 1] ^= 0x01;
  EXPECT_NE(compressed_headers.length(),
            decompressor_.DecompressData(compressed_headers, &visitor_));

  EXPECT_TRUE(visitor_.error());
  EXPECT_EQ("", visitor_.data());
}

}  // namespace
}  // namespace test
}  // namespace net