1 files changed, 7 insertions, 5 deletions
diff --git a/pw_rpc/java/test/dev/pigweed/pw_rpc/EndpointTest.java b/pw_rpc/java/test/dev/pigweed/pw_rpc/EndpointTest.java
index 91722ee70..041f69398 100644
--- a/pw_rpc/java/test/dev/pigweed/pw_rpc/EndpointTest.java
+++ b/pw_rpc/java/test/dev/pigweed/pw_rpc/EndpointTest.java
@@ -36,11 +36,13 @@ public final class EndpointTest {
   @Rule public final MockitoRule mockito = MockitoJUnit.rule();
 
   private static final Service SERVICE = new Service("pw.rpc.test1.TheTestService",
-      Service.unaryMethod("SomeUnary", SomeMessage.class, SomeMessage.class),
-      Service.serverStreamingMethod("SomeServerStreaming", SomeMessage.class, SomeMessage.class),
-      Service.clientStreamingMethod("SomeClientStreaming", SomeMessage.class, SomeMessage.class),
+      Service.unaryMethod("SomeUnary", SomeMessage.parser(), SomeMessage.parser()),
+      Service.serverStreamingMethod(
+          "SomeServerStreaming", SomeMessage.parser(), SomeMessage.parser()),
+      Service.clientStreamingMethod(
+          "SomeClientStreaming", SomeMessage.parser(), SomeMessage.parser()),
       Service.bidirectionalStreamingMethod(
-          "SomeBidiStreaming", SomeMessage.class, SomeMessage.class));
+          "SomeBidiStreaming", SomeMessage.parser(), SomeMessage.parser()));
 
   private static final Method METHOD = SERVICE.method("SomeUnary");
 
@@ -195,7 +197,7 @@ public final class EndpointTest {
                        .build()))
         .isTrue();
     assertThat(endpoint.processClientPacket(call.rpc().method(),
-                   packetBuilder().setType(PacketType.CLIENT_STREAM_END).build()))
+                   packetBuilder().setType(PacketType.CLIENT_REQUEST_COMPLETION).build()))
         .isTrue();
     assertThat(endpoint.processClientPacket(call.rpc().method(),
                    packetBuilder()