chore: update unit tests

Author: DMwangnima

pkg/generic/thrift/raw_test.go

@@ -63,6 +63,7 @@ func TestRawWriter_Write(t *testing.T) {
 
 	// normal message
 	buf = buf[:0]
+	out = bufiox.NewBytesWriter(&buf)
 	err = w.Write(context.Background(), out, []byte("hello world"), "method", true, nil)
 	test.Assert(t, err == nil)
 	err = out.Flush()

pkg/remote/codec/thrift/codec_apache_test.go

@@ -50,7 +50,7 @@ func TestGetSkippedStructBuffer(t *testing.T) {
 	_, err = skipThriftStruct(trans)
 	test.Assert(t, err != nil, err)
 	test.Assert(t, strings.Contains(err.Error(), "skipThriftStruct"))
-	test.Assert(t, strings.Contains(err.Error(), "EOF"))
+	test.Assert(t, strings.Contains(err.Error(), "EOF") || strings.Contains(err.Error(), "no remaining data"))
 }
 
 func TestApacheMarshalUnmarshal(t *testing.T) {