// Copyright 2018 The Chromium OS 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 "bsdiff/brotli_decompressor.h"
|
|
#include <memory>
|
#include <string>
|
#include <vector>
|
|
#include <gtest/gtest.h>
|
|
namespace {
|
|
// echo -n "Hello!" | brotli -9 | hexdump -v -e '" " 11/1 "0x%02x, " "\n"'
|
constexpr uint8_t kBrotliHello[] = {
|
0x8b, 0x02, 0x80, 0x48, 0x65, 0x6c, 0x6c, 0x6f, 0x21, 0x03,
|
};
|
|
} // namespace
|
|
namespace bsdiff {
|
|
class BrotliDecompressorTest : public testing::Test {
|
protected:
|
void SetUp() {
|
decompressor_.reset(new BrotliDecompressor());
|
EXPECT_NE(nullptr, decompressor_.get());
|
}
|
|
std::unique_ptr<BrotliDecompressor> decompressor_;
|
};
|
|
TEST_F(BrotliDecompressorTest, SmokeTest) {
|
EXPECT_TRUE(decompressor_->SetInputData(kBrotliHello, sizeof(kBrotliHello)));
|
std::vector<uint8_t> output_data(6);
|
EXPECT_TRUE(decompressor_->Read(output_data.data(), output_data.size()));
|
std::string hello = "Hello!";
|
EXPECT_EQ(std::vector<uint8_t>(hello.begin(), hello.end()), output_data);
|
}
|
|
TEST_F(BrotliDecompressorTest, ReadingFromEmptyFileTest) {
|
uint8_t data = 0;
|
EXPECT_TRUE(decompressor_->SetInputData(&data, 0));
|
|
uint8_t output_data[10];
|
EXPECT_FALSE(decompressor_->Read(output_data, sizeof(output_data)));
|
}
|
|
// Check that we fail to read from a truncated file.
|
TEST_F(BrotliDecompressorTest, ReadingFromTruncatedFileTest) {
|
// We feed only half of the compressed file.
|
EXPECT_TRUE(
|
decompressor_->SetInputData(kBrotliHello, sizeof(kBrotliHello) / 2));
|
uint8_t output_data[6];
|
EXPECT_FALSE(decompressor_->Read(output_data, sizeof(output_data)));
|
}
|
|
// Check that we fail to read more than it is available in the file.
|
TEST_F(BrotliDecompressorTest, ReadingMoreThanAvailableTest) {
|
EXPECT_TRUE(decompressor_->SetInputData(kBrotliHello, sizeof(kBrotliHello)));
|
uint8_t output_data[1000];
|
EXPECT_FALSE(decompressor_->Read(output_data, sizeof(output_data)));
|
}
|
|
} // namespace bsdiff
|
