All checks were successful
/ Builds (1.19) (pull_request) Successful in 3m11s
/ Builds (1.20) (pull_request) Successful in 2m57s
/ DCO (pull_request) Successful in 4m7s
/ Lint (pull_request) Successful in 2m25s
/ Tests (1.19) (pull_request) Successful in 3m9s
/ Tests (1.20) (pull_request) Successful in 3m18s
/ Vulncheck (pull_request) Successful in 1m15s
Unwrap error before checking for s3 error Signed-off-by: Denis Kirillov <d.kirillov@yadro.com>
63 lines
1.8 KiB
Go
63 lines
1.8 KiB
Go
package handler
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/xml"
|
|
"testing"
|
|
"time"
|
|
|
|
s3Errors "git.frostfs.info/TrueCloudLab/frostfs-s3-gw/api/errors"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestPeriodicWriter(t *testing.T) {
|
|
const dur = 100 * time.Millisecond
|
|
const whitespaces = 8
|
|
expected := []byte(xml.Header)
|
|
for i := 0; i < whitespaces; i++ {
|
|
expected = append(expected, []byte(" ")...)
|
|
}
|
|
|
|
t.Run("writes data", func(t *testing.T) {
|
|
buf := bytes.NewBuffer(nil)
|
|
stop := periodicXMLWriter(buf, dur)
|
|
|
|
// N number of whitespaces + half durations to guarantee at least N writes in buffer
|
|
time.Sleep(whitespaces*dur + dur/2)
|
|
require.True(t, stop())
|
|
require.Equal(t, expected, buf.Bytes())
|
|
|
|
t.Run("no additional data after stop", func(t *testing.T) {
|
|
time.Sleep(2 * dur)
|
|
require.Equal(t, expected, buf.Bytes())
|
|
})
|
|
})
|
|
|
|
t.Run("does not write data", func(t *testing.T) {
|
|
buf := bytes.NewBuffer(nil)
|
|
stop := periodicXMLWriter(buf, dur)
|
|
time.Sleep(dur / 2)
|
|
require.False(t, stop())
|
|
require.Empty(t, buf.Bytes())
|
|
|
|
t.Run("disabled", func(t *testing.T) {
|
|
stop = periodicXMLWriter(buf, 0)
|
|
require.False(t, stop())
|
|
require.Empty(t, buf.Bytes())
|
|
})
|
|
})
|
|
}
|
|
|
|
func TestMultipartUploadInvalidPart(t *testing.T) {
|
|
hc := prepareHandlerContext(t)
|
|
|
|
bktName, objName := "bucket-to-upload-part", "object-multipart"
|
|
createTestBucket(hc, bktName)
|
|
partSize := 8 // less than min part size
|
|
|
|
multipartUpload := createMultipartUpload(hc, bktName, objName, map[string]string{})
|
|
etag1, _ := uploadPart(hc, bktName, objName, multipartUpload.UploadID, 1, partSize)
|
|
etag2, _ := uploadPart(hc, bktName, objName, multipartUpload.UploadID, 2, partSize)
|
|
w := completeMultipartUploadBase(hc, bktName, objName, multipartUpload.UploadID, []string{etag1, etag2})
|
|
assertS3Error(hc.t, w, s3Errors.GetAPIError(s3Errors.ErrEntityTooSmall))
|
|
}
|