aboutsummaryrefslogtreecommitdiffstats
path: root/storage/test
diff options
context:
space:
mode:
Diffstat (limited to 'storage/test')
-rw-r--r--storage/test/storage_suite.go58
1 files changed, 50 insertions, 8 deletions
diff --git a/storage/test/storage_suite.go b/storage/test/storage_suite.go
index dc500bb..8686853 100644
--- a/storage/test/storage_suite.go
+++ b/storage/test/storage_suite.go
@@ -1,12 +1,17 @@
package test
import (
+ "encoding/hex"
+ "errors"
+ "fmt"
"io"
+ "io/ioutil"
"sort"
- . "gopkg.in/check.v1"
"gopkg.in/src-d/go-git.v4/config"
"gopkg.in/src-d/go-git.v4/core"
+
+ . "gopkg.in/check.v1"
)
type TestObject struct {
@@ -67,11 +72,11 @@ func (s *BaseStorageSuite) TestObjectStorageSetAndGet(c *C) {
o, err := s.ObjectStorage.Get(to.Type, h)
c.Assert(err, IsNil)
- c.Assert(o, Equals, to.Object)
+ c.Assert(objectEquals(o, to.Object), IsNil)
o, err = s.ObjectStorage.Get(core.AnyObject, h)
c.Assert(err, IsNil)
- c.Assert(o, Equals, to.Object)
+ c.Assert(objectEquals(o, to.Object), IsNil)
for _, t := range s.validTypes {
if t == to.Type {
@@ -85,7 +90,7 @@ func (s *BaseStorageSuite) TestObjectStorageSetAndGet(c *C) {
}
}
-func (s *BaseStorageSuite) TestObjectStorageGetIvalid(c *C) {
+func (s *BaseStorageSuite) TestObjectStorageGetInvalid(c *C) {
o := s.ObjectStorage.NewObject()
o.SetType(core.REFDeltaObject)
@@ -95,7 +100,9 @@ func (s *BaseStorageSuite) TestObjectStorageGetIvalid(c *C) {
func (s *BaseStorageSuite) TestObjectStorageIter(c *C) {
for _, o := range s.testObjects {
- s.ObjectStorage.Set(o.Object)
+ h, err := s.ObjectStorage.Set(o.Object)
+ c.Assert(err, IsNil)
+ c.Assert(h, Equals, o.Object.Hash())
}
for _, t := range s.validTypes {
@@ -105,7 +112,7 @@ func (s *BaseStorageSuite) TestObjectStorageIter(c *C) {
o, err := i.Next()
c.Assert(err, IsNil)
- c.Assert(o, Equals, s.testObjects[t].Object, comment)
+ c.Assert(objectEquals(o, s.testObjects[t].Object), IsNil)
o, err = i.Next()
c.Assert(o, IsNil)
@@ -125,7 +132,7 @@ func (s *BaseStorageSuite) TestObjectStorageIter(c *C) {
for _, to := range s.testObjects {
found := false
for _, o := range foundObjects {
- if to.Object == o {
+ if to.Object.Hash() == o.Hash() {
found = true
break
}
@@ -222,7 +229,7 @@ func (s *BaseStorageSuite) TestReferenceStorageGetNotFound(c *C) {
func (s *BaseStorageSuite) TestReferenceStorageIter(c *C) {
err := s.ReferenceStorage.Set(
- core.NewReferenceFromStrings("foo", "bc9968d75e48de59f0870ffb71f5e160bbbdcf52"),
+ core.NewReferenceFromStrings("refs/foo", "bc9968d75e48de59f0870ffb71f5e160bbbdcf52"),
)
c.Assert(err, IsNil)
@@ -283,3 +290,38 @@ func (s *BaseStorageSuite) TestConfigStorageRemotes(c *C) {
c.Assert(sorted[0], Equals, "bar")
c.Assert(sorted[1], Equals, "foo")
}
+
+func objectEquals(a core.Object, b core.Object) error {
+ ha := a.Hash()
+ hb := b.Hash()
+ if ha != hb {
+ return fmt.Errorf("hashes do not match: %s != %s",
+ ha.String(), hb.String())
+ }
+
+ ra, err := a.Reader()
+ if err != nil {
+ return fmt.Errorf("can't get reader on b: %q", err)
+ }
+
+ rb, err := b.Reader()
+ if err != nil {
+ return fmt.Errorf("can't get reader on a: %q", err)
+ }
+
+ ca, err := ioutil.ReadAll(ra)
+ if err != nil {
+ return fmt.Errorf("error reading a: %q", err)
+ }
+
+ cb, err := ioutil.ReadAll(rb)
+ if err != nil {
+ return fmt.Errorf("error reading b: %q", err)
+ }
+
+ if hex.EncodeToString(ca) != hex.EncodeToString(cb) {
+ return errors.New("content does not match")
+ }
+
+ return nil
+}