| Server IP : 153.92.12.68 / Your IP : 216.73.217.6 Web Server : LiteSpeed System : Linux id-dci-web1986.main-hosting.eu 5.14.0-611.26.1.el9_7.x86_64 #1 SMP PREEMPT_DYNAMIC Thu Jan 29 05:24:47 EST 2026 x86_64 User : u686484674 ( 686484674) PHP Version : 8.0.30 Disable Function : system, exec, shell_exec, passthru, mysql_list_dbs, ini_alter, dl, symlink, link, chgrp, leak, popen, apache_child_terminate, virtual, mb_send_mail MySQL : OFF | cURL : ON | WGET : ON | Perl : OFF | Python : OFF | Sudo : OFF | Pkexec : OFF Directory : /proc/self/root/opt/go/pkg/mod/github.com/aws/aws-sdk-go@v1.55.5/service/s3/s3crypto/ |
Upload File : |
package s3crypto
import (
"bytes"
"io/ioutil"
"net/http"
"os"
"testing"
"github.com/aws/aws-sdk-go/aws"
"github.com/aws/aws-sdk-go/aws/request"
"github.com/aws/aws-sdk-go/awstesting"
"github.com/aws/aws-sdk-go/internal/sdkio"
)
func TestBytesReadWriteSeeker_Read(t *testing.T) {
b := &bytesReadWriteSeeker{[]byte{1, 2, 3}, 0}
expected := []byte{1, 2, 3}
buf := make([]byte, 3)
n, err := b.Read(buf)
if err != nil {
t.Errorf("expected no error, but received %v", err)
}
if e, a := 3, n; e != a {
t.Errorf("expected %d, but received %d", e, a)
}
if !bytes.Equal(expected, buf) {
t.Error("expected equivalent byte slices, but received otherwise")
}
}
func TestBytesReadWriteSeeker_Write(t *testing.T) {
b := &bytesReadWriteSeeker{}
expected := []byte{1, 2, 3}
buf := make([]byte, 3)
n, err := b.Write([]byte{1, 2, 3})
if err != nil {
t.Errorf("expected no error, but received %v", err)
}
if e, a := 3, n; e != a {
t.Errorf("expected %d, but received %d", e, a)
}
n, err = b.Read(buf)
if err != nil {
t.Errorf("expected no error, but received %v", err)
}
if e, a := 3, n; e != a {
t.Errorf("expected %d, but received %d", e, a)
}
if !bytes.Equal(expected, buf) {
t.Error("expected equivalent byte slices, but received otherwise")
}
}
func TestBytesReadWriteSeeker_Seek(t *testing.T) {
b := &bytesReadWriteSeeker{[]byte{1, 2, 3}, 0}
expected := []byte{2, 3}
m, err := b.Seek(1, sdkio.SeekStart)
if err != nil {
t.Errorf("expected no error, but received %v", err)
}
if e, a := 1, int(m); e != a {
t.Errorf("expected %d, but received %d", e, a)
}
buf := make([]byte, 3)
n, err := b.Read(buf)
if err != nil {
t.Errorf("expected no error, but received %v", err)
}
if e, a := 2, n; e != a {
t.Errorf("expected %d, but received %d", e, a)
}
if !bytes.Equal(expected, buf[:n]) {
t.Error("expected equivalent byte slices, but received otherwise")
}
}
func TestGetWriterStore_TempFile(t *testing.T) {
response := http.Response{StatusCode: 200}
s := awstesting.NewClient(aws.NewConfig().WithMaxRetries(10))
s.Handlers.Validate.Clear()
s.Handlers.Send.Clear() // mock sending
s.Handlers.Send.PushBack(func(r *request.Request) {
r.HTTPResponse = &response
})
type testData struct {
Data string
}
out := &testData{}
r := s.NewRequest(&request.Operation{Name: "Operation"}, nil, out)
f, err := getWriterStore(r, "", true)
if err != nil {
t.Fatalf("expected no error, but received %v", err)
}
tempFile, ok := f.(*os.File)
if !ok {
t.Fatal("io.ReadWriteSeeker expected to be *os.file")
}
err = r.Send()
if err != nil {
t.Errorf("expected no error, but received %v", err)
}
if _, err := os.Stat(tempFile.Name()); !os.IsNotExist(err) {
t.Errorf("expected temp file be deleted, but still exists %v", tempFile.Name())
}
}
func TestGetWriterStore_TempFileWithRetry(t *testing.T) {
responses := []*http.Response{
{StatusCode: 500, Header: http.Header{}, Body: ioutil.NopCloser(&bytes.Buffer{})},
{StatusCode: 200, Header: http.Header{}, Body: ioutil.NopCloser(&bytes.Buffer{})},
}
s := awstesting.NewClient(aws.NewConfig().WithMaxRetries(10))
s.Handlers.Validate.Clear()
s.Handlers.Send.Clear() // mock sending
s.Handlers.Send.PushBack(func(r *request.Request) {
r.HTTPResponse = responses[0]
responses = responses[1:]
})
type testData struct {
Data string
}
out := &testData{}
r := s.NewRequest(&request.Operation{Name: "Operation"}, nil, out)
f, err := getWriterStore(r, "", true)
if err != nil {
t.Fatalf("expected no error, but received %v", err)
}
tempFile, ok := f.(*os.File)
if !ok {
t.Fatal("io.ReadWriteSeeker expected to be *os.file")
}
err = r.Send()
if err != nil {
t.Errorf("expected no error, but received %v", err)
}
if _, err := os.Stat(tempFile.Name()); !os.IsNotExist(err) {
t.Errorf("expected temp file be deleted, but still exists %v", tempFile.Name())
}
if v := len(responses); v != 0 {
t.Errorf("expect all retries to be used, have %v remaining", v)
}
}
func TestGetWriterStore_Memory(t *testing.T) {
response := http.Response{StatusCode: 200}
s := awstesting.NewClient(aws.NewConfig().WithMaxRetries(10))
s.Handlers.Validate.Clear()
s.Handlers.Send.Clear() // mock sending
s.Handlers.Send.PushBack(func(r *request.Request) {
r.HTTPResponse = &response
})
type testData struct {
Data string
}
out := &testData{}
r := s.NewRequest(&request.Operation{Name: "Operation"}, nil, out)
f, err := getWriterStore(r, "", false)
if err != nil {
t.Fatalf("expected no error, but received %v", err)
}
if _, ok := f.(*bytesReadWriteSeeker); !ok {
t.Fatal("io.ReadWriteSeeker expected to be *bytesReadWriteSeeker")
}
if err != nil {
t.Errorf("expected no error, but received %v", err)
}
err = r.Send()
if err != nil {
t.Errorf("expected no error, but received %v", err)
}
}