// 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 "pw_rpc/internal/method.h"
#include "pw_unit_test/framework.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::CallContext&, const internal::Packet&) {}

class ServiceTestMethod : public internal::Method {
 public:
  constexpr ServiceTestMethod(uint32_t id, char the_value)
      : internal::Method(id, InvokeIt, MethodType::kUnary), 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