// 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/client_server.h"

#include "gtest/gtest.h"
#include "pw_rpc/internal/packet.h"
#include "pw_rpc/internal/raw_method_union.h"
#include "pw_rpc/server_context.h"
#include "pw_rpc/service.h"
#include "pw_rpc_private/internal_test_utils.h"

namespace pw::rpc::internal {
namespace {

constexpr uint32_t kFakeChannelId = 1;
constexpr uint32_t kFakeServiceId = 3;
constexpr uint32_t kFakeMethodId = 10;

TestOutput<32> output;
rpc::Channel channels[] = {Channel::Create<kFakeChannelId>(&output)};

StatusWithSize FakeMethod(ServerContext&, ConstByteSpan, ByteSpan) {
  return StatusWithSize::Unimplemented();
}

class FakeService : public Service {
 public:
  FakeService(uint32_t id) : Service(id, kMethods) {}

  static constexpr std::array<RawMethodUnion, 1> kMethods = {
      RawMethod::Unary<FakeMethod>(kFakeMethodId),
  };
};

FakeService service(kFakeServiceId);

TEST(ClientServer, ProcessPacket_CallsServer) {
  ClientServer client_server(channels);
  client_server.server().RegisterService(service);

  Packet packet(
      PacketType::REQUEST, kFakeChannelId, kFakeServiceId, kFakeMethodId);
  std::array<std::byte, 32> buffer;
  Result result = packet.Encode(buffer);
  EXPECT_EQ(result.status(), OkStatus());

  EXPECT_EQ(client_server.ProcessPacket(result.value(), output), OkStatus());
}

TEST(ClientServer, ProcessPacket_CallsClient) {
  ClientServer client_server(channels);
  client_server.server().RegisterService(service);

  // Same packet as above, but type RESPONSE will skip the server and call into
  // the client.
  Packet packet(
      PacketType::RESPONSE, kFakeChannelId, kFakeServiceId, kFakeMethodId);
  std::array<std::byte, 32> buffer;
  Result result = packet.Encode(buffer);
  EXPECT_EQ(result.status(), OkStatus());

  // No calls are registered on the client, so this should fail.
  EXPECT_EQ(client_server.ProcessPacket(result.value(), output),
            Status::NotFound());
}

TEST(ClientServer, ProcessPacket_BadData) {
  ClientServer client_server(channels);
  EXPECT_EQ(client_server.ProcessPacket({}, output), Status::DataLoss());
}

}  // namespace
}  // namespace pw::rpc::internal