1// Copyright 2014 The Chromium Authors
2// Use of this source code is governed by a BSD-style license that can be
3// found in the LICENSE file.
4
5#include "third_party/zlib/google/compression_utils.h"
6
7#include <stddef.h>
8#include <stdint.h>
9
10#include <iterator>
11#include <string>
12
13#include "testing/gtest/include/gtest/gtest.h"
14
15namespace compression {
16
17namespace {
18
19// The data to be compressed by gzip. This is the hex representation of "hello
20// world".
21const uint8_t kData[] = {0x68, 0x65, 0x6c, 0x6c, 0x6f, 0x20,
22                         0x77, 0x6f, 0x72, 0x6c, 0x64};
23
24// This is the string representation of gzip compressed string above. It was
25// obtained by running echo -n "hello world" | gzip -c | hexdump -e '8 1 ",
26// 0x%x"' followed by 0'ing out the OS byte (10th byte) in the header. This is
27// so that the test passes on all platforms (that run various OS'es).
28const uint8_t kCompressedData[] = {
29    0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0xcb,
30    0x48, 0xcd, 0xc9, 0xc9, 0x57, 0x28, 0xcf, 0x2f, 0xca, 0x49, 0x01,
31    0x00, 0x85, 0x11, 0x4a, 0x0d, 0x0b, 0x00, 0x00, 0x00};
32
33}  // namespace
34
35TEST(CompressionUtilsTest, GzipCompression) {
36  std::string data(reinterpret_cast<const char*>(kData), std::size(kData));
37  std::string compressed_data;
38  EXPECT_TRUE(GzipCompress(data, &compressed_data));
39  std::string golden_compressed_data(
40      reinterpret_cast<const char*>(kCompressedData),
41      std::size(kCompressedData));
42  EXPECT_EQ(golden_compressed_data, compressed_data);
43}
44
45TEST(CompressionUtilsTest, GzipUncompression) {
46  std::string compressed_data(reinterpret_cast<const char*>(kCompressedData),
47                              std::size(kCompressedData));
48
49  std::string uncompressed_data;
50  EXPECT_TRUE(GzipUncompress(compressed_data, &uncompressed_data));
51
52  std::string golden_data(reinterpret_cast<const char*>(kData),
53                          std::size(kData));
54  EXPECT_EQ(golden_data, uncompressed_data);
55}
56
57TEST(CompressionUtilsTest, GzipUncompressionFromSpanToString) {
58  std::string uncompressed_data;
59  EXPECT_TRUE(GzipUncompress(kCompressedData, &uncompressed_data));
60
61  std::string golden_data(reinterpret_cast<const char*>(kData),
62                          std::size(kData));
63  EXPECT_EQ(golden_data, uncompressed_data);
64}
65
66// Checks that compressing/decompressing input > 256 bytes works as expected.
67TEST(CompressionUtilsTest, LargeInput) {
68  const size_t kSize = 32 * 1024;
69
70  // Generate a data string of |kSize| for testing.
71  std::string data;
72  data.resize(kSize);
73  for (size_t i = 0; i < kSize; ++i)
74    data[i] = static_cast<char>(i & 0xFF);
75
76  std::string compressed_data;
77  EXPECT_TRUE(GzipCompress(data, &compressed_data));
78
79  std::string uncompressed_data;
80  EXPECT_TRUE(GzipUncompress(compressed_data, &uncompressed_data));
81
82  EXPECT_EQ(data, uncompressed_data);
83}
84
85TEST(CompressionUtilsTest, InPlace) {
86  const std::string original_data(reinterpret_cast<const char*>(kData),
87                                  std::size(kData));
88  const std::string golden_compressed_data(
89      reinterpret_cast<const char*>(kCompressedData),
90      std::size(kCompressedData));
91
92  std::string data(original_data);
93  EXPECT_TRUE(GzipCompress(data, &data));
94  EXPECT_EQ(golden_compressed_data, data);
95  EXPECT_TRUE(GzipUncompress(data, &data));
96  EXPECT_EQ(original_data, data);
97}
98
99}  // namespace compression
100