// 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