| // Copyright 2023 The Dawn & Tint Authors |
| // |
| // Redistribution and use in source and binary forms, with or without |
| // modification, are permitted provided that the following conditions are met: |
| // |
| // 1. Redistributions of source code must retain the above copyright notice, this |
| // list of conditions and the following disclaimer. |
| // |
| // 2. Redistributions in binary form must reproduce the above copyright notice, |
| // this list of conditions and the following disclaimer in the documentation |
| // and/or other materials provided with the distribution. |
| // |
| // 3. Neither the name of the copyright holder nor the names of its |
| // contributors may be used to endorse or promote products derived from |
| // this software without specific prior written permission. |
| // |
| // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" |
| // AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE |
| // IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE |
| // DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE |
| // FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL |
| // DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR |
| // SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER |
| // CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, |
| // OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
| // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
| |
| #include "src/tint/utils/text/base64.h" |
| |
| #include <optional> |
| #include <vector> |
| |
| #include "gtest/gtest.h" |
| |
| #include "src/tint/utils/containers/transform.h" |
| #include "src/tint/utils/text/string.h" |
| |
| namespace tint::utils { |
| namespace { |
| |
| struct DecodeBase64Case { |
| char in; |
| std::optional<uint8_t> out; |
| }; |
| |
| using DecodeBase64Test = testing::TestWithParam<DecodeBase64Case>; |
| |
| TEST_P(DecodeBase64Test, Char) { |
| EXPECT_EQ(DecodeBase64(GetParam().in), GetParam().out); |
| } |
| |
| INSTANTIATE_TEST_SUITE_P(Valid, |
| DecodeBase64Test, |
| testing::Values(DecodeBase64Case{'A', 0}, |
| DecodeBase64Case{'B', 1}, |
| DecodeBase64Case{'C', 2}, |
| DecodeBase64Case{'D', 3}, |
| DecodeBase64Case{'E', 4}, |
| DecodeBase64Case{'F', 5}, |
| DecodeBase64Case{'G', 6}, |
| DecodeBase64Case{'H', 7}, |
| DecodeBase64Case{'I', 8}, |
| DecodeBase64Case{'J', 9}, |
| DecodeBase64Case{'K', 10}, |
| DecodeBase64Case{'L', 11}, |
| DecodeBase64Case{'M', 12}, |
| DecodeBase64Case{'N', 13}, |
| DecodeBase64Case{'O', 14}, |
| DecodeBase64Case{'P', 15}, |
| DecodeBase64Case{'Q', 16}, |
| DecodeBase64Case{'R', 17}, |
| DecodeBase64Case{'S', 18}, |
| DecodeBase64Case{'T', 19}, |
| DecodeBase64Case{'U', 20}, |
| DecodeBase64Case{'V', 21}, |
| DecodeBase64Case{'W', 22}, |
| DecodeBase64Case{'X', 23}, |
| DecodeBase64Case{'Y', 24}, |
| DecodeBase64Case{'Z', 25}, |
| DecodeBase64Case{'a', 26}, |
| DecodeBase64Case{'b', 27}, |
| DecodeBase64Case{'c', 28}, |
| DecodeBase64Case{'d', 29}, |
| DecodeBase64Case{'e', 30}, |
| DecodeBase64Case{'f', 31}, |
| DecodeBase64Case{'g', 32}, |
| DecodeBase64Case{'h', 33}, |
| DecodeBase64Case{'i', 34}, |
| DecodeBase64Case{'j', 35}, |
| DecodeBase64Case{'k', 36}, |
| DecodeBase64Case{'l', 37}, |
| DecodeBase64Case{'m', 38}, |
| DecodeBase64Case{'n', 39}, |
| DecodeBase64Case{'o', 40}, |
| DecodeBase64Case{'p', 41}, |
| DecodeBase64Case{'q', 42}, |
| DecodeBase64Case{'r', 43}, |
| DecodeBase64Case{'s', 44}, |
| DecodeBase64Case{'t', 45}, |
| DecodeBase64Case{'u', 46}, |
| DecodeBase64Case{'v', 47}, |
| DecodeBase64Case{'w', 48}, |
| DecodeBase64Case{'x', 49}, |
| DecodeBase64Case{'y', 50}, |
| DecodeBase64Case{'z', 51}, |
| DecodeBase64Case{'0', 52}, |
| DecodeBase64Case{'1', 53}, |
| DecodeBase64Case{'2', 54}, |
| DecodeBase64Case{'3', 55}, |
| DecodeBase64Case{'4', 56}, |
| DecodeBase64Case{'5', 57}, |
| DecodeBase64Case{'6', 58}, |
| DecodeBase64Case{'7', 59}, |
| DecodeBase64Case{'8', 60}, |
| DecodeBase64Case{'9', 61}, |
| DecodeBase64Case{'+', 62}, |
| DecodeBase64Case{'/', 63})); |
| |
| INSTANTIATE_TEST_SUITE_P(Invalid, |
| DecodeBase64Test, |
| testing::Values(DecodeBase64Case{'@', std::nullopt}, |
| DecodeBase64Case{'#', std::nullopt}, |
| DecodeBase64Case{'^', std::nullopt}, |
| DecodeBase64Case{'&', std::nullopt}, |
| DecodeBase64Case{'!', std::nullopt}, |
| DecodeBase64Case{'*', std::nullopt}, |
| DecodeBase64Case{'(', std::nullopt}, |
| DecodeBase64Case{')', std::nullopt}, |
| DecodeBase64Case{'-', std::nullopt}, |
| DecodeBase64Case{'.', std::nullopt}, |
| DecodeBase64Case{'\0', std::nullopt}, |
| DecodeBase64Case{'\n', std::nullopt})); |
| |
| INSTANTIATE_TEST_SUITE_P(Padding, |
| DecodeBase64Test, |
| testing::Values(DecodeBase64Case{'=', std::nullopt})); |
| |
| struct DecodeBase64FromCommentsCase { |
| std::string_view wgsl; |
| Vector<int, 0> expected; |
| }; |
| |
| static std::ostream& operator<<(std::ostream& o, const DecodeBase64FromCommentsCase& c) { |
| return o << "'" << ReplaceAll(c.wgsl, "\n", "") << "'"; |
| } |
| |
| using DecodeBase64FromCommentsTest = ::testing::TestWithParam<DecodeBase64FromCommentsCase>; |
| |
| TEST_P(DecodeBase64FromCommentsTest, None) { |
| auto got_bytes = DecodeBase64FromComments(GetParam().wgsl); |
| auto got = Transform(got_bytes, [](std::byte byte) { return static_cast<int>(byte); }); |
| EXPECT_EQ(got, GetParam().expected); |
| } |
| |
| INSTANTIATE_TEST_SUITE_P(, |
| DecodeBase64FromCommentsTest, |
| testing::ValuesIn(std::vector<DecodeBase64FromCommentsCase>{ |
| {"", Empty}, |
| {"//", Empty}, |
| {"abc", Empty}, |
| {"abc//", Empty}, |
| { |
| R"(a |
| b |
| c)", |
| Empty, |
| }, |
| {"// abc", {26, 27, 28}}, |
| {"a // bc", {27, 28}}, |
| {"ab // c", {28}}, |
| {"// a.b.c", {26, 27, 28}}, |
| { |
| R"(a |
| b |
| c)", |
| Empty, |
| }, |
| { |
| R"(a |
| // b |
| c)", |
| {27}, |
| }, |
| { |
| R"(// a |
| // b |
| // c)", |
| {26, 27, 28}, |
| }, |
| { |
| R"(/* a |
| b |
| c |
| */)", |
| {26, 27, 28}, |
| }, |
| { |
| R"(/* a |
| b |
| */ |
| c)", |
| {26, 27}, |
| }, |
| { |
| R"(a/* |
| b |
| */ |
| c)", |
| {27}, |
| }, |
| { |
| "x/*a*/b/*c*/y", |
| {26, 28}, |
| }, |
| { |
| "x/*a/*b*/c*/z", |
| {26, 27, 28}, |
| }, |
| })); |
| |
| } // namespace |
| } // namespace tint::utils |
