mirror of
https://github.com/minio/minio.git
synced 2024-12-26 23:25:54 -05:00
111 lines
3.2 KiB
Go
111 lines
3.2 KiB
Go
package storage
|
|
|
|
import (
|
|
"bytes"
|
|
"math/rand"
|
|
"strconv"
|
|
|
|
. "gopkg.in/check.v1"
|
|
)
|
|
|
|
func APITestSuite(c *C, create func() Storage) {
|
|
testCreateBucket(c, create)
|
|
testMultipleObjectCreation(c, create)
|
|
testPaging(c, create)
|
|
testObjectOverwriteFails(c, create)
|
|
testNonExistantBucketOperations(c, create)
|
|
testBucketRecreateFails(c, create)
|
|
}
|
|
|
|
func testCreateBucket(c *C, create func() Storage) {
|
|
// TODO
|
|
}
|
|
|
|
func testMultipleObjectCreation(c *C, create func() Storage) {
|
|
objects := make(map[string][]byte)
|
|
storage := create()
|
|
err := storage.StoreBucket("bucket")
|
|
c.Check(err, IsNil)
|
|
for i := 0; i < 10; i++ {
|
|
randomPerm := rand.Perm(10)
|
|
randomString := ""
|
|
for _, num := range randomPerm {
|
|
randomString = randomString + strconv.Itoa(num)
|
|
}
|
|
key := "obj" + strconv.Itoa(i)
|
|
objects[key] = []byte(randomString)
|
|
err := storage.StoreObject("bucket", key, bytes.NewBufferString(randomString))
|
|
c.Check(err, IsNil)
|
|
}
|
|
|
|
// ensure no duplicates
|
|
etags := make(map[string]string)
|
|
for key, value := range objects {
|
|
var byteBuffer bytes.Buffer
|
|
storage.CopyObjectToWriter(&byteBuffer, "bucket", key)
|
|
c.Check(bytes.Equal(value, byteBuffer.Bytes()), Equals, true)
|
|
|
|
metadata, err := storage.GetObjectMetadata("bucket", key)
|
|
c.Check(err, IsNil)
|
|
c.Check(metadata.Size, Equals, int64(len(value)))
|
|
|
|
_, ok := etags[metadata.ETag]
|
|
c.Check(ok, Equals, false)
|
|
etags[metadata.ETag] = metadata.ETag
|
|
}
|
|
}
|
|
|
|
func testPaging(c *C, create func() Storage) {
|
|
storage := create()
|
|
storage.StoreBucket("bucket")
|
|
storage.ListObjects("bucket", "", 5)
|
|
objects, isTruncated, err := storage.ListObjects("bucket", "", 5)
|
|
c.Check(len(objects), Equals, 0)
|
|
c.Check(isTruncated, Equals, false)
|
|
c.Check(err, IsNil)
|
|
for i := 1; i <= 5; i++ {
|
|
key := "obj" + strconv.Itoa(i)
|
|
storage.StoreObject("bucket", key, bytes.NewBufferString(key))
|
|
objects, isTruncated, err = storage.ListObjects("bucket", "", 5)
|
|
c.Check(len(objects), Equals, i)
|
|
c.Check(isTruncated, Equals, false)
|
|
c.Check(err, IsNil)
|
|
}
|
|
for i := 6; i <= 10; i++ {
|
|
key := "obj" + strconv.Itoa(i)
|
|
storage.StoreObject("bucket", key, bytes.NewBufferString(key))
|
|
objects, isTruncated, err = storage.ListObjects("bucket", "", 5)
|
|
c.Check(len(objects), Equals, 5)
|
|
c.Check(isTruncated, Equals, true)
|
|
c.Check(err, IsNil)
|
|
}
|
|
}
|
|
|
|
func testObjectOverwriteFails(c *C, create func() Storage) {
|
|
storage := create()
|
|
storage.StoreBucket("bucket")
|
|
err := storage.StoreObject("bucket", "object", bytes.NewBufferString("one"))
|
|
c.Check(err, IsNil)
|
|
err = storage.StoreObject("bucket", "object", bytes.NewBufferString("three"))
|
|
c.Check(err, Not(IsNil))
|
|
var bytesBuffer bytes.Buffer
|
|
length, err := storage.CopyObjectToWriter(&bytesBuffer, "bucket", "object")
|
|
c.Check(length, Equals, int64(len("one")))
|
|
c.Check(err, IsNil)
|
|
c.Check(string(bytesBuffer.Bytes()), Equals, "one")
|
|
}
|
|
|
|
func testNonExistantBucketOperations(c *C, create func() Storage) {
|
|
storage := create()
|
|
err := storage.StoreObject("bucket", "object", bytes.NewBufferString("one"))
|
|
c.Check(err, Not(IsNil))
|
|
}
|
|
|
|
func testBucketRecreateFails(c *C, create func() Storage) {
|
|
storage := create()
|
|
err := storage.StoreBucket("string")
|
|
c.Check(err, IsNil)
|
|
err = storage.StoreBucket("string")
|
|
c.Check(err, Not(IsNil))
|
|
}
|