fix test

antlabs · May 17, 2024 · 1702051 · 1702051
1 parent 837e5e6
commit 1702051
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 4 deletions.
diff --git a/deflate/compression_decode_test.go b/deflate/compression_decode_test.go
@@ -54,7 +54,7 @@ func TestDecompressNoContextTakeover(t *testing.T) {
 			}
 			// 压缩下一段数据
 			var encode *CompressContextTakeover
-			gotPayload, err := encode.Compress(&tt.args.payload)
+			gotPayload, err := encode.Compress(&tt.args.payload, 0)
 			if (err != nil) != tt.wantErr {
 				t.Errorf("CompressNoContextTakeover() error = %v, wantErr %v", err, tt.wantErr)
 				return
@@ -134,7 +134,7 @@ func TestDecompressNoContextTakeover2(t *testing.T) {
 			for i := 0; i < tt.loop; i++ {
 				// 压缩下一段数据
 				var encode *CompressContextTakeover
-				gotPayload, err := encode.Compress(&needData)
+				gotPayload, err := encode.Compress(&needData, 0)
 				if (err != nil) != tt.wantErr {
 					t.Errorf("CompressNoContextTakeover() error = %v, wantErr %v", err, tt.wantErr)
 					return

diff --git a/deflate/compression_encode.go b/deflate/compression_encode.go
@@ -96,7 +96,6 @@ func (e *CompressContextTakeover) Compress(payload *[]byte, bit uint8) (encodePa
 
 	var dict []byte
 	if e != nil {
-		bit = e.bit
 		dict = e.dict.GetData()
 	}
 	w, p := newCompressContextTakeover(nil, DefaultCompressionLevel, bit)

diff --git a/deflate/compression_encode_test.go b/deflate/compression_encode_test.go
@@ -44,7 +44,7 @@ func TestEnCompressContextTakeover_Compress(t *testing.T) {
 				t.Errorf("NewCompressContextTakeover() error = %v, wantErr %v", err, tt.wantErr)
 				return
 			}
-			gotEncodePayload, err := e.Compress(&tt.args.payload)
+			gotEncodePayload, err := e.Compress(&tt.args.payload, 0)
 			if (err != nil) != tt.wantErr {
 				t.Errorf("CompressContextTakeover.Compress() error = %v, wantErr %v", err, tt.wantErr)
 				return