1
0
Fork 0
mirror of https://github.com/LadybirdBrowser/ladybird.git synced 2025-06-09 17:44:56 +09:00
ladybird/Userland/Applications/Spreadsheet/Writers/Test/TestXSVWriter.cpp
2023-01-29 19:16:44 -07:00

70 lines
1.6 KiB
C++

/*
* 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<DeprecatedString>> 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);
}