You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
124 lines
3.7 KiB
124 lines
3.7 KiB
package encodedstorage
|
|
|
|
import (
|
|
"bytes"
|
|
"io/ioutil"
|
|
"os"
|
|
"strconv"
|
|
"testing"
|
|
|
|
"github.com/minio-io/minio/pkgs/storage"
|
|
. "gopkg.in/check.v1"
|
|
)
|
|
|
|
type EncodedStorageSuite struct{}
|
|
|
|
var _ = Suite(&EncodedStorageSuite{})
|
|
|
|
func Test(t *testing.T) { TestingT(t) }
|
|
|
|
func makeTempTestDir() (string, error) {
|
|
return ioutil.TempDir("/tmp", "minio-test-")
|
|
}
|
|
|
|
func (s *EncodedStorageSuite) TestFileStoragePutAtRootPath(c *C) {
|
|
rootDir, err := makeTempTestDir()
|
|
c.Assert(err, IsNil)
|
|
defer os.RemoveAll(rootDir)
|
|
|
|
var objectStorage storage.ObjectStorage
|
|
objectStorage, err = NewStorage(rootDir, 10, 6, 1024)
|
|
c.Assert(err, IsNil)
|
|
objectBuffer := bytes.NewBuffer([]byte("object1"))
|
|
objectStorage.Put("path1", objectBuffer)
|
|
|
|
// assert object1 was created in correct path
|
|
objectResult1, err := objectStorage.Get("path1")
|
|
c.Assert(err, IsNil)
|
|
object1, _ := ioutil.ReadAll(objectResult1)
|
|
c.Assert(string(object1), Equals, "object1")
|
|
|
|
// objectList, err := objectStorage.List("/")
|
|
// c.Assert(err, IsNil)
|
|
// c.Assert(objectList[0].Path, Equals, "path1")
|
|
}
|
|
|
|
func (s *EncodedStorageSuite) TestFileStoragePutDirPath(c *C) {
|
|
rootDir, err := makeTempTestDir()
|
|
c.Assert(err, IsNil)
|
|
defer os.RemoveAll(rootDir)
|
|
|
|
var objectStorage storage.ObjectStorage
|
|
objectStorage, err = NewStorage(rootDir, 10, 6, 1024)
|
|
c.Assert(err, IsNil)
|
|
|
|
objectBuffer1 := bytes.NewBuffer([]byte("object1"))
|
|
objectStorage.Put("path1/path2/path3", objectBuffer1)
|
|
|
|
// assert object1 was created in correct path
|
|
objectResult1, err := objectStorage.Get("path1/path2/path3")
|
|
c.Assert(err, IsNil)
|
|
object1, _ := ioutil.ReadAll(objectResult1)
|
|
c.Assert(string(object1), Equals, "object1")
|
|
|
|
// add second object
|
|
objectBuffer2 := bytes.NewBuffer([]byte("object2"))
|
|
err = objectStorage.Put("path2/path2/path2", objectBuffer2)
|
|
c.Assert(err, IsNil)
|
|
|
|
// add third object
|
|
objectBuffer3 := bytes.NewBuffer([]byte("object3"))
|
|
err = objectStorage.Put("object3", objectBuffer3)
|
|
c.Assert(err, IsNil)
|
|
|
|
// TODO support list
|
|
// objectList, err := objectStorage.List("/")
|
|
// c.Assert(err, IsNil)
|
|
// c.Assert(objectList[0], Equals, storage.ObjectDescription{Path: "object3", IsDir: false, Hash: ""})
|
|
// c.Assert(objectList[1], Equals, storage.ObjectDescription{Path: "path1", IsDir: true, Hash: ""})
|
|
// c.Assert(objectList[2], Equals, storage.ObjectDescription{Path: "path2", IsDir: true, Hash: ""})
|
|
// c.Assert(len(objectList), Equals, 3)
|
|
//
|
|
// objectList, err = objectStorage.List("/path1")
|
|
// c.Assert(err, IsNil)
|
|
// c.Assert(objectList[0], Equals, storage.ObjectDescription{Path: "path2", IsDir: true, Hash: ""})
|
|
// c.Assert(len(objectList), Equals, 1)
|
|
//
|
|
// objectList, err = objectStorage.List("/path1/path2")
|
|
// c.Assert(err, IsNil)
|
|
// c.Assert(objectList[0], Equals, storage.ObjectDescription{Path: "path3", IsDir: false, Hash: ""})
|
|
// c.Assert(len(objectList), Equals, 1)
|
|
//
|
|
// objectList, err = objectStorage.List("/path1/path2/path3")
|
|
// c.Assert(err, Not(IsNil))
|
|
// c.Assert(objectList, IsNil)
|
|
}
|
|
|
|
func (s *EncodedStorageSuite) TestObjectWithChunking(c *C) {
|
|
rootDir, err := makeTempTestDir()
|
|
c.Assert(err, IsNil)
|
|
defer os.RemoveAll(rootDir)
|
|
|
|
var objectStorage storage.ObjectStorage
|
|
objectStorage, err = NewStorage(rootDir, 10, 6, 1024)
|
|
c.Assert(err, IsNil)
|
|
|
|
var buffer bytes.Buffer
|
|
for i := 0; i <= 2048; i++ {
|
|
buffer.Write([]byte(strconv.Itoa(i)))
|
|
}
|
|
|
|
reader := bytes.NewReader(buffer.Bytes())
|
|
|
|
err = objectStorage.Put("object", reader)
|
|
c.Assert(err, IsNil)
|
|
|
|
objectStorage2, err := NewStorage(rootDir, 10, 6, 1024)
|
|
c.Assert(err, IsNil)
|
|
objectResult, err := objectStorage2.Get("object")
|
|
c.Assert(err, IsNil)
|
|
result, err := ioutil.ReadAll(objectResult)
|
|
c.Assert(err, IsNil)
|
|
c.Assert(bytes.Compare(result, buffer.Bytes()), Equals, 0)
|
|
|
|
}
|
|
|