Alexei Frolov | bbbedff | 2020-03-11 14:21:38 -0700 | [diff] [blame] | 1 | // Copyright 2020 The Pigweed Authors |
| 2 | // |
| 3 | // Licensed under the Apache License, Version 2.0 (the "License"); you may not |
| 4 | // use this file except in compliance with the License. You may obtain a copy of |
| 5 | // the License at |
| 6 | // |
| 7 | // https://www.apache.org/licenses/LICENSE-2.0 |
| 8 | // |
| 9 | // Unless required by applicable law or agreed to in writing, software |
| 10 | // distributed under the License is distributed on an "AS IS" BASIS, WITHOUT |
| 11 | // WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the |
| 12 | // License for the specific language governing permissions and limitations under |
| 13 | // the License. |
| 14 | |
| 15 | #include "pw_protobuf/find.h" |
| 16 | |
| 17 | #include "gtest/gtest.h" |
| 18 | |
| 19 | namespace pw::protobuf { |
| 20 | namespace { |
| 21 | |
| 22 | // clang-format off |
| 23 | constexpr uint8_t encoded_proto[] = { |
| 24 | // type=int32, k=1, v=42 |
| 25 | 0x08, 0x2a, |
| 26 | // type=sint32, k=2, v=-13 |
| 27 | 0x10, 0x19, |
| 28 | // type=bool, k=3, v=false |
| 29 | 0x18, 0x00, |
| 30 | // type=double, k=4, v=3.14159 |
| 31 | 0x21, 0x6e, 0x86, 0x1b, 0xf0, 0xf9, 0x21, 0x09, 0x40, |
| 32 | // type=fixed32, k=5, v=0xdeadbeef |
| 33 | 0x2d, 0xef, 0xbe, 0xad, 0xde, |
| 34 | // type=string, k=6, v="Hello world" |
| 35 | 0x32, 0x0b, 'H', 'e', 'l', 'l', 'o', ' ', 'w', 'o', 'r', 'l', 'd', |
| 36 | |
| 37 | // type=message, k=7, len=2 |
| 38 | 0x3a, 0x02, |
| 39 | // (nested) type=uint32, k=1, v=3 |
| 40 | 0x08, 0x03 |
| 41 | }; |
| 42 | |
| 43 | TEST(FindDecodeHandler, SingleLevel_FindsExistingField) { |
Alexei Frolov | fe9723c | 2020-05-11 13:20:47 -0700 | [diff] [blame] | 44 | CallbackDecoder decoder; |
Alexei Frolov | bbbedff | 2020-03-11 14:21:38 -0700 | [diff] [blame] | 45 | FindDecodeHandler finder(3); |
| 46 | |
| 47 | decoder.set_handler(&finder); |
Wyatt Hepler | e2cbadf | 2020-06-22 11:21:45 -0700 | [diff] [blame] | 48 | decoder.Decode(std::as_bytes(std::span(encoded_proto))); |
Alexei Frolov | bbbedff | 2020-03-11 14:21:38 -0700 | [diff] [blame] | 49 | |
| 50 | EXPECT_TRUE(finder.found()); |
| 51 | EXPECT_TRUE(decoder.cancelled()); |
| 52 | } |
| 53 | |
| 54 | TEST(FindDecodeHandler, SingleLevel_DoesntFindNonExistingField) { |
Alexei Frolov | fe9723c | 2020-05-11 13:20:47 -0700 | [diff] [blame] | 55 | CallbackDecoder decoder; |
Alexei Frolov | bbbedff | 2020-03-11 14:21:38 -0700 | [diff] [blame] | 56 | FindDecodeHandler finder(8); |
| 57 | |
| 58 | decoder.set_handler(&finder); |
Wyatt Hepler | e2cbadf | 2020-06-22 11:21:45 -0700 | [diff] [blame] | 59 | decoder.Decode(std::as_bytes(std::span(encoded_proto))); |
Alexei Frolov | bbbedff | 2020-03-11 14:21:38 -0700 | [diff] [blame] | 60 | |
| 61 | EXPECT_FALSE(finder.found()); |
| 62 | EXPECT_FALSE(decoder.cancelled()); |
| 63 | } |
| 64 | |
| 65 | TEST(FindDecodeHandler, MultiLevel_FindsExistingNestedField) { |
Alexei Frolov | fe9723c | 2020-05-11 13:20:47 -0700 | [diff] [blame] | 66 | CallbackDecoder decoder; |
Alexei Frolov | bbbedff | 2020-03-11 14:21:38 -0700 | [diff] [blame] | 67 | FindDecodeHandler nested_finder(1); |
| 68 | FindDecodeHandler finder(7, &nested_finder); |
| 69 | |
| 70 | decoder.set_handler(&finder); |
Wyatt Hepler | e2cbadf | 2020-06-22 11:21:45 -0700 | [diff] [blame] | 71 | decoder.Decode(std::as_bytes(std::span(encoded_proto))); |
Alexei Frolov | bbbedff | 2020-03-11 14:21:38 -0700 | [diff] [blame] | 72 | |
| 73 | EXPECT_TRUE(finder.found()); |
| 74 | EXPECT_TRUE(nested_finder.found()); |
| 75 | EXPECT_TRUE(decoder.cancelled()); |
| 76 | } |
| 77 | |
| 78 | TEST(FindDecodeHandler, MultiLevel_DoesntFindNonExistingNestedField) { |
Alexei Frolov | fe9723c | 2020-05-11 13:20:47 -0700 | [diff] [blame] | 79 | CallbackDecoder decoder; |
Alexei Frolov | bbbedff | 2020-03-11 14:21:38 -0700 | [diff] [blame] | 80 | FindDecodeHandler nested_finder(3); |
| 81 | FindDecodeHandler finder(7, &nested_finder); |
| 82 | |
| 83 | decoder.set_handler(&finder); |
Wyatt Hepler | e2cbadf | 2020-06-22 11:21:45 -0700 | [diff] [blame] | 84 | decoder.Decode(std::as_bytes(std::span(encoded_proto))); |
Alexei Frolov | bbbedff | 2020-03-11 14:21:38 -0700 | [diff] [blame] | 85 | |
| 86 | EXPECT_TRUE(finder.found()); |
| 87 | EXPECT_FALSE(nested_finder.found()); |
| 88 | EXPECT_FALSE(decoder.cancelled()); |
| 89 | } |
| 90 | |
| 91 | } // namespace |
| 92 | } // namespace pw::protobuf |