feat: read response from 201 on file upload

This commit is contained in:
Rohan Verma
2018-12-26 13:36:35 +05:30
parent e4c7d93da1
commit 8079b244c7
4 changed files with 71 additions and 14 deletions
+30 -1
View File
@@ -11,6 +11,11 @@ func TestS3_FileUpload(t *testing.T) {
return
}
defer testTxt.Close()
testPng, err := os.Open("avatar.png")
if err != nil {
return
}
defer testPng.Close()
type fields struct {
AccessKey string
@@ -44,13 +49,37 @@ func TestS3_FileUpload(t *testing.T) {
},
wantErr: false,
},
{
name: "Upload avatar.png",
fields: fields{
AccessKey: os.Getenv("AWS_S3_ACCESS_KEY"),
SecretKey: os.Getenv("AWS_S3_SECRET_KEY"),
Region: os.Getenv("AWS_S3_REGION"),
},
args: args{
UploadInput{
Bucket: os.Getenv("AWS_S3_BUCKET"),
Bucket: "zerodha-testbucket",
ObjectKey: "xyz/image.png",
ContentType: "image/png",
FileName: "avatar.png",
Body: testPng,
},
},
wantErr: false,
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
s3 := New(tt.fields.Region, tt.fields.AccessKey, tt.fields.SecretKey)
if err := s3.FileUpload(tt.args.u); (err != nil) != tt.wantErr {
resp, err := s3.FileUpload(tt.args.u)
if (err != nil) != tt.wantErr {
t.Errorf("S3.FileUpload() error = %v, wantErr %v", err, tt.wantErr)
}
// check for empty response
if (resp == UploadResponse{}) {
t.Errorf("S3.FileUpload() returned empty response, %v", resp)
}
})
}
}