Skip to content

Commit 1ecb657

Browse files
fix TestPutGetStream
1 parent 6040d01 commit 1ecb657

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

connection_util.go

+2-2
Original file line numberDiff line numberDiff line change
@@ -165,11 +165,11 @@ func getFileStream(ctx context.Context) (*bytes.Buffer, error) {
165165

166166
// read a small amount of data to check if file stream will be used
167167
buf := make([]byte, defaultStringBufferSize)
168-
_, err := r.Read(buf)
168+
n, err := r.Read(buf)
169169
if err != nil {
170170
return nil, err
171171
}
172-
return bytes.NewBuffer(buf), nil
172+
return bytes.NewBuffer(buf[:n]), nil
173173
}
174174

175175
func getFileTransferOptions(ctx context.Context) *SnowflakeFileTransferOptions {

0 commit comments

Comments
 (0)