Skip to content

Commit 0660e07

Browse files
committed
Fix
1 parent 1e0a688 commit 0660e07

File tree

1 file changed

+4
-7
lines changed

1 file changed

+4
-7
lines changed

testing-common/src/main/java/io/opentelemetry/instrumentation/testing/recording/ResponseHeaderScrubber.java

Lines changed: 4 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -5,23 +5,20 @@
55

66
package io.opentelemetry.instrumentation.testing.recording;
77

8-
import com.github.tomakehurst.wiremock.common.FileSource;
9-
import com.github.tomakehurst.wiremock.extension.Parameters;
10-
import com.github.tomakehurst.wiremock.extension.ResponseTransformer;
8+
import com.github.tomakehurst.wiremock.extension.ResponseTransformerV2;
119
import com.github.tomakehurst.wiremock.http.HttpHeader;
1210
import com.github.tomakehurst.wiremock.http.HttpHeaders;
13-
import com.github.tomakehurst.wiremock.http.Request;
1411
import com.github.tomakehurst.wiremock.http.Response;
12+
import com.github.tomakehurst.wiremock.stubbing.ServeEvent;
1513

16-
public final class ResponseHeaderScrubber extends ResponseTransformer {
14+
public final class ResponseHeaderScrubber implements ResponseTransformerV2 {
1715
@Override
1816
public String getName() {
1917
return "scrub-response-header";
2018
}
2119

2220
@Override
23-
public Response transform(
24-
Request request, Response response, FileSource fileSource, Parameters parameters) {
21+
public Response transform(Response response, ServeEvent serveEvent) {
2522
HttpHeaders scrubbed = HttpHeaders.noHeaders();
2623
for (HttpHeader header : response.getHeaders().all()) {
2724
switch (header.key()) {

0 commit comments

Comments
 (0)