fix TestPutGetStream

snowflakedb · Dec 11, 2024 · 1ecb657 · 1ecb657
1 parent 6040d01
commit 1ecb657
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/connection_util.go b/connection_util.go
@@ -165,11 +165,11 @@ func getFileStream(ctx context.Context) (*bytes.Buffer, error) {
 
 	// read a small amount of data to check if file stream will be used
 	buf := make([]byte, defaultStringBufferSize)
-	_, err := r.Read(buf)
+	n, err := r.Read(buf)
 	if err != nil {
 		return nil, err
 	}
-	return bytes.NewBuffer(buf), nil
+	return bytes.NewBuffer(buf[:n]), nil
 }
 
 func getFileTransferOptions(ctx context.Context) *SnowflakeFileTransferOptions {