// Copyright 2023 The Centipede Authors.
//
// 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
//
//      https://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 "./centipede/runner_flags.h"

#include "gtest/gtest.h"

namespace centipede {

namespace {

TEST(RunnerFlags, Empty) {
  RunnerFlags runner_flags("");
  EXPECT_TRUE(runner_flags.ToString().empty());
}

TEST(RunnerFlags, Malformed) {
  RunnerFlags runner_flags("a=x:b=:=c");
  EXPECT_EQ(runner_flags.ToString(), ":a=x:b:");
}

TEST(RunnerFlags, HasFlag) {
  RunnerFlags runner_flags(":a:b=x:");

  EXPECT_TRUE(runner_flags.HasFlag("a"));
  EXPECT_TRUE(runner_flags.HasFlag("b"));
  EXPECT_FALSE(runner_flags.HasFlag("c"));
}

TEST(RunnerFlags, GetFlagValue) {
  RunnerFlags runner_flags(":a=x:b:");

  EXPECT_EQ(runner_flags.GetFlagValue("a"), "x");
  EXPECT_TRUE(runner_flags.GetFlagValue("b").empty());
  EXPECT_TRUE(runner_flags.GetFlagValue("c").empty());
}

TEST(RunnerFlags, RepeatedFlag) {
  RunnerFlags runner_flags(":a=x:a=y:");
  EXPECT_EQ(runner_flags.GetFlagValue("a"), "y");
}

TEST(RunnerFlags, SetFlagValue) {
  RunnerFlags runner_flags(":a=x:b:");

  // Multiple updates.
  runner_flags.SetFlagValue("a", "red");
  EXPECT_EQ(runner_flags.GetFlagValue("a"), "red");
  runner_flags.SetFlagValue("a", "green");
  EXPECT_EQ(runner_flags.GetFlagValue("a"), "green");

  // Changing a flag to valueless.
  runner_flags.SetFlagValue("a", "");
  EXPECT_TRUE(runner_flags.GetFlagValue("a").empty());

  // Adding value to an existing flag without value.
  runner_flags.SetFlagValue("b", "yellow");
  EXPECT_EQ(runner_flags.GetFlagValue("b"), "yellow");

  // Adding a new flag.
  runner_flags.SetFlagValue("c", "blue");
  EXPECT_EQ(runner_flags.GetFlagValue("c"), "blue");
}

TEST(RunnerFlags, ToString) {
  RunnerFlags runner_flags(":b=x:a:c=y:");
  EXPECT_EQ(runner_flags.ToString(), ":b=x:a:c=y:");
}
}  // namespace
}  // namespace centipede