pw_rpc/java/test/dev/pigweed/pw_rpc/PacketsTest.java

// Copyright 2021 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.

package dev.pigweed.pw_rpc;

import static com.google.common.truth.Truth.assertThat;

import com.google.protobuf.ExtensionRegistryLite;
import dev.pigweed.pw.rpc.internal.Packet.PacketType;
import dev.pigweed.pw.rpc.internal.Packet.RpcPacket;
import org.junit.Test;
import org.junit.runner.RunWith;

public final class PacketsTest {
  private static final Service SERVICE =
      new Service("Greetings", Service.unaryMethod("Hello", RpcPacket.class, RpcPacket.class));

  private static final PendingRpc RPC =
      PendingRpc.create(new Channel(123, null), SERVICE, SERVICE.method("Hello"));

  private static final RpcPacket PACKET = RpcPacket.newBuilder()
                                              .setChannelId(123)
                                              .setServiceId(RPC.service().id())
                                              .setMethodId(RPC.method().id())
                                              .build();

  @Test
  public void request() throws Exception {
    RpcPacket payload = RpcPacket.newBuilder().setType(PacketType.SERVER_STREAM).build();
    RpcPacket packet = RpcPacket.parseFrom(
        Packets.request(RPC, payload), ExtensionRegistryLite.getEmptyRegistry());
    assertThat(packet).isEqualTo(
        packet().setType(PacketType.REQUEST).setPayload(payload.toByteString()).build());
  }

  @Test
  public void cancel() throws Exception {
    RpcPacket packet =
        RpcPacket.parseFrom(Packets.cancel(RPC), ExtensionRegistryLite.getEmptyRegistry());
    assertThat(packet).isEqualTo(
        packet().setType(PacketType.CLIENT_ERROR).setStatus(Status.CANCELLED.code()).build());
  }

  @Test
  public void error() throws Exception {
    RpcPacket packet = RpcPacket.parseFrom(
        Packets.error(PACKET, Status.ALREADY_EXISTS), ExtensionRegistryLite.getEmptyRegistry());
    assertThat(packet).isEqualTo(
        packet().setType(PacketType.CLIENT_ERROR).setStatus(Status.ALREADY_EXISTS.code()).build());
  }

  private static RpcPacket.Builder packet() {
    return RpcPacket.newBuilder()
        .setChannelId(123)
        .setServiceId(Ids.calculate("Greetings"))
        .setMethodId(Ids.calculate("Hello"));
  }
}