third_party/pigweed/repo/pw_protobuf/find_test.cc

   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) {
  44   CallbackDecoder decoder;
  45   FindDecodeHandler finder(3);
  46
  47   decoder.set_handler(&finder);
  48   decoder.Decode(std::as_bytes(std::span(encoded_proto)));
  49
  50   EXPECT_TRUE(finder.found());
  51   EXPECT_TRUE(decoder.cancelled());
  52 }
  53
  54 TEST(FindDecodeHandler, SingleLevel_DoesntFindNonExistingField) {
  55   CallbackDecoder decoder;
  56   FindDecodeHandler finder(8);
  57
  58   decoder.set_handler(&finder);
  59   decoder.Decode(std::as_bytes(std::span(encoded_proto)));
  60
  61   EXPECT_FALSE(finder.found());
  62   EXPECT_FALSE(decoder.cancelled());
  63 }
  64
  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   decoder.Decode(std::as_bytes(std::span(encoded_proto)));
  72
  73   EXPECT_TRUE(finder.found());
  74   EXPECT_TRUE(nested_finder.found());
  75   EXPECT_TRUE(decoder.cancelled());
  76 }
  77
  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   decoder.Decode(std::as_bytes(std::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