增加测试用例

internal/compressions/writer_brotli_test.go

@@ -35,6 +35,33 @@ func BenchmarkBrotliWriter_Write(b *testing.B) {
 	}
 }
 
+func BenchmarkBrotliWriter_Write_Parallel(b *testing.B) {
+	var data = []byte(strings.Repeat("A", 1024))
+
+	b.RunParallel(func(pb *testing.PB) {
+		for pb.Next() {
+			var buf = &bytes.Buffer{}
+			writer, err := compressions.NewBrotliWriter(buf, 5)
+			if err != nil {
+				b.Fatal(err)
+			}
+
+			for j := 0; j < 100; j++ {
+				_, err = writer.Write(data)
+				if err != nil {
+					b.Fatal(err)
+				}
+
+				/**err = writer.Flush()
+				if err != nil {
+					b.Fatal(err)
+				}**/
+			}
+
+			_ = writer.Close()
+		}
+	})
+}
 
 func BenchmarkBrotliWriter_Write_Small(b *testing.B) {
 	var data = []byte(strings.Repeat("A", 16))
@@ -62,7 +89,6 @@ func BenchmarkBrotliWriter_Write_Small(b *testing.B) {
 	}
 }
 
-
 func BenchmarkBrotliWriter_Write_Large(b *testing.B) {
 	var data = []byte(strings.Repeat("A", 4096))
 
@@ -87,4 +113,4 @@ func BenchmarkBrotliWriter_Write_Large(b *testing.B) {
 
 		_ = writer.Close()
 	}
-}
+}