// Copyright 2023 PingCAP, Inc.
//
// 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 <Common/FieldVisitors.h>
#include <Common/RedactHelpers.h>
#include <TestUtils/TiFlashTestBasic.h>
#include <common/types.h>

#include <sstream>

namespace DB::tests
{

TEST(RedactLogTest, Basic)
{
    const char * test_key = "\x01\x0a\xff";
    const size_t key_sz = strlen(test_key);

    const /*DB::HandleID*/ Int64 test_handle = 10009;

    Redact::setRedactLog(RedactMode::Disable);
    EXPECT_EQ(Redact::keyToDebugString(test_key, key_sz), "010AFF");
    EXPECT_EQ(Redact::keyToHexString(test_key, key_sz), "010AFF");
    EXPECT_EQ(Redact::handleToDebugString(test_handle), "10009");
    std::stringstream ss;
    Redact::keyToDebugString(test_key, key_sz, ss);
    EXPECT_EQ(ss.str(), "010AFF");

    Redact::setRedactLog(RedactMode::Marker);
    EXPECT_EQ(Redact::keyToDebugString(test_key, key_sz), "‹010AFF›");
    EXPECT_EQ(Redact::keyToHexString(test_key, key_sz), "010AFF");
    EXPECT_EQ(Redact::handleToDebugString(test_handle), "‹10009›");
    ss.str("");
    Redact::keyToDebugString(test_key, key_sz, ss);
    EXPECT_EQ(ss.str(), "‹010AFF›");

    Redact::setRedactLog(RedactMode::Enable);
    EXPECT_EQ(Redact::keyToDebugString(test_key, key_sz), "?");
    EXPECT_EQ(Redact::keyToHexString(test_key, key_sz), "010AFF"); // Unaffected by readact-log status
    EXPECT_EQ(Redact::handleToDebugString(test_handle), "?");
    ss.str("");
    Redact::keyToDebugString(test_key, key_sz, ss);
    EXPECT_EQ(ss.str(), "?");

    Redact::setRedactLog(RedactMode::Disable); // restore flags
}

TEST(RedactLogTest, ToMarkerString)
{
    for (const auto & [input, expect] : //
         std::vector<std::pair<String, String>>{
             {"", "‹›"},
             {"abcdefg", "‹abcdefg›"},
             {"中文", "‹中文›"},
         })
    {
        EXPECT_EQ(Redact::toMarkerString(input, true), expect) << input;
    }

    for (const auto & [input, expect] : //
         std::vector<std::pair<String, String>>{
             {"plain text", "‹plain text›"},
             {"‹›", "‹‹‹›››"},
             {"abc‹›de‹github›fg", "‹abc‹‹››de‹‹github››fg›"},
             {"abc‹", "‹abc‹‹›"},
             {"abc›def", "‹abc››def›"},
             {"abc‹github", "‹abc‹‹github›"},
             {"测试‹中文", "‹测试‹‹中文›"},
         })
    {
        EXPECT_EQ(Redact::toMarkerString(input, false), expect) << input;
    }
}

TEST(RedactLogTest, FieldVisitorToDebugStringTest)
{
    std::vector<std::pair<Field, String>> fields{
        {Field(), "‹NULL›"},
        {Field(static_cast<Int64>(-65536)), "‹-65536›"},
        {Field(static_cast<UInt64>(65536)), "‹65536›"},
        {Field(3.1415926), "‹3.1415926›"},
        {Field(String("plain text")), "‹'plain text'›"},
        {Field(String("中文‹测试")), "‹'中文‹‹测试'›"},
        {Field(String("abc‹›de‹github›fg")), "‹'abc‹‹››de‹‹github››fg'›"},
    };

    Redact::setRedactLog(RedactMode::Marker);
    for (const auto & [input, expect] : fields)
    {
        auto output = applyVisitor(FieldVisitorToDebugString(), input);
        EXPECT_EQ(output, expect);
    }

    Redact::setRedactLog(RedactMode::Disable);
}

} // namespace DB::tests