s3manager-web/delete-bucket_test.go

47 lines
978 B
Go
Raw Normal View History

2017-04-03 14:08:01 +02:00
package main
2017-04-02 17:10:36 +02:00
import (
"errors"
"net/http"
"net/http/httptest"
"testing"
"github.com/stretchr/testify/assert"
)
2017-04-03 14:08:01 +02:00
func TestDeleteBucketHandler(t *testing.T) {
2017-04-02 17:10:36 +02:00
assert := assert.New(t)
2017-04-03 14:08:01 +02:00
tests := map[string]struct {
s3Client S3Client
2017-04-02 17:10:36 +02:00
expectedStatusCode int
expectedBody string
}{
2017-04-03 14:08:01 +02:00
"success": {
s3Client: &S3ClientMock{},
2017-04-02 17:10:36 +02:00
expectedStatusCode: http.StatusNoContent,
expectedBody: "",
},
2017-04-03 14:08:01 +02:00
"s3 error": {
s3Client: &S3ClientMock{
2017-04-02 17:10:36 +02:00
Err: errors.New("internal S3 error"),
},
expectedStatusCode: http.StatusInternalServerError,
expectedBody: "error removing bucket\n",
},
}
for _, tc := range tests {
req, err := http.NewRequest("DELETE", "/api/buckets/bucketName", nil)
assert.NoError(err)
rr := httptest.NewRecorder()
2017-04-03 14:08:01 +02:00
handler := DeleteBucketHandler(tc.s3Client)
2017-04-02 17:10:36 +02:00
handler.ServeHTTP(rr, req)
2017-04-03 14:08:01 +02:00
assert.Equal(tc.expectedStatusCode, rr.Code)
assert.Equal(tc.expectedBody, rr.Body.String())
2017-04-02 17:10:36 +02:00
}
}