mirror-chatterino2/tests/src/UtilTwitch.cpp

#include "Test.hpp"
#include "util/Twitch.hpp"

#include <QApplication>
#include <QDebug>
#include <QtConcurrent>

#include <chrono>
#include <thread>

using namespace chatterino;

TEST(UtilTwitch, StripUserName)
{
    struct TestCase {
        QString inputUserName;
        QString expectedUserName;
    };

    std::vector<TestCase> tests{
        {
            "pajlada",
            "pajlada",
        },
        {
            "Pajlada",
            "Pajlada",
        },
        {
            "@Pajlada",
            "Pajlada",
        },
        {
            "@Pajlada,",
            "Pajlada",
        },
        {
            "@@Pajlada,",
            "@Pajlada",
        },
        {
            "@@Pajlada,,",
            "@Pajlada,",
        },
        {
            "",
            "",
        },
        {
            "@",
            "",
        },
        {
            ",",
            "",
        },
        {
            // We purposefully don't handle spaces at the end, as all expected usages of this function split the message up by space and strip the parameters by themselves
            ", ",
            ", ",
        },
        {
            // We purposefully don't handle spaces at the start, as all expected usages of this function split the message up by space and strip the parameters by themselves
            " @",
            " @",
        },
    };

    for (const auto &[inputUserName, expectedUserName] : tests)
    {
        QString userName = inputUserName;
        stripUserName(userName);

        EXPECT_EQ(userName, expectedUserName)
            << userName << " (" << inputUserName
            << ") did not match expected value " << expectedUserName;
    }
}

TEST(UtilTwitch, StripChannelName)
{
    struct TestCase {
        QString inputChannelName;
        QString expectedChannelName;
    };

    std::vector<TestCase> tests{
        {
            "pajlada",
            "pajlada",
        },
        {
            "Pajlada",
            "Pajlada",
        },
        {
            "@Pajlada",
            "Pajlada",
        },
        {
            "#Pajlada",
            "Pajlada",
        },
        {
            "#Pajlada,",
            "Pajlada",
        },
        {
            "#Pajlada,",
            "Pajlada",
        },
        {
            "@@Pajlada,",
            "@Pajlada",
        },
        {
            // We only strip one character off the front
            "#@Pajlada,",
            "@Pajlada",
        },
        {
            "@@Pajlada,,",
            "@Pajlada,",
        },
        {
            "",
            "",
        },
        {
            "@",
            "",
        },
        {
            ",",
            "",
        },
        {
            // We purposefully don't handle spaces at the end, as all expected usages of this function split the message up by space and strip the parameters by themselves
            ", ",
            ", ",
        },
        {
            // We purposefully don't handle spaces at the start, as all expected usages of this function split the message up by space and strip the parameters by themselves
            " #",
            " #",
        },
    };

    for (const auto &[inputChannelName, expectedChannelName] : tests)
    {
        QString userName = inputChannelName;
        stripChannelName(userName);

        EXPECT_EQ(userName, expectedChannelName)
            << userName << " (" << inputChannelName
            << ") did not match expected value " << expectedChannelName;
    }
}

TEST(UtilTwitch, ParseUserNameOrID)
{
    struct TestCase {
        QString input;
        QString expectedUserName;
        QString expectedUserID;
    };

    std::vector<TestCase> tests{
        {
            "pajlada",
            "pajlada",
            {},
        },
        {
            "Pajlada",
            "Pajlada",
            {},
        },
        {
            "@Pajlada",
            "Pajlada",
            {},
        },
        {
            "#Pajlada",
            "Pajlada",
            {},
        },
        {
            "#Pajlada,",
            "Pajlada",
            {},
        },
        {
            "#Pajlada,",
            "Pajlada",
            {},
        },
        {
            "@@Pajlada,",
            "@Pajlada",
            {},
        },
        {
            // We only strip one character off the front
            "#@Pajlada,",
            "@Pajlada",
            {},
        },
        {
            "@@Pajlada,,",
            "@Pajlada,",
            {},
        },
        {
            "",
            "",
            {},
        },
        {
            "@",
            "",
            {},
        },
        {
            ",",
            "",
            {},
        },
        {
            // We purposefully don't handle spaces at the end, as all expected usages of this function split the message up by space and strip the parameters by themselves
            ", ",
            ", ",
            {},
        },
        {
            // We purposefully don't handle spaces at the start, as all expected usages of this function split the message up by space and strip the parameters by themselves
            " #",
            " #",
            {},
        },
        {
            "id:123",
            {},
            "123",
        },
        {
            "id:",
            {},
            "",
        },
    };

    for (const auto &[input, expectedUserName, expectedUserID] : tests)
    {
        auto [actualUserName, actualUserID] = parseUserNameOrID(input);

        EXPECT_EQ(actualUserName, expectedUserName)
            << "name " << actualUserName << " (" << input
            << ") did not match expected value " << expectedUserName;

        EXPECT_EQ(actualUserID, expectedUserID)
            << "id " << actualUserID << " (" << input
            << ") did not match expected value " << expectedUserID;
    }
}

TEST(UtilTwitch, UserLoginRegexp)
{
    struct TestCase {
        QString inputUserLogin;
        bool matches;
    };

    std::vector<TestCase> tests{
        {
            "pajlada",
            true,
        },
        {
            // Login names must not start with an underscore
            "_pajlada",
            false,
        },
        {
            "@pajlada",
            false,
        },
        {
            "pajlada!",
            false,
        },
        {
            "pajlada,",
            false,
        },
        {
            // Login names must not contain capital letters
            "Pajlada",
            false,
        },
        {"k", true},
        {"testaccount_420", true},
        {"3v", true},
        {"ron", true},
        {"bits", true},
    };

    const auto &regexp = twitchUserLoginRegexp();

    for (const auto &[inputUserLogin, expectedMatch] : tests)
    {
        const auto &match = regexp.match(inputUserLogin);
        auto actual = regexp.match(inputUserLogin);

        EXPECT_EQ(match.hasMatch(), expectedMatch)
            << inputUserLogin << " did not match as expected";
    }
}

TEST(UtilTwitch, UserNameRegexp)
{
    struct TestCase {
        QString inputUserLogin;
        bool matches;
    };

    std::vector<TestCase> tests{
        {"PAJLADA", true},
        {
            // User names must not start with an underscore
            "_pajlada",
            false,
        },
        {
            "@pajlada",
            false,
        },
        {
            "pajlada!",
            false,
        },
        {
            "pajlada,",
            false,
        },
        {
            "Pajlada",
            true,
        },
        {"k", true},
        {"testaccount_420", true},
        {"3v", true},
        {"ron", true},
        {"bits", true},
        {"3V", true},
        {"Ron", true},
        {"Bits", true},
    };

    const auto &regexp = twitchUserNameRegexp();

    for (const auto &[inputUserLogin, expectedMatch] : tests)
    {
        const auto &match = regexp.match(inputUserLogin);
        auto actual = regexp.match(inputUserLogin);

        EXPECT_EQ(match.hasMatch(), expectedMatch)
            << inputUserLogin << " did not match as expected";
    }
}

TEST(UtilTwitch, CleanHelixColor)
{
    struct TestCase {
        QString inputColor;
        QString expectedColor;
    };

    std::vector<TestCase> tests{
        {"foo", "foo"},
        {"BlueViolet", "blue_violet"},
        {"blueviolet", "blue_violet"},
        {"DODGERBLUE", "dodger_blue"},
        {"blUEviolet", "blue_violet"},
        {"caDEtblue", "cadet_blue"},
        {"doDGerblue", "dodger_blue"},
        {"goLDenrod", "golden_rod"},
        {"hoTPink", "hot_pink"},
        {"orANgered", "orange_red"},
        {"seAGreen", "sea_green"},
        {"spRInggreen", "spring_green"},
        {"yeLLowgreen", "yellow_green"},
        {"xDxD", "xdxd"},
    };

    for (const auto &[inputColor, expectedColor] : tests)
    {
        QString actualColor = inputColor;
        cleanHelixColorName(actualColor);

        EXPECT_EQ(actualColor, expectedColor)
            << inputColor << " cleaned up to " << actualColor << " instead of "
            << expectedColor;
    }
}