1 files changed, 51 insertions, 0 deletions
diff --git a/storage/memory/storage_test.go b/storage/memory/storage_test.go
new file mode 100644
index 0000000..4059dfa
--- /dev/null
+++ b/storage/memory/storage_test.go
@@ -0,0 +1,51 @@
+package memory
+
+import (
+	. "gopkg.in/check.v1"
+	"gopkg.in/src-d/go-git.v2/core"
+)
+
+type ObjectStorageSuite struct{}
+
+var _ = Suite(&ObjectStorageSuite{})
+
+func (s *ObjectStorageSuite) TestNew(c *C) {
+	os := NewObjectStorage()
+
+	o, err := os.New()
+	c.Assert(err, IsNil)
+	c.Assert(o.Size(), Equals, int64(0))
+}
+
+func (s *ObjectStorageSuite) TestSet(c *C) {
+	os := NewObjectStorage()
+
+	o, err := os.New()
+	c.Assert(err, IsNil)
+
+	o.SetType(core.CommitObject)
+	o.SetSize(3)
+	o.Writer().Write([]byte("foo"))
+
+	h, err := os.Set(o)
+	c.Assert(h.String(), Equals, "bc9968d75e48de59f0870ffb71f5e160bbbdcf52")
+}
+
+func (s *ObjectStorageSuite) TestGet(c *C) {
+	os := NewObjectStorage()
+
+	o, err := os.New()
+	c.Assert(err, IsNil)
+
+	o.SetType(core.CommitObject)
+	o.SetSize(3)
+	o.Writer().Write([]byte("foo"))
+
+	h, err := os.Set(o)
+	c.Assert(err, IsNil)
+
+	ro, err := os.Get(h)
+	c.Assert(err, IsNil)
+
+	c.Assert(ro, DeepEquals, o)
+}