#include "gtest/gtest.h"
#include "proto_cxx_utils.h"

namespace ApiBooster {
namespace {

// Validate C++ to proto type name conversion.
TEST(ProtoCxxUtils, CxxToProtoType) {
  EXPECT_EQ("", ProtoCxxUtils::cxxToProtoType(""));
  EXPECT_EQ("foo", ProtoCxxUtils::cxxToProtoType("foo"));
  EXPECT_EQ("foo.bar", ProtoCxxUtils::cxxToProtoType("foo::bar"));
  EXPECT_EQ("foo.bar", ProtoCxxUtils::cxxToProtoType("foo::bar::FooCase"));
  EXPECT_EQ("foo.bar.Baz.Blah", ProtoCxxUtils::cxxToProtoType("foo::bar::Baz_Blah"));
}

// Validate proto to C++ type name conversion.
TEST(ProtoCxxUtils, ProtoToCxxType) {
  EXPECT_EQ("", ProtoCxxUtils::protoToCxxType("", false, false));
  EXPECT_EQ("", ProtoCxxUtils::protoToCxxType("", true, false));
  EXPECT_EQ("foo", ProtoCxxUtils::protoToCxxType("foo", false, false));
  EXPECT_EQ("foo", ProtoCxxUtils::protoToCxxType("foo", true, false));
  EXPECT_EQ("bar", ProtoCxxUtils::protoToCxxType("foo.bar", false, false));
  EXPECT_EQ("foo::bar", ProtoCxxUtils::protoToCxxType("foo.bar", true, false));
  EXPECT_EQ("foo::Bar", ProtoCxxUtils::protoToCxxType("foo.Bar", true, false));
  EXPECT_EQ("foo", ProtoCxxUtils::protoToCxxType("foo.Bar", true, true));
  EXPECT_EQ("foo::Bar::Baz", ProtoCxxUtils::protoToCxxType("foo.Bar.Baz", true, false));
  EXPECT_EQ("foo::Bar::Baz::Blah", ProtoCxxUtils::protoToCxxType("foo.Bar.Baz.Blah", true, false));
  EXPECT_EQ("foo::Bar::Baz", ProtoCxxUtils::protoToCxxType("foo.Bar.Baz.Blah", true, true));
}

// Validate proto field accessor upgrades.
TEST(ProtoCxxUtils, RenameMethod) {
  const absl::node_hash_map<std::string, std::string> renames = {
      {"foo", "bar"},
      {"bar", "baz"},
  };
  EXPECT_EQ(absl::nullopt, ProtoCxxUtils::renameMethod("whatevs", renames));
  EXPECT_EQ("bar", ProtoCxxUtils::renameMethod("foo", renames));
  EXPECT_EQ("baz", ProtoCxxUtils::renameMethod("bar", renames));

  EXPECT_EQ("clear_bar", ProtoCxxUtils::renameMethod("clear_foo", renames));
  EXPECT_EQ("set_bar", ProtoCxxUtils::renameMethod("set_foo", renames));
  EXPECT_EQ("has_bar", ProtoCxxUtils::renameMethod("has_foo", renames));
  EXPECT_EQ("mutable_bar", ProtoCxxUtils::renameMethod("mutable_foo", renames));
  EXPECT_EQ("set_allocated_bar", ProtoCxxUtils::renameMethod("set_allocated_foo", renames));
  EXPECT_EQ("release_bar", ProtoCxxUtils::renameMethod("release_foo", renames));
  EXPECT_EQ("add_bar", ProtoCxxUtils::renameMethod("add_foo", renames));
  EXPECT_EQ("bar_size", ProtoCxxUtils::renameMethod("foo_size", renames));
}

// Validate proto constant upgrades.
TEST(ProtoCxxUtils, RenameConstant) {
  const absl::node_hash_map<std::string, std::string> renames = {
      {"foo_bar", "bar_foo"},
      {"foo_baz", "baz"},
  };
  EXPECT_EQ(absl::nullopt, ProtoCxxUtils::renameConstant("whatevs", renames));
  EXPECT_EQ("kBarFoo", ProtoCxxUtils::renameConstant("kFooBar", renames));
  EXPECT_EQ("kBaz", ProtoCxxUtils::renameConstant("kFooBaz", renames));
}

// Validate proto enum value upgrades.
TEST(ProtoCxxUtils, RenameEnumValue) {
  const absl::node_hash_map<std::string, std::string> renames = {
      {"FOO_BAR", "BAR_FOO"},
  };
  EXPECT_EQ(absl::nullopt, ProtoCxxUtils::renameEnumValue("FOO_BAZ", renames));
  EXPECT_EQ("BAR_FOO", ProtoCxxUtils::renameEnumValue("FOO_BAR", renames));
}

} // namespace
} // namespace ApiBooster