third_party/pigweed/repo/pw_rpc/service_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_rpc/service.h"
  16
  17 #include "gtest/gtest.h"
  18 #include "pw_rpc/internal/method.h"
  19
  20 namespace pw::rpc {
  21
  22 class ServiceTestHelper {
  23  public:
  24   static const internal::Method* FindMethod(Service& service, uint32_t id) {
  25     return service.FindMethod(id);
  26   }
  27 };
  28
  29 namespace {
  30
  31 void InvokeIt(const internal::Method&,
  32               internal::ServerCall&,
  33               const internal::Packet&) {}
  34
  35 class ServiceTestMethod : public internal::Method {
  36  public:
  37   constexpr ServiceTestMethod(uint32_t id, char the_value)
  38       : internal::Method(id, InvokeIt), value(the_value) {}
  39
  40   char value;  // Add a member so the class is larger than the base Method.
  41 };
  42
  43 class ServiceTestMethodUnion : public internal::MethodUnion {
  44  public:
  45   constexpr ServiceTestMethodUnion(ServiceTestMethod&& method)
  46       : impl_({.service_test = method}) {}
  47
  48   constexpr const internal::Method& method() const { return impl_.method; }
  49
  50  private:
  51   union {
  52     internal::Method method;
  53     ServiceTestMethod service_test;
  54   } impl_;
  55 };
  56
  57 class TestService : public Service {
  58  public:
  59   constexpr TestService() : Service(0xabcd, kMethods) {}
  60
  61   static constexpr std::array<ServiceTestMethodUnion, 3> kMethods = {
  62       ServiceTestMethod(123, 'a'),
  63       ServiceTestMethod(456, 'b'),
  64       ServiceTestMethod(789, 'c'),
  65   };
  66 };
  67
  68 TEST(Service, MultipleMethods_FindMethod_Present) {
  69   TestService service;
  70   EXPECT_EQ(ServiceTestHelper::FindMethod(service, 123),
  71             &TestService::kMethods[0].method());
  72   EXPECT_EQ(ServiceTestHelper::FindMethod(service, 456),
  73             &TestService::kMethods[1].method());
  74   EXPECT_EQ(ServiceTestHelper::FindMethod(service, 789),
  75             &TestService::kMethods[2].method());
  76 }
  77
  78 TEST(Service, MultipleMethods_FindMethod_NotPresent) {
  79   TestService service;
  80   EXPECT_EQ(ServiceTestHelper::FindMethod(service, 0), nullptr);
  81   EXPECT_EQ(ServiceTestHelper::FindMethod(service, 457), nullptr);
  82   EXPECT_EQ(ServiceTestHelper::FindMethod(service, 999), nullptr);
  83 }
  84
  85 class EmptyTestService : public Service {
  86  public:
  87   constexpr EmptyTestService() : Service(0xabcd, kMethods) {}
  88   static constexpr std::array<ServiceTestMethodUnion, 0> kMethods = {};
  89 };
  90
  91 TEST(Service, NoMethods_FindMethod_NotPresent) {
  92   EmptyTestService service;
  93   EXPECT_EQ(ServiceTestHelper::FindMethod(service, 123), nullptr);
  94   EXPECT_EQ(ServiceTestHelper::FindMethod(service, 456), nullptr);
  95   EXPECT_EQ(ServiceTestHelper::FindMethod(service, 789), nullptr);
  96 }
  97
  98 }  // namespace
  99 }  // namespace pw::rpc