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
|