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
TEST(FindDecodeHandler,SingleLevel_FindsExistingField)43 TEST(FindDecodeHandler, SingleLevel_FindsExistingField) {
44 CallbackDecoder decoder;
45 FindDecodeHandler finder(3);
46
47 decoder.set_handler(&finder);
48 ASSERT_EQ(Status::Cancelled(), decoder.Decode(as_bytes(span(encoded_proto))));
49
50 EXPECT_TRUE(finder.found());
51 EXPECT_TRUE(decoder.cancelled());
52 }
53
TEST(FindDecodeHandler,SingleLevel_DoesntFindNonExistingField)54 TEST(FindDecodeHandler, SingleLevel_DoesntFindNonExistingField) {
55 CallbackDecoder decoder;
56 FindDecodeHandler finder(8);
57
58 decoder.set_handler(&finder);
59 ASSERT_EQ(OkStatus(), decoder.Decode(as_bytes(span(encoded_proto))));
60
61 EXPECT_FALSE(finder.found());
62 EXPECT_FALSE(decoder.cancelled());
63 }
64
TEST(FindDecodeHandler,MultiLevel_FindsExistingNestedField)65 TEST(FindDecodeHandler, MultiLevel_FindsExistingNestedField) {
66 CallbackDecoder decoder;
67 FindDecodeHandler nested_finder(1);
68 FindDecodeHandler finder(7, &nested_finder);
69
70 decoder.set_handler(&finder);
71 ASSERT_EQ(Status::Cancelled(), decoder.Decode(as_bytes(span(encoded_proto))));
72
73 EXPECT_TRUE(finder.found());
74 EXPECT_TRUE(nested_finder.found());
75 EXPECT_TRUE(decoder.cancelled());
76 }
77
TEST(FindDecodeHandler,MultiLevel_DoesntFindNonExistingNestedField)78 TEST(FindDecodeHandler, MultiLevel_DoesntFindNonExistingNestedField) {
79 CallbackDecoder decoder;
80 FindDecodeHandler nested_finder(3);
81 FindDecodeHandler finder(7, &nested_finder);
82
83 decoder.set_handler(&finder);
84 ASSERT_EQ(OkStatus(), decoder.Decode(as_bytes(span(encoded_proto))));
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
93