fix test

Azure · Apr 27, 2021 · 6de3ac7 · 6de3ac7
1 parent a793e92
commit 6de3ac7
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/sdk/internal/testframework/recording.go b/sdk/internal/testframework/recording.go
@@ -265,7 +265,7 @@ func getOptionalEnv(name string, defaultValue string) *string {
 func (r *Recording) matchRequest(req *http.Request, rec cassette.Request) bool {
 	isMatch := compareMethods(req, rec, r.c) &&
 		compareURLs(req, rec, r.c) &&
-		// compareHeaders(req, rec, r.c) &&
+		compareHeaders(req, rec, r.c) &&
 		compareBodies(req, rec, r.c)
 
 	return isMatch

diff --git a/sdk/internal/testframework/recording_sanitizer_test.go b/sdk/internal/testframework/recording_sanitizer_test.go
@@ -107,7 +107,10 @@ func (s *recordingSanitizerTests) TestAddUrlSanitizerSanitizes() {
 	target.AddUrlSanitizer(func(url *string) {
 		*url = strings.Replace(*url, secret, SanitizedValue, -1)
 	})
-	target.AddBodysanitizer(func(body *string) {
+	target.AddBodysanitizerReq(func(body *string) {
+		*body = strings.Replace(*body, secret, SanitizedValue, -1)
+	})
+	target.AddBodysanitizerResp(func(body *string) {
 		*body = strings.Replace(*body, secret, SanitizedValue, -1)
 	})