#include "libcockatrice/utility/svg_recolor.h"

#include <QByteArray>
#include <gtest/gtest.h>

namespace
{

TEST(SvgRecolor, ReplacesEverySentinel)
{
    const QByteArray svg = R"(<svg><path fill="currentColor"/><path stroke="currentColor"/></svg>)";
    const QByteArray result = resolveSvgColorSentinel(svg, "#ff8800");

    EXPECT_EQ(result.count("#ff8800"), 2);
    EXPECT_FALSE(result.contains("currentColor"));
}

TEST(SvgRecolor, ReplacesSentinelRegardlessOfCase)
{
    const QByteArray svg = R"(<svg><path fill="CurrentColor"/><path stroke="CURRENTCOLOR"/></svg>)";

    EXPECT_EQ(resolveSvgColorSentinel(svg, "#ff8800").count("#ff8800"), 2);
}

TEST(SvgRecolor, LeavesTheRestOfTheDocumentIntact)
{
    const QByteArray svg =
        R"(<svg viewBox="0 0 24 24"><!-- an ordinary comment --><path d="M0 0h24v24" fill="currentColor"/></svg>)";
    const QByteArray result = resolveSvgColorSentinel(svg, "#123456");

    EXPECT_TRUE(result.contains(R"(viewBox="0 0 24 24")"));
    EXPECT_TRUE(result.contains(R"(d="M0 0h24v24")"));
    EXPECT_TRUE(result.contains("an ordinary comment"));
}

TEST(SvgRecolor, DoesNotModifyTheInput)
{
    const QByteArray svg = R"(<svg><path fill="currentColor"/></svg>)";
    const QByteArray original = svg;

    resolveSvgColorSentinel(svg, "#000000");

    EXPECT_EQ(svg, original);
}

TEST(SvgRecolor, PreservesBytesThatAreNotValidUtf8)
{
    QByteArray svg = "<svg>";
    svg.append("\xff\xfe");
    svg.append(R"(<path fill="currentColor"/></svg>)");

    const QByteArray result = resolveSvgColorSentinel(svg, "#ff8800");

    EXPECT_TRUE(result.contains("\xff\xfe"));
    EXPECT_FALSE(result.contains("currentColor"));
    EXPECT_EQ(result.count("#ff8800"), 1);
}

TEST(SvgRecolor, NoSentinelIsANoOp)
{
    const QByteArray svg = R"(<svg><path fill="#abcdef"/></svg>)";

    EXPECT_EQ(resolveSvgColorSentinel(svg, "#ffffff"), svg);
}

TEST(SvgStrokeWidth, RewritesEveryAttribute)
{
    const QByteArray svg = R"(<svg stroke-width="2"><path stroke-width="1.5"/><path stroke-width='3'/></svg>)";

    const QByteArray result = resolveSvgStrokeWidth(svg, 4.25);

    EXPECT_EQ(result.count(R"(stroke-width="4.25")"), 2);
    EXPECT_EQ(result.count(R"(stroke-width='4.25')"), 1);
    EXPECT_FALSE(result.contains("1.5"));
    EXPECT_FALSE(result.contains("3'/"));
}

TEST(SvgStrokeWidth, LeavesCssDeclarationsAlone)
{
    const QByteArray svg = R"(<svg><path style="stroke-width:2"/><path stroke-width="2"/></svg>)";

    const QByteArray result = resolveSvgStrokeWidth(svg, 3.0);

    EXPECT_TRUE(result.contains(R"(style="stroke-width:2")"));
    EXPECT_EQ(result.count(R"(stroke-width="3.00")"), 1);
}

TEST(SvgStrokeWidth, NoAttributeIsANoOp)
{
    const QByteArray svg = R"(<svg stroke-linecap="round"><path d="M0 0h24"/></svg>)";

    EXPECT_EQ(resolveSvgStrokeWidth(svg, 3.0), svg);
}

TEST(SvgStrokeWidth, DoesNotModifyTheInputAndPreservesBytes)
{
    QByteArray svg = "<svg>";
    svg.append("\xff\xfe");
    svg.append(R"(<path stroke-width="2"/></svg>)");
    const QByteArray original = svg;

    const QByteArray result = resolveSvgStrokeWidth(svg, 3.5);

    EXPECT_EQ(svg, original);
    EXPECT_TRUE(result.contains("\xff\xfe"));
    EXPECT_TRUE(result.contains(R"(stroke-width="3.50")"));
}

TEST(SvgStrokeWidth, TruncatedAttributeTerminates)
{
    // A quote that is never closed must not over-read or hang.
    const QByteArray svg = R"(<svg><path stroke-width="2/></svg>)";

    const QByteArray result = resolveSvgStrokeWidth(svg, 3.0);

    EXPECT_TRUE(result.startsWith("<svg>"));
}

} // namespace