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:
FindMethod(Service & service,uint32_t id)24 static const internal::Method* FindMethod(Service& service, uint32_t id) {
25 return service.FindMethod(id);
26 }
27 };
28
29 namespace {
30
InvokeIt(const internal::Method &,internal::ServerCall &,const internal::Packet &)31 void InvokeIt(const internal::Method&,
32 internal::ServerCall&,
33 const internal::Packet&) {}
34
35 class ServiceTestMethod : public internal::Method {
36 public:
ServiceTestMethod(uint32_t id,char the_value)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:
ServiceTestMethodUnion(ServiceTestMethod && method)45 constexpr ServiceTestMethodUnion(ServiceTestMethod&& method)
46 : impl_({.service_test = method}) {}
47
method() const48 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:
TestService()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
TEST(Service,MultipleMethods_FindMethod_Present)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
TEST(Service,MultipleMethods_FindMethod_NotPresent)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:
EmptyTestService()87 constexpr EmptyTestService() : Service(0xabcd, kMethods) {}
88 static constexpr std::array<ServiceTestMethodUnion, 0> kMethods = {};
89 };
90
TEST(Service,NoMethods_FindMethod_NotPresent)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
100