A full restructure

pkg/storage/encodedstorage/encoded_storage_test.go

@@ -0,0 +1,99 @@
+package encodedstorage
+
+import (
+	"bytes"
+	"io/ioutil"
+	"os"
+	"strconv"
+	"testing"
+
+	"github.com/minio-io/minio/pkg/storage"
+	"github.com/minio-io/minio/pkg/utils"
+	. "gopkg.in/check.v1"
+)
+
+type EncodedStorageSuite struct{}
+
+var _ = Suite(&EncodedStorageSuite{})
+
+func Test(t *testing.T) { TestingT(t) }
+
+func (s *EncodedStorageSuite) TestFileStoragePutAtRootPath(c *C) {
+	rootDir, err := utils.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].Name, Equals, "path1")
+}
+
+func (s *EncodedStorageSuite) TestFileStoragePutDirPath(c *C) {
+	rootDir, err := utils.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)
+}
+
+func (s *EncodedStorageSuite) TestObjectWithChunking(c *C) {
+	rootDir, err := utils.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)
+
+}