mirror of
https://github.com/expo/bsdiff.git
synced 2026-11-15 03:38:06 -10:00
Tianjie Xu
77833b6477
We modified the patch writer to have the capability to produce the smallest patch if multiple compressors are given. This type is supported for bsdf2 format only. Example usage: bsdiff --format=bsdf2 --type=bz2:brotli <src_file> <tgt_file> <patch_file> Test: diff & patch an apk file Change-Id: Icd4322f21975b82c5ee09f8feb9321f52c9813a4
122 lines
4.0 KiB
C++
122 lines
4.0 KiB
C++
// Copyright 2017 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/bsdiff_arguments.h"
|
|
|
|
#include <gtest/gtest.h>
|
|
|
|
namespace bsdiff {
|
|
|
|
TEST(BsdiffArgumentsTest, ParseCompressorTypeTest) {
|
|
std::set<CompressorType> types;
|
|
EXPECT_TRUE(BsdiffArguments::ParseCompressorTypes("Brotli", &types));
|
|
EXPECT_EQ(1U, types.size());
|
|
EXPECT_NE(types.end(), types.find(CompressorType::kBrotli));
|
|
|
|
types.clear();
|
|
|
|
EXPECT_TRUE(BsdiffArguments::ParseCompressorTypes("bz2", &types));
|
|
EXPECT_EQ(1U, types.size());
|
|
EXPECT_NE(types.end(), types.find(CompressorType::kBZ2));
|
|
|
|
types.clear();
|
|
|
|
EXPECT_FALSE(BsdiffArguments::ParseCompressorTypes("invalid", &types));
|
|
}
|
|
|
|
TEST(BsdiffArgumentsTest, ParseMultipleCompressorTypeTest) {
|
|
std::set<CompressorType> types;
|
|
EXPECT_TRUE(BsdiffArguments::ParseCompressorTypes("bz2:brotli:nocompression",
|
|
&types));
|
|
EXPECT_EQ(3U, types.size());
|
|
EXPECT_NE(types.end(), types.find(CompressorType::kBrotli));
|
|
EXPECT_NE(types.end(), types.find(CompressorType::kBZ2));
|
|
EXPECT_NE(types.end(), types.find(CompressorType::kNoCompression));
|
|
|
|
types.clear();
|
|
|
|
// No space in the type string.
|
|
EXPECT_FALSE(
|
|
BsdiffArguments::ParseCompressorTypes("bz2 : nocompression", &types));
|
|
}
|
|
|
|
TEST(BsdiffArgumentsTest, ParseBsdiffFormatTest) {
|
|
BsdiffFormat format;
|
|
EXPECT_TRUE(BsdiffArguments::ParseBsdiffFormat("bsdf2", &format));
|
|
EXPECT_EQ(BsdiffFormat::kBsdf2, format);
|
|
|
|
EXPECT_TRUE(BsdiffArguments::ParseBsdiffFormat("Legacy", &format));
|
|
EXPECT_EQ(BsdiffFormat::kLegacy, format);
|
|
|
|
EXPECT_TRUE(BsdiffArguments::ParseBsdiffFormat("bsdiff40", &format));
|
|
EXPECT_EQ(BsdiffFormat::kLegacy, format);
|
|
|
|
EXPECT_TRUE(BsdiffArguments::ParseBsdiffFormat("endsley", &format));
|
|
EXPECT_EQ(BsdiffFormat::kEndsley, format);
|
|
|
|
EXPECT_FALSE(BsdiffArguments::ParseBsdiffFormat("Other", &format));
|
|
}
|
|
|
|
TEST(BsdiffArgumentsTest, ParseQualityTest) {
|
|
int quality;
|
|
EXPECT_TRUE(BsdiffArguments::ParseQuality("9", &quality, 0, 11));
|
|
EXPECT_EQ(9, quality);
|
|
|
|
// Check the out of range quality values.
|
|
EXPECT_FALSE(BsdiffArguments::ParseQuality("30", &quality, 0, 11));
|
|
EXPECT_FALSE(BsdiffArguments::ParseQuality("1234567890", &quality, 0, 1000));
|
|
EXPECT_FALSE(BsdiffArguments::ParseQuality("aabb", &quality, 0, 1000));
|
|
}
|
|
|
|
TEST(BsdiffArgumentsTest, ParseMinLengthTest) {
|
|
size_t len;
|
|
EXPECT_TRUE(BsdiffArguments::ParseMinLength("11", &len));
|
|
EXPECT_EQ(11U, len);
|
|
|
|
// Check the out of range quality values.
|
|
EXPECT_FALSE(BsdiffArguments::ParseMinLength("-1", &len));
|
|
EXPECT_FALSE(BsdiffArguments::ParseMinLength("aabb", &len));
|
|
}
|
|
|
|
TEST(BsdiffArgumentsTest, ArgumentsValidTest) {
|
|
// Default arguments using BsdiffFormat::kLegacy and CompressorType::kBZ2
|
|
// should be valid.
|
|
EXPECT_TRUE(BsdiffArguments().IsValid());
|
|
|
|
// brotli is not supported for BsdiffFormat::kLegacy.
|
|
EXPECT_FALSE(
|
|
BsdiffArguments(BsdiffFormat::kLegacy, {CompressorType::kBrotli}, -1)
|
|
.IsValid());
|
|
|
|
EXPECT_TRUE(
|
|
BsdiffArguments(BsdiffFormat::kBsdf2, {CompressorType::kBrotli}, 9)
|
|
.IsValid());
|
|
|
|
// Compression quality out of range for brotli.
|
|
EXPECT_FALSE(
|
|
BsdiffArguments(BsdiffFormat::kBsdf2, {CompressorType::kBrotli}, 20)
|
|
.IsValid());
|
|
}
|
|
|
|
TEST(BsdiffArgumentsTest, ParseArgumentsSmokeTest) {
|
|
std::vector<const char*> args = {"bsdiff", "--format=bsdf2",
|
|
"--type=brotli:bz2", "--brotli_quality=9",
|
|
"--minlen=12"};
|
|
|
|
BsdiffArguments arguments;
|
|
EXPECT_TRUE(
|
|
arguments.ParseCommandLine(args.size(), const_cast<char**>(args.data())));
|
|
|
|
EXPECT_EQ(BsdiffFormat::kBsdf2, arguments.format());
|
|
|
|
std::vector<CompressorType> types = {CompressorType::kBZ2,
|
|
CompressorType::kBrotli};
|
|
EXPECT_EQ(types, arguments.compressor_types());
|
|
|
|
EXPECT_EQ(9, arguments.brotli_quality());
|
|
EXPECT_EQ(12, arguments.min_length());
|
|
}
|
|
|
|
} // namespace bsdiff
|