/*
 * Copyright (c) 2020, the SerenityOS developers.
 *
 * SPDX-License-Identifier: BSD-2-Clause
 */

#include <LibTest/TestCase.h>

#include "../CSV.h"
#include <AK/MemoryStream.h>

TEST_CASE(can_write)
{
    Vector<Vector<int>> data = {
        { 1, 2, 3 },
        { 4, 5, 6 },
        { 7, 8, 9 },
    };

    AllocatingMemoryStream stream;
    MUST(Writer::CSV::generate(stream, data));

    auto expected_output = R"~(1,2,3
4,5,6
7,8,9
)~"sv;

    auto buffer = MUST(stream.read_until_eof());
    EXPECT_EQ(StringView { buffer.bytes() }, expected_output);
}

TEST_CASE(can_write_with_header)
{
    Vector<Vector<int>> data = {
        { 1, 2, 3 },
        { 4, 5, 6 },
        { 7, 8, 9 },
    };

    AllocatingMemoryStream stream;
    MUST(Writer::CSV::generate(stream, data, { "A"sv, "B\""sv, "C"sv }));

    auto expected_output = R"~(A,"B""",C
1,2,3
4,5,6
7,8,9
)~"sv;

    auto buffer = MUST(stream.read_until_eof());
    EXPECT_EQ(StringView { buffer.bytes() }, expected_output);
}

TEST_CASE(can_write_with_different_behaviors)
{
    Vector<Vector<ByteString>> data = {
        { "Well", "Hello\"", "Friends" },
        { "We\"ll", "Hello,", "   Friends" },
    };

    AllocatingMemoryStream stream;
    MUST(Writer::CSV::generate(stream, data, { "A"sv, "B\""sv, "C"sv }, Writer::WriterBehavior::QuoteOnlyInFieldStart | Writer::WriterBehavior::WriteHeaders));

    auto expected_output = R"~(A,B",C
Well,Hello",Friends
We"ll,"Hello,",   Friends
)~"sv;

    auto buffer = MUST(stream.read_until_eof());
    EXPECT_EQ(StringView { buffer.bytes() }, expected_output);
}