s3manager-web/buckets-view_test.go

57 lines
1.2 KiB
Go
Raw Normal View History

2017-04-03 22:46:01 +02:00
package main
import (
"errors"
"net/http"
"net/http/httptest"
"testing"
minio "github.com/minio/minio-go"
"github.com/stretchr/testify/assert"
)
func TestBucketsViewHandler(t *testing.T) {
assert := assert.New(t)
tests := map[string]struct {
s3 S3Client
expectedStatusCode int
expectedBodyContains string
}{
"success": {
s3: &S3ClientMock{
Buckets: []minio.BucketInfo{
{Name: "testBucket"},
},
},
expectedStatusCode: http.StatusOK,
expectedBodyContains: "testBucket",
},
"success (bo buckets)": {
s3: &S3ClientMock{},
expectedStatusCode: http.StatusOK,
expectedBodyContains: "No buckets yet",
},
"s3 error": {
s3: &S3ClientMock{
Err: errors.New("internal S3 error"),
},
expectedStatusCode: http.StatusInternalServerError,
expectedBodyContains: "error listing buckets\n",
},
}
for _, tc := range tests {
req, err := http.NewRequest("GET", "/buckets", nil)
assert.NoError(err)
rr := httptest.NewRecorder()
handler := BucketsViewHandler(tc.s3)
handler.ServeHTTP(rr, req)
assert.Equal(tc.expectedStatusCode, rr.Code)
assert.Contains(rr.Body.String(), tc.expectedBodyContains)
}
}