blob: f71bcc8ea79e61f7c5cf98c3f11d219f046edc88 [file] [log] [blame]
Alex Deymo338f3402018-03-23 13:40:17 +01001// Copyright 2018 The Chromium OS Authors. All rights reserved.
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 "bsdiff/brotli_decompressor.h"
6
7#include <memory>
8#include <string>
9#include <vector>
10
11#include <gtest/gtest.h>
12
13namespace {
14
15// echo -n "Hello!" | brotli -9 | hexdump -v -e '" " 11/1 "0x%02x, " "\n"'
16constexpr uint8_t kBrotliHello[] = {
17 0x8b, 0x02, 0x80, 0x48, 0x65, 0x6c, 0x6c, 0x6f, 0x21, 0x03,
18};
19
20} // namespace
21
22namespace bsdiff {
23
24class BrotliDecompressorTest : public testing::Test {
25 protected:
26 void SetUp() {
27 decompressor_.reset(new BrotliDecompressor());
28 EXPECT_NE(nullptr, decompressor_.get());
29 }
30
31 std::unique_ptr<BrotliDecompressor> decompressor_;
32};
33
34TEST_F(BrotliDecompressorTest, SmokeTest) {
35 EXPECT_TRUE(decompressor_->SetInputData(kBrotliHello, sizeof(kBrotliHello)));
36 std::vector<uint8_t> output_data(6);
37 EXPECT_TRUE(decompressor_->Read(output_data.data(), output_data.size()));
38 std::string hello = "Hello!";
39 EXPECT_EQ(std::vector<uint8_t>(hello.begin(), hello.end()), output_data);
40}
41
42TEST_F(BrotliDecompressorTest, ReadingFromEmptyFileTest) {
43 uint8_t data = 0;
44 EXPECT_TRUE(decompressor_->SetInputData(&data, 0));
45
46 uint8_t output_data[10];
47 EXPECT_FALSE(decompressor_->Read(output_data, sizeof(output_data)));
48}
49
50// Check that we fail to read from a truncated file.
51TEST_F(BrotliDecompressorTest, ReadingFromTruncatedFileTest) {
52 // We feed only half of the compressed file.
53 EXPECT_TRUE(
54 decompressor_->SetInputData(kBrotliHello, sizeof(kBrotliHello) / 2));
55 uint8_t output_data[6];
56 EXPECT_FALSE(decompressor_->Read(output_data, sizeof(output_data)));
57}
58
59// Check that we fail to read more than it is available in the file.
60TEST_F(BrotliDecompressorTest, ReadingMoreThanAvailableTest) {
61 EXPECT_TRUE(decompressor_->SetInputData(kBrotliHello, sizeof(kBrotliHello)));
62 uint8_t output_data[1000];
63 EXPECT_FALSE(decompressor_->Read(output_data, sizeof(output_data)));
64}
65
66} // namespace bsdiff