1 files changed, 92 insertions, 0 deletions
diff --git a/storage/test/storage_suite.go b/storage/test/storage_suite.go
new file mode 100644
index 0000000..2463d9d
--- /dev/null
+++ b/storage/test/storage_suite.go
@@ -0,0 +1,92 @@
+package test
+
+import (
+	. "gopkg.in/check.v1"
+	"gopkg.in/src-d/go-git.v4/core"
+	"io"
+)
+
+type TestObject struct {
+	Object core.Object
+	Hash   string
+	Type   core.ObjectType
+}
+
+func RunObjectStorageSuite(c *C, os core.ObjectStorage) {
+	commit := &core.MemoryObject{}
+	commit.SetType(core.CommitObject)
+	tree := &core.MemoryObject{}
+	tree.SetType(core.TreeObject)
+	blob := &core.MemoryObject{}
+	blob.SetType(core.BlobObject)
+	tag := &core.MemoryObject{}
+	tag.SetType(core.TagObject)
+
+	testObjects := map[core.ObjectType]TestObject{
+		core.CommitObject: TestObject{commit, "dcf5b16e76cce7425d0beaef62d79a7d10fce1f5", core.CommitObject},
+		core.TreeObject:   TestObject{tree, "4b825dc642cb6eb9a060e54bf8d69288fbee4904", core.TreeObject},
+		core.BlobObject:   TestObject{blob, "e69de29bb2d1d6434b8b29ae775ad8c2e48c5391", core.BlobObject},
+		core.TagObject:    TestObject{tag, "d994c6bb648123a17e8f70a966857c546b2a6f94", core.TagObject},
+	}
+
+	validTypes := []core.ObjectType{core.CommitObject, core.BlobObject, core.TagObject, core.TreeObject}
+
+	for _, to := range testObjects {
+		comment := Commentf("failed for type %s", to.Type.String())
+
+		h, err := os.Set(to.Object)
+		c.Assert(err, IsNil)
+		c.Assert(h.String(), Equals, to.Hash, comment)
+
+		o, err := os.Get(h, to.Type)
+		c.Assert(err, IsNil)
+		c.Assert(o, Equals, to.Object)
+
+		o, err = os.Get(h, core.AnyObject)
+		c.Assert(err, IsNil)
+		c.Assert(o, Equals, to.Object)
+
+		for _, validType := range validTypes {
+			if validType == to.Type {
+				continue
+			}
+			o, err = os.Get(h, validType)
+			c.Assert(o, IsNil)
+			c.Assert(err, Equals, core.ErrObjectNotFound)
+		}
+	}
+
+	for _, validType := range validTypes {
+		comment := Commentf("failed for type %s)", validType.String())
+		i, err := os.Iter(validType)
+		c.Assert(err, IsNil, comment)
+
+		o, err := i.Next()
+		c.Assert(err, IsNil)
+		c.Assert(o, Equals, testObjects[validType].Object, comment)
+
+		o, err = i.Next()
+		c.Assert(o, IsNil)
+		c.Assert(err, Equals, io.EOF, comment)
+	}
+
+	i, err := os.Iter(core.AnyObject)
+	c.Assert(err, IsNil)
+
+	foundObjects := []core.Object{}
+	i.ForEach(func(o core.Object) error {
+		foundObjects = append(foundObjects, o)
+		return nil
+	})
+	c.Assert(foundObjects, HasLen, len(testObjects))
+	for _, to := range testObjects {
+		found := false
+		for _, o := range foundObjects {
+			if to.Object == o {
+				found = true
+				break
+			}
+		}
+		c.Assert(found, Equals, true, Commentf("Object of type %s not found", to.Type.String()))
+	}
+}