refactor test case

context_test.go

@@ -3425,24 +3425,46 @@ func TestContextSetCookieData(t *testing.T) {
 	})
 }
 
-func TestParallelHeaderWrite(t *testing.T) {
+func TestParallelHeaderAccess(t *testing.T) {
-	c, _ := CreateTestContext(httptest.NewRecorder())
+	t.Parallel()
-	wg := sync.WaitGroup{}
+	const iterations = 1000
-	wg.Add(1)
+	const goroutines = 8
-	go func() {
+
-		defer wg.Done()
+	testCases := []struct {
-		for i := 0; i < 1000; i++ {
+		name        string
-			c.Header("key", "value")
+		writerCount int
-		}
+		readerCount int
-	}()
+	}{
-	wg.Add(1)
+		{"parallel_write_only", goroutines, 0},
-	go func() {
+		{"parallel_write_and_read", goroutines / 2, goroutines / 2},
-		defer wg.Done()
+	}
-		for i := 0; i < 1000; i++ {
+
-			c.Header("key", "value")
+	for _, tc := range testCases {
-		}
+		t.Run(tc.name, func(t *testing.T) {
-	}()
+			c, _ := CreateTestContext(httptest.NewRecorder())
-	wg.Wait()
+			c.Request, _ = http.NewRequest(http.MethodGet, "/", nil)
+			wg := sync.WaitGroup{}
+			for i := 0; i < tc.writerCount; i++ {
+				wg.Add(1)
+				go func() {
+					defer wg.Done()
+					for _ = range iterations {
+						c.Header("key", "value")
+					}
+				}()
+			}
+			for i := 0; i < tc.readerCount; i++ {
+				wg.Add(1)
+				go func() {
+					defer wg.Done()
+					for _ = range iterations {
+						_ = c.GetHeader("key")
+					}
+				}()
+			}
+			wg.Wait()
+		})
+	}
 }
 
 func TestGetMapFromFormData(t *testing.T) {