blob: 959323f602fb0c40e60737c1da8762030a63f9fd [file] [log] [blame]
Wyatt Heplerd1591422020-09-15 10:04:41 -07001// 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"
Wyatt Heplercbd09c22020-09-15 11:17:24 -070018#include "pw_rpc/internal/method.h"
Wyatt Heplerd1591422020-09-15 10:04:41 -070019
20namespace pw::rpc {
21
22class ServiceTestHelper {
23 public:
Wyatt Heplercbd09c22020-09-15 11:17:24 -070024 static const internal::Method* FindMethod(Service& service, uint32_t id) {
Wyatt Heplerd1591422020-09-15 10:04:41 -070025 return service.FindMethod(id);
26 }
27};
28
29namespace {
30
Wyatt Heplercbd09c22020-09-15 11:17:24 -070031void InvokeIt(const internal::Method&,
Wyatt Heplerd1591422020-09-15 10:04:41 -070032 internal::ServerCall&,
33 const internal::Packet&) {}
34
Wyatt Heplercbd09c22020-09-15 11:17:24 -070035class ServiceTestMethod : public internal::Method {
Wyatt Heplerd1591422020-09-15 10:04:41 -070036 public:
37 constexpr ServiceTestMethod(uint32_t id, char the_value)
Wyatt Heplercbd09c22020-09-15 11:17:24 -070038 : internal::Method(id, InvokeIt), value(the_value) {}
Wyatt Heplerd1591422020-09-15 10:04:41 -070039
40 char value; // Add a member so the class is larger than the base Method.
41};
42
Alexei Frolova4d71502020-10-14 12:43:14 -070043class 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
Wyatt Heplerd1591422020-09-15 10:04:41 -070057class TestService : public Service {
58 public:
59 constexpr TestService() : Service(0xabcd, kMethods) {}
60
Alexei Frolova4d71502020-10-14 12:43:14 -070061 static constexpr std::array<ServiceTestMethodUnion, 3> kMethods = {
Wyatt Heplerd1591422020-09-15 10:04:41 -070062 ServiceTestMethod(123, 'a'),
63 ServiceTestMethod(456, 'b'),
64 ServiceTestMethod(789, 'c'),
Alexei Frolova4d71502020-10-14 12:43:14 -070065 };
Wyatt Heplerd1591422020-09-15 10:04:41 -070066};
67
68TEST(Service, MultipleMethods_FindMethod_Present) {
69 TestService service;
70 EXPECT_EQ(ServiceTestHelper::FindMethod(service, 123),
Alexei Frolova4d71502020-10-14 12:43:14 -070071 &TestService::kMethods[0].method());
Wyatt Heplerd1591422020-09-15 10:04:41 -070072 EXPECT_EQ(ServiceTestHelper::FindMethod(service, 456),
Alexei Frolova4d71502020-10-14 12:43:14 -070073 &TestService::kMethods[1].method());
Wyatt Heplerd1591422020-09-15 10:04:41 -070074 EXPECT_EQ(ServiceTestHelper::FindMethod(service, 789),
Alexei Frolova4d71502020-10-14 12:43:14 -070075 &TestService::kMethods[2].method());
Wyatt Heplerd1591422020-09-15 10:04:41 -070076}
77
78TEST(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
85class EmptyTestService : public Service {
86 public:
87 constexpr EmptyTestService() : Service(0xabcd, kMethods) {}
Alexei Frolova4d71502020-10-14 12:43:14 -070088 static constexpr std::array<ServiceTestMethodUnion, 0> kMethods = {};
Wyatt Heplerd1591422020-09-15 10:04:41 -070089};
90
91TEST(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