summaryrefslogtreecommitdiff
path: root/bsdiff_arguments_unittest.cc
blob: 412fac58c1f301854342583964968ec9c64ccc4c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
// 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