1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
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()))
}
}
|
