Add resetAll method to WireMockGrpcService

src/main/java/org/wiremock/grpc/dsl/WireMockGrpcService.java

@@ -20,6 +20,8 @@
 
 import com.github.tomakehurst.wiremock.client.CountMatchingStrategy;
 import com.github.tomakehurst.wiremock.client.WireMock;
+import com.github.tomakehurst.wiremock.http.RequestMethod;
+import com.github.tomakehurst.wiremock.matching.RequestPattern;
 import com.github.tomakehurst.wiremock.stubbing.StubMapping;
 import org.wiremock.annotations.Beta;
 
@@ -51,4 +53,23 @@ public GrpcVerification verify(int count, String method) {
   public GrpcVerification verify(CountMatchingStrategy countMatch, String method) {
     return new GrpcVerification(wireMock, countMatch, serviceName, method);
   }
+
+  /** Resets all active mocks for the current gRPC service */
+  public void resetAllForService() {
+    final String servicePath = "/" + serviceName;
+
+    wireMock.allStubMappings().getMappings().stream()
+        .filter(
+            mapping -> {
+              final RequestPattern requestMatcher = mapping.getRequest();
+              final RequestMethod requestMethod = requestMatcher.getMethod();
+              final String requestPath = requestMatcher.getUrlPath();
+
+              return (requestMethod != null)
+                  && (requestPath != null)
+                  && requestMethod.match(RequestMethod.POST).isExactMatch()
+                  && requestPath.startsWith(servicePath);
+            })
+        .forEach(wireMock::removeStubMapping);
+  }
 }

src/test/java/org/wiremock/grpc/GrpcAcceptanceTest.java

@@ -23,6 +23,7 @@
 import static org.junit.jupiter.api.Assertions.assertThrows;
 import static org.wiremock.grpc.dsl.WireMockGrpc.*;
 
+import com.example.grpc.AnotherGreetingServiceGrpc;
 import com.example.grpc.GreetingServiceGrpc;
 import com.example.grpc.request.HelloRequest;
 import com.example.grpc.response.HelloResponse;
@@ -46,6 +47,7 @@ public class GrpcAcceptanceTest {
   WireMockGrpcService mockGreetingService;
   ManagedChannel channel;
   GreetingsClient greetingsClient;
+  WireMock wireMock;
 
   @RegisterExtension
   public static WireMockExtension wm =
@@ -59,8 +61,8 @@ public class GrpcAcceptanceTest {
 
   @BeforeEach
   void init() {
-    mockGreetingService =
-        new WireMockGrpcService(new WireMock(wm.getPort()), GreetingServiceGrpc.SERVICE_NAME);
+    wireMock = new WireMock(wm.getPort());
+    mockGreetingService = new WireMockGrpcService(wireMock, GreetingServiceGrpc.SERVICE_NAME);
 
     channel = ManagedChannelBuilder.forAddress("localhost", wm.getPort()).usePlaintext().build();
     greetingsClient = new GreetingsClient(channel);
@@ -286,4 +288,28 @@ void randomDelay() {
     assertThat(greeting, is("Delayed hello"));
     assertThat(stopwatch.elapsed(MILLISECONDS), greaterThanOrEqualTo(500L));
   }
+
+  @Test
+  void resetAllForService() {
+    WireMockGrpcService mockAnotherGreetingService =
+        new WireMockGrpcService(wireMock, AnotherGreetingServiceGrpc.SERVICE_NAME);
+
+    mockAnotherGreetingService.stubFor(
+        method("greeting").willReturn(message(HelloResponse.newBuilder().setGreeting("Hi"))));
+
+    mockGreetingService.stubFor(
+        method("greeting").willReturn(message(HelloResponse.newBuilder().setGreeting("Hi"))));
+
+    mockGreetingService.stubFor(
+        method("oneGreetingEmptyReply").willReturn(message(Empty.newBuilder())));
+
+    // Starting point assertion
+    assertThat(wireMock.allStubMappings().getMappings(), iterableWithSize(3));
+
+    mockGreetingService.resetAllForService();
+    assertThat(wireMock.allStubMappings().getMappings(), iterableWithSize(1));
+
+    mockAnotherGreetingService.resetAllForService();
+    assertThat(wireMock.allStubMappings().getMappings(), iterableWithSize(0));
+  }
 }

src/test/proto/ExampleServices.proto

@@ -31,3 +31,10 @@ service GreetingService {
     rpc oneGreetingManyReplies(com.example.grpc.request.HelloRequest) returns (stream com.example.grpc.response.HelloResponse);
     rpc oneGreetingEmptyReply(com.example.grpc.request.HelloRequest) returns (google.protobuf.Empty);
 }
+
+service AnotherGreetingService {
+    rpc greeting(com.example.grpc.request.HelloRequest) returns (com.example.grpc.response.HelloResponse);
+    rpc manyGreetingsOneReply(stream com.example.grpc.request.HelloRequest) returns (com.example.grpc.response.HelloResponse);
+    rpc oneGreetingManyReplies(com.example.grpc.request.HelloRequest) returns (stream com.example.grpc.response.HelloResponse);
+    rpc oneGreetingEmptyReply(com.example.grpc.request.HelloRequest) returns (google.protobuf.Empty);
+}