#include <library/cpp/testing/gtest/gtest.h>

#include <library/cpp/yt/string/string.h>

namespace NYT {
namespace {

////////////////////////////////////////////////////////////////////////////////

struct TTestCase
{
    const char* UnderCase;
    const char* CamelCase;
};

static std::vector<TTestCase> TestCases {
    { "kenny", "Kenny" },
    { "south_park", "SouthPark" },
    { "a", "A" },
    { "a_b_c", "ABC" },
    { "reed_solomon_6_3", "ReedSolomon_6_3" },
    { "lrc_12_2_2", "Lrc_12_2_2" },
    { "0", "0" },
    { "0_1_2", "0_1_2" },
    { "int64", "Int64" }
};

////////////////////////////////////////////////////////////////////////////////

TEST(TStringTest, UnderscoreCaseToCamelCase)
{
    for (const auto& testCase : TestCases) {
        auto result = UnderscoreCaseToCamelCase(testCase.UnderCase);
        EXPECT_STREQ(testCase.CamelCase, result.c_str())
            << "Original: \"" << testCase.UnderCase << '"';
    }
}

TEST(TStringTest, CamelCaseToUnderscoreCase)
{
    for (const auto& testCase : TestCases) {
        auto result = CamelCaseToUnderscoreCase(testCase.CamelCase);
        EXPECT_STREQ(testCase.UnderCase, result.c_str())
            << "Original: \"" << testCase.CamelCase << '"';
    }
}

TEST(TStringTest, TrimWhitespaces)
{
    EXPECT_EQ("", TrimLeadingWhitespaces(""));
    EXPECT_EQ("", TrimLeadingWhitespaces("   "));
    EXPECT_EQ("foo", TrimLeadingWhitespaces("   foo"));
    EXPECT_EQ("foo ", TrimLeadingWhitespaces("   foo "));
    EXPECT_EQ("f oo ", TrimLeadingWhitespaces("   f oo "));

    EXPECT_EQ("", Trim("", ""));
    EXPECT_EQ("", Trim(" ", " "));
    EXPECT_EQ(" ", Trim(" ", "\t"));
    EXPECT_EQ("", Trim(" \t", " \t"));
    EXPECT_EQ("foo", Trim(" foo ", " "));
    EXPECT_EQ("f", Trim(" f ", " "));
}

////////////////////////////////////////////////////////////////////////////////

} // namespace
} // namespace NYT