A full restructure

pkg/storage/appendstorage/append_storage_test.go

@@ -0,0 +1,116 @@
+package appendstorage
+
+import (
+	"bytes"
+	"io/ioutil"
+	"os"
+	"testing"
+
+	"github.com/minio-io/minio/pkg/storage"
+	"github.com/minio-io/minio/pkg/utils"
+	. "gopkg.in/check.v1"
+)
+
+type AppendStorageSuite struct{}
+
+var _ = Suite(&AppendStorageSuite{})
+
+func Test(t *testing.T) { TestingT(t) }
+
+func (s *AppendStorageSuite) TestAppendStoragePutAtRootPath(c *C) {
+	rootDir, err := utils.MakeTempTestDir()
+	c.Assert(err, IsNil)
+	defer os.RemoveAll(rootDir)
+
+	var objectStorage storage.ObjectStorage
+	objectStorage, err = NewStorage(rootDir, 0)
+	c.Assert(err, IsNil)
+
+	err = objectStorage.Put("path1", bytes.NewBuffer([]byte("object1")))
+	c.Assert(err, IsNil)
+
+	// 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")
+
+	err = objectStorage.Put("path2", bytes.NewBuffer([]byte("object2")))
+	c.Assert(err, IsNil)
+
+	// assert object1 was created in correct path
+	objectResult2, err := objectStorage.Get("path2")
+	c.Assert(err, IsNil)
+	object2, _ := ioutil.ReadAll(objectResult2)
+	c.Assert(string(object2), Equals, "object2")
+
+	objectResult1, err = objectStorage.Get("path1")
+	c.Assert(err, IsNil)
+	object1, _ = ioutil.ReadAll(objectResult1)
+	c.Assert(string(object1), Equals, "object1")
+}
+
+func (s *AppendStorageSuite) TestAppendStoragePutDirPath(c *C) {
+	rootDir, err := utils.MakeTempTestDir()
+	c.Assert(err, IsNil)
+	defer os.RemoveAll(rootDir)
+
+	var objectStorage storage.ObjectStorage
+	objectStorage, err = NewStorage(rootDir, 0)
+	c.Assert(err, IsNil)
+
+	// add object 1
+	objectStorage.Put("path1/path2/path3", bytes.NewBuffer([]byte("object")))
+
+	// 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, "object")
+
+	// add object 2
+	objectStorage.Put("path1/path1/path1", bytes.NewBuffer([]byte("object2")))
+
+	// assert object1 was created in correct path
+	objectResult2, err := objectStorage.Get("path1/path1/path1")
+	c.Assert(err, IsNil)
+	object2, _ := ioutil.ReadAll(objectResult2)
+	c.Assert(string(object2), Equals, "object2")
+}
+
+func (s *AppendStorageSuite) TestSerialization(c *C) {
+	rootDir, err := utils.MakeTempTestDir()
+	c.Assert(err, IsNil)
+	defer os.RemoveAll(rootDir)
+
+	objectStorage, err := NewStorage(rootDir, 0)
+	c.Assert(err, IsNil)
+
+	err = objectStorage.Put("path1", bytes.NewBuffer([]byte("object1")))
+	c.Assert(err, IsNil)
+	err = objectStorage.Put("path2", bytes.NewBuffer([]byte("object2")))
+	c.Assert(err, IsNil)
+	err = objectStorage.Put("path3/obj3", bytes.NewBuffer([]byte("object3")))
+	c.Assert(err, IsNil)
+
+	es := objectStorage.(*appendStorage)
+	es.file.Close()
+
+	objectStorage2, err := NewStorage(rootDir, 0)
+	c.Assert(err, IsNil)
+
+	objectResult1, err := objectStorage2.Get("path1")
+	c.Assert(err, IsNil)
+	object1, _ := ioutil.ReadAll(objectResult1)
+	c.Assert(string(object1), Equals, "object1")
+
+	objectResult2, err := objectStorage2.Get("path2")
+	c.Assert(err, IsNil)
+	object2, _ := ioutil.ReadAll(objectResult2)
+	c.Assert(string(object2), Equals, "object2")
+
+	objectResult3, err := objectStorage2.Get("path3/obj3")
+	c.Assert(err, IsNil)
+	object3, _ := ioutil.ReadAll(objectResult3)
+	c.Assert(string(object3), Equals, "object3")
+}