package test import ( "io" . "gopkg.in/check.v1" "gopkg.in/src-d/go-git.v4/config" "gopkg.in/src-d/go-git.v4/core" ) type TestObject struct { Object core.Object Hash string Type core.ObjectType } type BaseStorageSuite struct { ObjectStorage core.ObjectStorage ReferenceStorage core.ReferenceStorage ConfigStore config.ConfigStorage validTypes []core.ObjectType testObjects map[core.ObjectType]TestObject } func NewBaseStorageSuite( os core.ObjectStorage, rs core.ReferenceStorage, cs config.ConfigStorage, ) BaseStorageSuite { 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) return BaseStorageSuite{ ObjectStorage: os, ReferenceStorage: rs, ConfigStore: cs, validTypes: []core.ObjectType{ core.CommitObject, core.BlobObject, core.TagObject, core.TreeObject, }, testObjects: map[core.ObjectType]TestObject{ core.CommitObject: {commit, "dcf5b16e76cce7425d0beaef62d79a7d10fce1f5", core.CommitObject}, core.TreeObject: {tree, "4b825dc642cb6eb9a060e54bf8d69288fbee4904", core.TreeObject}, core.BlobObject: {blob, "e69de29bb2d1d6434b8b29ae775ad8c2e48c5391", core.BlobObject}, core.TagObject: {tag, "d994c6bb648123a17e8f70a966857c546b2a6f94", core.TagObject}, }} } func (s *BaseStorageSuite) TestObjectStorageSetAndGet(c *C) { for _, to := range s.testObjects { comment := Commentf("failed for type %s", to.Type.String()) h, err := s.ObjectStorage.Set(to.Object) c.Assert(err, IsNil) c.Assert(h.String(), Equals, to.Hash, comment) o, err := s.ObjectStorage.Get(to.Type, h) c.Assert(err, IsNil) c.Assert(o, Equals, to.Object) o, err = s.ObjectStorage.Get(core.AnyObject, h) c.Assert(err, IsNil) c.Assert(o, Equals, to.Object) for _, t := range s.validTypes { if t == to.Type { continue } o, err = s.ObjectStorage.Get(t, h) c.Assert(o, IsNil) c.Assert(err, Equals, core.ErrObjectNotFound) } } } func (s *BaseStorageSuite) TestObjectStorageGetIvalid(c *C) { o := s.ObjectStorage.NewObject() o.SetType(core.REFDeltaObject) _, err := s.ObjectStorage.Set(o) c.Assert(err, NotNil) } func (s *BaseStorageSuite) TestObjectStorageIter(c *C) { for _, o := range s.testObjects { s.ObjectStorage.Set(o.Object) } for _, t := range s.validTypes { comment := Commentf("failed for type %s)", t.String()) i, err := s.ObjectStorage.Iter(t) c.Assert(err, IsNil, comment) o, err := i.Next() c.Assert(err, IsNil) c.Assert(o, Equals, s.testObjects[t].Object, comment) o, err = i.Next() c.Assert(o, IsNil) c.Assert(err, Equals, io.EOF, comment) } i, err := s.ObjectStorage.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(s.testObjects)) for _, to := range s.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())) } } func (s *BaseStorageSuite) TestTxObjectStorageSetAndCommit(c *C) { tx := s.ObjectStorage.Begin() for _, o := range s.testObjects { h, err := tx.Set(o.Object) c.Assert(err, IsNil) c.Assert(h.String(), Equals, o.Hash) } iter, err := s.ObjectStorage.Iter(core.AnyObject) c.Assert(err, IsNil) _, err = iter.Next() c.Assert(err, Equals, io.EOF) err = tx.Commit() c.Assert(err, IsNil) iter, err = s.ObjectStorage.Iter(core.AnyObject) c.Assert(err, IsNil) var count int iter.ForEach(func(o core.Object) error { count++ return nil }) c.Assert(count, Equals, 4) } func (s *BaseStorageSuite) TestTxObjectStorageSetAndGet(c *C) { tx := s.ObjectStorage.Begin() for _, expected := range s.testObjects { h, err := tx.Set(expected.Object) c.Assert(err, IsNil) c.Assert(h.String(), Equals, expected.Hash) o, err := tx.Get(expected.Type, core.NewHash(expected.Hash)) c.Assert(o.Hash().String(), DeepEquals, expected.Hash) } } func (s *BaseStorageSuite) TestTxObjectStorageGetNotFound(c *C) { tx := s.ObjectStorage.Begin() o, err := tx.Get(core.AnyObject, core.ZeroHash) c.Assert(o, IsNil) c.Assert(err, Equals, core.ErrObjectNotFound) } func (s *BaseStorageSuite) TestTxObjectStorageSetAndRollback(c *C) { tx := s.ObjectStorage.Begin() for _, o := range s.testObjects { h, err := tx.Set(o.Object) c.Assert(err, IsNil) c.Assert(h.String(), Equals, o.Hash) } err := tx.Rollback() c.Assert(err, IsNil) iter, err := s.ObjectStorage.Iter(core.AnyObject) c.Assert(err, IsNil) _, err = iter.Next() c.Assert(err, Equals, io.EOF) } func (s *BaseStorageSuite) TestReferenceStorageSetAndGet(c *C) { err := s.ReferenceStorage.Set( core.NewReferenceFromStrings("foo", "bc9968d75e48de59f0870ffb71f5e160bbbdcf52"), ) c.Assert(err, IsNil) err = s.ReferenceStorage.Set( core.NewReferenceFromStrings("bar", "482e0eada5de4039e6f216b45b3c9b683b83bfa"), ) c.Assert(err, IsNil) e, err := s.ReferenceStorage.Get(core.ReferenceName("foo")) c.Assert(err, IsNil) c.Assert(e.Hash().String(), Equals, "bc9968d75e48de59f0870ffb71f5e160bbbdcf52") } func (s *BaseStorageSuite) TestReferenceStorageGetNotFound(c *C) { r, err := s.ReferenceStorage.Get(core.ReferenceName("bar")) c.Assert(err, Equals, core.ErrReferenceNotFound) c.Assert(r, IsNil) } func (s *BaseStorageSuite) TestReferenceStorageIter(c *C) { err := s.ReferenceStorage.Set( core.NewReferenceFromStrings("foo", "bc9968d75e48de59f0870ffb71f5e160bbbdcf52"), ) c.Assert(err, IsNil) i, err := s.ReferenceStorage.Iter() c.Assert(err, IsNil) e, err := i.Next() c.Assert(err, IsNil) c.Assert(e.Hash().String(), Equals, "bc9968d75e48de59f0870ffb71f5e160bbbdcf52") e, err = i.Next() c.Assert(e, IsNil) c.Assert(err, Equals, io.EOF) } func (s *BaseStorageSuite) TestConfigStorageSetGetAndDelete(c *C) { err := s.ConfigStore.SetRemote(&config.RemoteConfig{ Name: "foo", URL: "http://foo/bar.git", }) c.Assert(err, IsNil) r, err := s.ConfigStore.Remote("foo") c.Assert(err, IsNil) c.Assert(r.Name, Equals, "foo") err = s.ConfigStore.DeleteRemote("foo") c.Assert(err, IsNil) r, err = s.ConfigStore.Remote("foo") c.Assert(err, Equals, config.ErrRemoteConfigNotFound) c.Assert(r, IsNil) } func (s *BaseStorageSuite) TestConfigStorageSetInvalid(c *C) { err := s.ConfigStore.SetRemote(&config.RemoteConfig{}) c.Assert(err, NotNil) } func (s *BaseStorageSuite) TestConfigStorageRemotes(c *C) { s.ConfigStore.SetRemote(&config.RemoteConfig{ Name: "foo", URL: "http://foo/bar.git", }) s.ConfigStore.SetRemote(&config.RemoteConfig{ Name: "bar", URL: "http://foo/bar.git", }) r, err := s.ConfigStore.Remotes() c.Assert(err, IsNil) c.Assert(r, HasLen, 2) c.Assert(r[0].Name, Equals, "foo") c.Assert(r[1].Name, Equals, "bar") }