1*a3a45f30SXin Li // Copyright 2018 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/bz2_decompressor.h"
6*a3a45f30SXin Li
7*a3a45f30SXin Li #include <memory>
8*a3a45f30SXin Li
9*a3a45f30SXin Li #include <gtest/gtest.h>
10*a3a45f30SXin Li
11*a3a45f30SXin Li namespace {
12*a3a45f30SXin Li
13*a3a45f30SXin Li // echo -n "Hello!" | bzip2 -9 | hexdump -v -e '" " 11/1 "0x%02x, " "\n"'
14*a3a45f30SXin Li constexpr uint8_t kBZ2Hello[] = {
15*a3a45f30SXin Li 0x42, 0x5a, 0x68, 0x39, 0x31, 0x41, 0x59, 0x26, 0x53, 0x59, 0x1a,
16*a3a45f30SXin Li 0xea, 0x74, 0xba, 0x00, 0x00, 0x00, 0x95, 0x00, 0x20, 0x00, 0x00,
17*a3a45f30SXin Li 0x40, 0x02, 0x04, 0xa0, 0x00, 0x21, 0x83, 0x41, 0x9a, 0x02, 0x5c,
18*a3a45f30SXin Li 0x2e, 0x2e, 0xe4, 0x8a, 0x70, 0xa1, 0x20, 0x35, 0xd4, 0xe9, 0x74,
19*a3a45f30SXin Li };
20*a3a45f30SXin Li
21*a3a45f30SXin Li } // namespace
22*a3a45f30SXin Li
23*a3a45f30SXin Li namespace bsdiff {
24*a3a45f30SXin Li
25*a3a45f30SXin Li class BZ2DecompressorTest : public testing::Test {
26*a3a45f30SXin Li protected:
SetUp()27*a3a45f30SXin Li void SetUp() {
28*a3a45f30SXin Li decompressor_.reset(new BZ2Decompressor());
29*a3a45f30SXin Li EXPECT_NE(nullptr, decompressor_.get());
30*a3a45f30SXin Li }
31*a3a45f30SXin Li
32*a3a45f30SXin Li std::unique_ptr<BZ2Decompressor> decompressor_;
33*a3a45f30SXin Li };
34*a3a45f30SXin Li
TEST_F(BZ2DecompressorTest,ReadingFromEmptyFileTest)35*a3a45f30SXin Li TEST_F(BZ2DecompressorTest, ReadingFromEmptyFileTest) {
36*a3a45f30SXin Li uint8_t data = 0;
37*a3a45f30SXin Li EXPECT_TRUE(decompressor_->SetInputData(&data, 0));
38*a3a45f30SXin Li
39*a3a45f30SXin Li uint8_t output_data[10];
40*a3a45f30SXin Li EXPECT_FALSE(decompressor_->Read(output_data, sizeof(output_data)));
41*a3a45f30SXin Li }
42*a3a45f30SXin Li
43*a3a45f30SXin Li // Check that we fail to read from a truncated file.
TEST_F(BZ2DecompressorTest,ReadingFromTruncatedFileTest)44*a3a45f30SXin Li TEST_F(BZ2DecompressorTest, ReadingFromTruncatedFileTest) {
45*a3a45f30SXin Li // We feed only half of the compressed file.
46*a3a45f30SXin Li EXPECT_TRUE(decompressor_->SetInputData(kBZ2Hello, sizeof(kBZ2Hello) / 2));
47*a3a45f30SXin Li uint8_t output_data[6];
48*a3a45f30SXin Li EXPECT_FALSE(decompressor_->Read(output_data, sizeof(output_data)));
49*a3a45f30SXin Li }
50*a3a45f30SXin Li
51*a3a45f30SXin Li // Check that we fail to read more than it is available in the file.
TEST_F(BZ2DecompressorTest,ReadingMoreThanAvailableTest)52*a3a45f30SXin Li TEST_F(BZ2DecompressorTest, ReadingMoreThanAvailableTest) {
53*a3a45f30SXin Li EXPECT_TRUE(decompressor_->SetInputData(kBZ2Hello, sizeof(kBZ2Hello)));
54*a3a45f30SXin Li uint8_t output_data[1000];
55*a3a45f30SXin Li EXPECT_FALSE(decompressor_->Read(output_data, sizeof(output_data)));
56*a3a45f30SXin Li }
57*a3a45f30SXin Li
58*a3a45f30SXin Li } // namespace bsdiff
59