accept ::1 as valid X-Forwarded-For value in tests

puma · Feb 3, 2022 · 0ecde59 · 0ecde59
1 parent c25b052
commit 0ecde59
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/cmd/puma-dev/main_test.go b/cmd/puma-dev/main_test.go
@@ -322,7 +322,7 @@ func runPlatformAgnosticTestScenarios(t *testing.T) {
 
 		dumpedHeaders := parseDumpedHTTPHeadersFromBody(getURLWithHost(t, reqURL, statusHost))
 
-		assert.Equal(t, "127.0.0.1", dumpedHeaders["HTTP_X_FORWARDED_FOR"])
+		assert.Regexp(t, `127\.0\.0\.1|::1`, dumpedHeaders["HTTP_X_FORWARDED_FOR"])
 		assert.Equal(t, "http", dumpedHeaders["HTTP_X_FORWARDED_PROTO"])
 	})
 
@@ -332,7 +332,7 @@ func runPlatformAgnosticTestScenarios(t *testing.T) {
 
 		dumpedHeaders := parseDumpedHTTPHeadersFromBody(getURLWithHost(t, reqURL, statusHost))
 
-		assert.Equal(t, "127.0.0.1", dumpedHeaders["HTTP_X_FORWARDED_FOR"])
+		assert.Regexp(t, `^127\.0\.0\.1|::1`, dumpedHeaders["HTTP_X_FORWARDED_FOR"])
 		assert.Equal(t, "https", dumpedHeaders["HTTP_X_FORWARDED_PROTO"])
 	})
 }