| // Copyright 2024 The Pigweed Authors |
| // |
| // 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 |
| // |
| // https://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 "pw_tokenizer_private/csv.h" |
| |
| #include "pw_unit_test/framework.h" |
| |
| namespace { |
| |
| using ::pw::tokenizer::ParseCsv; |
| |
| TEST(CsvParser, ReadFile) { |
| static constexpr const char kCsv[] = |
| "abc,def,ghi\n" |
| "\"\",\"\"\"\",\n" |
| "123,\"\",\"4\""; |
| |
| const auto result = ParseCsv(kCsv); |
| EXPECT_EQ(result.size(), 3u); |
| |
| EXPECT_EQ(result[0][0], "abc"); |
| EXPECT_EQ(result[0][1], "def"); |
| EXPECT_EQ(result[0][2], "ghi"); |
| EXPECT_EQ(result[1][0], ""); |
| EXPECT_EQ(result[1][1], "\""); |
| EXPECT_EQ(result[1][2], ""); |
| EXPECT_EQ(result[2][0], "123"); |
| EXPECT_EQ(result[2][1], ""); |
| EXPECT_EQ(result[2][2], "4"); |
| } |
| |
| TEST(CsvParser, EmptyLines) { |
| static constexpr const char kCsv[] = "\n\n\r\n\r\n"; |
| const auto result = ParseCsv(kCsv); |
| EXPECT_EQ(result.size(), 0u); |
| } |
| |
| TEST(CsvParser, EmptyLinesWithTextInterspersed) { |
| static constexpr const char kCsv[] = "\n\n\r \n\r\n\r\n\r,\r\n"; |
| const auto result = ParseCsv(kCsv); |
| ASSERT_EQ(result.size(), 2u); |
| ASSERT_EQ(result[0].size(), 1u); |
| EXPECT_EQ(result[0][0], " "); |
| |
| ASSERT_EQ(result[1].size(), 2u); |
| EXPECT_EQ(result[1][0], ""); |
| EXPECT_EQ(result[1][1], ""); |
| } |
| |
| TEST(CsvParser, VaryingColumns) { |
| static constexpr const char kCsv[] = |
| "\n" |
| "a\r\n" |
| "b\r\n" |
| ",\r\n" |
| "c,d,,\r\n" |
| " , ,\"\n\"\n" |
| "e,fg,hijk,lmno "; |
| |
| const std::vector<std::vector<std::string>> kExpected = { |
| {"a"}, |
| {"b"}, |
| {"", ""}, |
| {"c", "d", "", ""}, |
| {" ", " ", "\n"}, |
| {"e", "fg", "hijk", "lmno "}}; |
| const auto result = ParseCsv(kCsv); |
| ASSERT_EQ(result.size(), 6u); |
| EXPECT_EQ(kExpected, result); |
| } |
| |
| TEST(CsvParser, Error_NoLines) { |
| const auto result = ParseCsv(R"(11,"abc"., 13 )"); |
| ASSERT_EQ(result.size(), 0u); |
| } |
| |
| TEST(CsvParser, Error_SkipsOnlyErrors) { |
| const auto result = ParseCsv( |
| "a,b,c\n" |
| "1,\"2\".,3\n" |
| "d,e\r\n" |
| "\"456\n\r\" 789\r\n\r\n" |
| "f,g,h\n" |
| "\"0"); |
| const std::vector<std::vector<std::string>> kExpected = { |
| {"a", "b", "c"}, |
| {"d", "e"}, |
| {"f", "g", "h"}, |
| }; |
| EXPECT_EQ(result, kExpected); |
| } |
| |
| } // namespace |