clang-tools-extra/clangd/unittests/tweaks/ScopifyEnumTests.cpp

//===-- DefineOutline.cpp ---------------------------------------*- C++ -*-===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//

#include "TweakTesting.h"
#include "gtest/gtest.h"

namespace clang::clangd {
namespace {

TWEAK_TEST(ScopifyEnum);

TEST_F(ScopifyEnumTest, TriggersOnUnscopedEnumDecl) {
  FileName = "Test.hpp";
  // Not available for scoped enum.
  EXPECT_UNAVAILABLE(R"cpp(enum class ^E { V };)cpp");

  // Not available for non-definition.
  EXPECT_UNAVAILABLE(R"cpp(
enum E { V };
enum ^E;
)cpp");
}

TEST_F(ScopifyEnumTest, ApplyTestWithPrefix) {
  std::string Original = R"cpp(
enum ^E { EV1, EV2, EV3 };
enum E;
E func(E in)
{
  E out = EV1;
  if (in == EV2)
    out = E::EV3;
  return out;
}
)cpp";
  std::string Expected = R"cpp(
enum class E { V1, V2, V3 };
enum class E;
E func(E in)
{
  E out = E::V1;
  if (in == E::V2)
    out = E::V3;
  return out;
}
)cpp";
  FileName = "Test.cpp";
  SCOPED_TRACE(Original);
  EXPECT_EQ(apply(Original), Expected);
}

TEST_F(ScopifyEnumTest, ApplyTestWithPrefixAndUnderscore) {
  std::string Original = R"cpp(
enum ^E { E_V1, E_V2, E_V3 };
enum E;
E func(E in)
{
  E out = E_V1;
  if (in == E_V2)
    out = E::E_V3;
  return out;
}
)cpp";
  std::string Expected = R"cpp(
enum class E { V1, V2, V3 };
enum class E;
E func(E in)
{
  E out = E::V1;
  if (in == E::V2)
    out = E::V3;
  return out;
}
)cpp";
  FileName = "Test.cpp";
  SCOPED_TRACE(Original);
  EXPECT_EQ(apply(Original), Expected);
}

TEST_F(ScopifyEnumTest, ApplyTestWithoutPrefix) {
  std::string Original = R"cpp(
enum ^E { V1, V2, V3 };
enum E;
E func(E in)
{
  E out = V1;
  if (in == V2)
    out = E::V3;
  return out;
}
)cpp";
  std::string Expected = R"cpp(
enum class E { V1, V2, V3 };
enum class E;
E func(E in)
{
  E out = E::V1;
  if (in == E::V2)
    out = E::V3;
  return out;
}
)cpp";
  FileName = "Test.cpp";
  SCOPED_TRACE(Original);
  EXPECT_EQ(apply(Original), Expected);
}

} // namespace
} // namespace clang::clangd