1*a3a45f30SXin Li // Copyright 2017 The Chromium OS Authors. All rights reserved.
2*a3a45f30SXin Li // Use of this source code is governed by a BSD-style license that can be
3*a3a45f30SXin Li // found in the LICENSE file.
4*a3a45f30SXin Li
5*a3a45f30SXin Li #include "bsdiff/brotli_compressor.h"
6*a3a45f30SXin Li
7*a3a45f30SXin Li #include <gtest/gtest.h>
8*a3a45f30SXin Li
9*a3a45f30SXin Li #include "bsdiff/brotli_decompressor.h"
10*a3a45f30SXin Li
11*a3a45f30SXin Li namespace {
12*a3a45f30SXin Li
13*a3a45f30SXin Li constexpr uint8_t kHelloWorld[] = {
14*a3a45f30SXin Li 0x48, 0x65, 0x6c, 0x6c, 0x6f, 0x20, 0x57, 0x6f, 0x72, 0x6c, 0x64, 0x0a,
15*a3a45f30SXin Li };
16*a3a45f30SXin Li } // namespace
17*a3a45f30SXin Li
18*a3a45f30SXin Li namespace bsdiff {
19*a3a45f30SXin Li
TEST(BrotliCompressorTest,BrotliCompressorSmoke)20*a3a45f30SXin Li TEST(BrotliCompressorTest, BrotliCompressorSmoke) {
21*a3a45f30SXin Li BrotliCompressor brotli_compressor(11);
22*a3a45f30SXin Li EXPECT_TRUE(brotli_compressor.Write(kHelloWorld, sizeof(kHelloWorld)));
23*a3a45f30SXin Li EXPECT_TRUE(brotli_compressor.Finish());
24*a3a45f30SXin Li std::vector<uint8_t> compressed_data = brotli_compressor.GetCompressedData();
25*a3a45f30SXin Li EXPECT_GT(compressed_data.size(), static_cast<size_t>(0));
26*a3a45f30SXin Li
27*a3a45f30SXin Li // Run decompressor and check we can get the exact same data as kHelloWorld.
28*a3a45f30SXin Li std::vector<uint8_t> decompressed_data(sizeof(kHelloWorld));
29*a3a45f30SXin Li BrotliDecompressor brotli_decompressor;
30*a3a45f30SXin Li EXPECT_TRUE(brotli_decompressor.SetInputData(compressed_data.data(),
31*a3a45f30SXin Li compressed_data.size()));
32*a3a45f30SXin Li EXPECT_TRUE(
33*a3a45f30SXin Li brotli_decompressor.Read(decompressed_data.data(), sizeof(kHelloWorld)));
34*a3a45f30SXin Li EXPECT_TRUE(brotli_decompressor.Close());
35*a3a45f30SXin Li EXPECT_EQ(
36*a3a45f30SXin Li std::vector<uint8_t>(kHelloWorld, kHelloWorld + sizeof(kHelloWorld)),
37*a3a45f30SXin Li decompressed_data);
38*a3a45f30SXin Li }
39*a3a45f30SXin Li
TEST(BrotliCompressorTest,BrotliCompressorEmptyStream)40*a3a45f30SXin Li TEST(BrotliCompressorTest, BrotliCompressorEmptyStream) {
41*a3a45f30SXin Li uint8_t unused_value = 0;
42*a3a45f30SXin Li BrotliCompressor brotli_compressor(11);
43*a3a45f30SXin Li EXPECT_TRUE(brotli_compressor.Write(&unused_value, 0));
44*a3a45f30SXin Li EXPECT_TRUE(brotli_compressor.Finish());
45*a3a45f30SXin Li
46*a3a45f30SXin Li std::vector<uint8_t> compressed_data = brotli_compressor.GetCompressedData();
47*a3a45f30SXin Li
48*a3a45f30SXin Li // Check that we can close the decompressor without errors.
49*a3a45f30SXin Li BrotliDecompressor brotli_decompressor;
50*a3a45f30SXin Li EXPECT_TRUE(brotli_decompressor.SetInputData(compressed_data.data(),
51*a3a45f30SXin Li compressed_data.size()));
52*a3a45f30SXin Li EXPECT_TRUE(brotli_decompressor.Close());
53*a3a45f30SXin Li }
54*a3a45f30SXin Li
55*a3a45f30SXin Li } // namespace bsdiff
56