You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
100 lines
3.0 KiB
100 lines
3.0 KiB
4 months ago
|
// Copyright 2020 The Pigweed Authors
|
||
|
|
//
|
||
|
|
// Licensed under the Apache License, Version 2.0 (the "License"); you may not
|
||
|
|
// use this file except in compliance with the License. You may obtain a copy of
|
||
|
|
// the License at
|
||
|
|
//
|
||
|
|
// https://www.apache.org/licenses/LICENSE-2.0
|
||
|
|
//
|
||
|
|
// Unless required by applicable law or agreed to in writing, software
|
||
|
|
// distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
||
|
|
// WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
||
|
|
// License for the specific language governing permissions and limitations under
|
||
|
|
// the License.
|
||
|
|
|
||
|
|
#include "pw_rpc/service.h"
|
||
|
|
|
||
|
|
#include "gtest/gtest.h"
|
||
|
|
#include "pw_rpc/internal/method.h"
|
||
|
|
|
||
|
|
namespace pw::rpc {
|
||
|
|
|
||
|
|
class ServiceTestHelper {
|
||
|
|
public:
|
||
|
|
static const internal::Method* FindMethod(Service& service, uint32_t id) {
|
||
|
|
return service.FindMethod(id);
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
namespace {
|
||
|
|
|
||
|
|
void InvokeIt(const internal::Method&,
|
||
|
|
internal::ServerCall&,
|
||
|
|
const internal::Packet&) {}
|
||
|
|
|
||
|
|
class ServiceTestMethod : public internal::Method {
|
||
|
|
public:
|
||
|
|
constexpr ServiceTestMethod(uint32_t id, char the_value)
|
||
|
|
: internal::Method(id, InvokeIt), value(the_value) {}
|
||
|
|
|
||
|
|
char value; // Add a member so the class is larger than the base Method.
|
||
|
|
};
|
||
|
|
|
||
|
|
class ServiceTestMethodUnion : public internal::MethodUnion {
|
||
|
|
public:
|
||
|
|
constexpr ServiceTestMethodUnion(ServiceTestMethod&& method)
|
||
|
|
: impl_({.service_test = method}) {}
|
||
|
|
|
||
|
|
constexpr const internal::Method& method() const { return impl_.method; }
|
||
|
|
|
||
|
|
private:
|
||
|
|
union {
|
||
|
|
internal::Method method;
|
||
|
|
ServiceTestMethod service_test;
|
||
|
|
} impl_;
|
||
|
|
};
|
||
|
|
|
||
|
|
class TestService : public Service {
|
||
|
|
public:
|
||
|
|
constexpr TestService() : Service(0xabcd, kMethods) {}
|
||
|
|
|
||
|
|
static constexpr std::array<ServiceTestMethodUnion, 3> kMethods = {
|
||
|
|
ServiceTestMethod(123, 'a'),
|
||
|
|
ServiceTestMethod(456, 'b'),
|
||
|
|
ServiceTestMethod(789, 'c'),
|
||
|
|
};
|
||
|
|
};
|
||
|
|
|
||
|
|
TEST(Service, MultipleMethods_FindMethod_Present) {
|
||
|
|
TestService service;
|
||
|
|
EXPECT_EQ(ServiceTestHelper::FindMethod(service, 123),
|
||
|
|
&TestService::kMethods[0].method());
|
||
|
|
EXPECT_EQ(ServiceTestHelper::FindMethod(service, 456),
|
||
|
|
&TestService::kMethods[1].method());
|
||
|
|
EXPECT_EQ(ServiceTestHelper::FindMethod(service, 789),
|
||
|
|
&TestService::kMethods[2].method());
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST(Service, MultipleMethods_FindMethod_NotPresent) {
|
||
|
|
TestService service;
|
||
|
|
EXPECT_EQ(ServiceTestHelper::FindMethod(service, 0), nullptr);
|
||
|
|
EXPECT_EQ(ServiceTestHelper::FindMethod(service, 457), nullptr);
|
||
|
|
EXPECT_EQ(ServiceTestHelper::FindMethod(service, 999), nullptr);
|
||
|
|
}
|
||
|
|
|
||
|
|
class EmptyTestService : public Service {
|
||
|
|
public:
|
||
|
|
constexpr EmptyTestService() : Service(0xabcd, kMethods) {}
|
||
|
|
static constexpr std::array<ServiceTestMethodUnion, 0> kMethods = {};
|
||
|
|
};
|
||
|
|
|
||
|
|
TEST(Service, NoMethods_FindMethod_NotPresent) {
|
||
|
|
EmptyTestService service;
|
||
|
|
EXPECT_EQ(ServiceTestHelper::FindMethod(service, 123), nullptr);
|
||
|
|
EXPECT_EQ(ServiceTestHelper::FindMethod(service, 456), nullptr);
|
||
|
|
EXPECT_EQ(ServiceTestHelper::FindMethod(service, 789), nullptr);
|
||
|
|
}
|
||
|
|
|
||
|
|
} // namespace
|
||
|
|
} // namespace pw::rpc
|