fix TestPutGetStream

sfc-gh-ext-simba-jl · sfc-gh-ext-simba-jl · commit 1ecb65756a08 · 2024-12-11T02:52:52.000-08:00
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 {

Original file line number	Diff line number	Diff line change
`@@ -165,11 +165,11 @@ func getFileStream(ctx context.Context) (*bytes.Buffer, error) {`
`165`	`165`
`166`	`166`	`// read a small amount of data to check if file stream will be used`
`167`	`167`	`buf := make([]byte, defaultStringBufferSize)`
`168`		`- _, err := r.Read(buf)`
	`168`	`+ n, err := r.Read(buf)`
`169`	`169`	`if err != nil {`
`170`	`170`	`return nil, err`
`171`	`171`	`}`
`172`		`- return bytes.NewBuffer(buf), nil`
	`172`	`+ return bytes.NewBuffer(buf[:n]), nil`
`173`	`173`	`}`
`174`	`174`
`175`	`175`	`func getFileTransferOptions(ctx context.Context) *SnowflakeFileTransferOptions {`