/*
 * Copyright 2016 Google Inc. All rights reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#include "options.h"

#include <gtest/gtest.h>
#include <gmock/gmock.h>
#include <mutex>
#include <string.h>
#include <vector>
#include "url.h"

namespace speedtest {
namespace {

char *CopyString(const char *s) {
  size_t len = strlen(s);
  char *ret = new char[len + 1];
  strcpy(ret, s);
  return ret;
}

std::mutex options_lock;

void TestOptions(bool valid,
                 std::vector<const char *> args,
                 Options *options) {
  int argc = args.size() + 1;
  char **argv = new char *[argc];
  argv[0] = CopyString("speedtest");
  int i = 1;
  for (const char *s : args) {
    argv[i++] = CopyString(s);
  }
  {
    std::lock_guard<std::mutex> lock(options_lock);
    EXPECT_EQ(valid, ParseOptions(argc, argv, options));
  }
  for (int i = 0; i < argc; i++) {
    delete argv[i];
  }
  delete[] argv;
}

void TestOptions(bool valid, std::vector<const char *> args) {
  Options options;
  TestOptions(valid, args, &options);
}

void TestValidOptions(std::vector<const char *> args, Options *options) {
  TestOptions(true, args, options);
}

void TestValidOptions(std::vector<const char *> args) {
  TestOptions(true, args);
}

void TestInvalidOptions(std::vector<const char *> args) {
  TestOptions(false, args);
}

TEST(OptionsTest, UnknownOption_Invalid) {
  TestInvalidOptions({"-x"});
}

TEST(OptionsTest, MissingArgs_Invalid) {
  TestInvalidOptions({"-d", "-u"});
}

TEST(OptionsTest, Empty_ValidDefault) {
  Options options;
  TestValidOptions({}, &options);
  EXPECT_FALSE(options.usage);
  EXPECT_FALSE(options.verbose);
  EXPECT_TRUE(options.global);
  EXPECT_EQ(http::Url("any.speed.gfsvc.com"), options.global_host);
  EXPECT_FALSE(options.disable_dns_cache);
  EXPECT_EQ(0, options.max_connections);
  EXPECT_EQ(0, options.progress_millis);
  EXPECT_FALSE(options.exponential_moving_average);

  EXPECT_EQ(0, options.num_downloads);
  EXPECT_EQ(0, options.download_size);
  EXPECT_EQ(0, options.num_uploads);
  EXPECT_EQ(0, options.upload_size);
  EXPECT_EQ(0, options.min_transfer_runtime);
  EXPECT_EQ(0, options.max_transfer_runtime);
  EXPECT_EQ(0, options.min_transfer_intervals);
  EXPECT_EQ(0, options.max_transfer_intervals);
  EXPECT_EQ(0, options.max_transfer_variance);
  EXPECT_EQ(0, options.interval_millis);
  EXPECT_EQ(0, options.ping_runtime);
  EXPECT_EQ(0, options.ping_timeout);
  EXPECT_THAT(options.hosts, testing::IsEmpty());
}

TEST(OptionsTest, Usage_Valid) {
  {
    Options options;
    TestValidOptions({"-h"}, &options);
    EXPECT_TRUE(options.usage);
  }
  {
    Options options;
    TestValidOptions({"--help"}, &options);
    EXPECT_TRUE(options.usage);
  }
}

TEST(OptionsTest, OneHost_Valid) {
  Options options;
  TestValidOptions({"efgh"}, &options);
  EXPECT_THAT(options.hosts, testing::ElementsAre(http::Url("efgh")));
}

TEST(OptionsTest, ShortOptions_Valid) {
  Options options;
  TestValidOptions({"-v",
                    "-s", "5122",
                    "-t", "7653",
                    "-d", "20",
                    "-u", "15",
                    "-p", "500",
                    "-g", "speed.gfsvc.com",
                    "-a", "CrOS",
                    "foo.speed.googlefiber.net",
                    "bar.speed.googlefiber.net"},
                    &options);
  EXPECT_TRUE(options.verbose);
  EXPECT_EQ(20, options.num_downloads);
  EXPECT_EQ(5122, options.download_size);
  EXPECT_EQ(15, options.num_uploads);
  EXPECT_EQ(7653, options.upload_size);
  EXPECT_EQ(500, options.progress_millis);
  EXPECT_FALSE(options.global);
  EXPECT_EQ(http::Url("speed.gfsvc.com"), options.global_host);
  EXPECT_EQ("CrOS", options.user_agent);

  EXPECT_EQ(0, options.max_connections);
  EXPECT_FALSE(options.disable_dns_cache);
  EXPECT_FALSE(options.exponential_moving_average);
  EXPECT_EQ(0, options.min_transfer_runtime);
  EXPECT_EQ(0, options.max_transfer_runtime);
  EXPECT_EQ(0, options.min_transfer_intervals);
  EXPECT_EQ(0, options.max_transfer_intervals);
  EXPECT_EQ(0, options.max_transfer_variance);
  EXPECT_EQ(0, options.interval_millis);
  EXPECT_EQ(0, options.ping_runtime);
  EXPECT_EQ(0, options.ping_timeout);

  EXPECT_THAT(options.hosts, testing::UnorderedElementsAre(
      http::Url("foo.speed.googlefiber.net"),
      http::Url("bar.speed.googlefiber.net")));
}

TEST(OptionsTest, LongOptions_Valid) {
  Options options;
  TestValidOptions({"--verbose",
                    "--global_host", "speed.gfsvc.com",
                    "--user_agent", "CrOS",
                    "--progress_millis", "1000",
                    "--disable_dns_cache",
                    "--max_connections", "23",
                    "--exponential_moving_average",
                    "--num_downloads", "16",
                    "--download_size", "5122",
                    "--num_uploads", "12",
                    "--upload_size", "7653",
                    "--min_transfer_runtime", "7500",
                    "--max_transfer_runtime", "13500",
                    "--min_transfer_intervals", "13",
                    "--max_transfer_intervals", "22",
                    "--max_transfer_variance", "0.12",
                    "--interval_millis", "250",
                    "--ping_runtime", "2500",
                    "--ping_timeout", "300",
                    "foo.speed.googlefiber.net",
                    "bar.speed.googlefiber.net"},
                    &options);
  EXPECT_TRUE(options.verbose);
  EXPECT_FALSE(options.global);
  EXPECT_EQ(http::Url("speed.gfsvc.com"), options.global_host);
  EXPECT_EQ("CrOS", options.user_agent);
  EXPECT_EQ(1000, options.progress_millis);
  EXPECT_TRUE(options.disable_dns_cache);
  EXPECT_EQ(23, options.max_connections);
  EXPECT_TRUE(options.exponential_moving_average);
  EXPECT_EQ(16, options.num_downloads);
  EXPECT_EQ(5122, options.download_size);
  EXPECT_EQ(12, options.num_uploads);
  EXPECT_EQ(7653, options.upload_size);
  EXPECT_EQ("CrOS", options.user_agent);
  EXPECT_EQ(7500, options.min_transfer_runtime);
  EXPECT_EQ(13500, options.max_transfer_runtime);
  EXPECT_EQ(13, options.min_transfer_intervals);
  EXPECT_EQ(22, options.max_transfer_intervals);
  EXPECT_EQ(0.12, options.max_transfer_variance);
  EXPECT_EQ(250, options.interval_millis);
  EXPECT_EQ(2500, options.ping_runtime);
  EXPECT_EQ(300, options.ping_timeout);
  EXPECT_THAT(options.hosts, testing::UnorderedElementsAre(
      http::Url("foo.speed.googlefiber.net"),
      http::Url("bar.speed.googlefiber.net")));
}

}  // namespace
}  // namespace speedtest