Update fsttest/recorder.go

Co-authored-by: Cameron Walters (cee-dub) <cameron.walters@fastly.com>
fastly · Jun 28, 2024 · c7df4c4 · c7df4c4
1 parent af333b6
commit c7df4c4
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/fsttest/recorder.go b/fsttest/recorder.go
@@ -29,11 +29,10 @@ func NewRecorder() *ResponseRecorder {
 
 // Header returns the response headers to mutate within a handler.
 func (r *ResponseRecorder) Header() fsthttp.Header {
-	if !r.headersDone {
+	if !r.headersSent {
 		return r.HeaderMap
 	}
-	// Once the send the headers, return a copy so any changes
-	// are discarded.
+	// Once sent, return a copy so any changes are discarded.
 	return r.HeaderMap.Clone()
 }