import (
"git.curoverse.com/arvados.git/sdk/go/blockdigest"
- "reflect"
+ . "gopkg.in/check.v1"
"testing"
)
+// Gocheck boilerplate
+func Test(t *testing.T) {
+ TestingT(t)
+}
+
+// Tests that require the Keep server running
+type MySuite struct{}
+
+var _ = Suite(&MySuite{})
+
// This captures the result we expect from
// ReadCollections.Summarize(). Because CollectionUuidToIndex is
// indeterminate, we replace BlockToCollectionIndices with
BlockToCollectionUuids map[blockdigest.BlockDigest][]string
}
-func CompareSummarizedReadCollections(t *testing.T,
+func CompareSummarizedReadCollections(c *C,
summarized ReadCollections,
expected ExpectedSummary) {
- if !reflect.DeepEqual(summarized.OwnerToCollectionSize,
- expected.OwnerToCollectionSize) {
- t.Fatalf("Expected summarized OwnerToCollectionSize to look like %+v but instead it is %+v",
- expected.OwnerToCollectionSize,
- summarized.OwnerToCollectionSize)
- }
+ c.Assert(summarized.OwnerToCollectionSize, DeepEquals,
+ expected.OwnerToCollectionSize)
- if !reflect.DeepEqual(summarized.BlockToReplication,
- expected.BlockToReplication) {
- t.Fatalf("Expected summarized BlockToReplication to look like %+v but instead it is %+v",
- expected.BlockToReplication,
- summarized.BlockToReplication)
- }
+ c.Assert(summarized.BlockToReplication, DeepEquals,
+ expected.BlockToReplication)
summarizedBlockToCollectionUuids :=
make(map[blockdigest.BlockDigest]map[string]struct{})
}
}
- if !reflect.DeepEqual(summarizedBlockToCollectionUuids,
- expectedBlockToCollectionUuids) {
- t.Fatalf("Expected summarized BlockToCollectionUuids to look like %+v but instead it is %+v", expectedBlockToCollectionUuids, summarizedBlockToCollectionUuids)
- }
+ c.Assert(summarizedBlockToCollectionUuids, DeepEquals,
+ expectedBlockToCollectionUuids)
}
-func TestSummarizeSimple(t *testing.T) {
+func (s *MySuite) TestSummarizeSimple(checker *C) {
rc := MakeTestReadCollections([]TestCollectionSpec{TestCollectionSpec{
ReplicationLevel: 5,
Blocks: []int{1, 2},
BlockToCollectionUuids: map[blockdigest.BlockDigest][]string{blockDigest1: []string{c.Uuid}, blockDigest2: []string{c.Uuid}},
}
- CompareSummarizedReadCollections(t, rc, expected)
+ CompareSummarizedReadCollections(checker, rc, expected)
}
-func TestSummarizeOverlapping(t *testing.T) {
+func (s *MySuite) TestSummarizeOverlapping(checker *C) {
rc := MakeTestReadCollections([]TestCollectionSpec{
TestCollectionSpec{
ReplicationLevel: 5,
},
}
- CompareSummarizedReadCollections(t, rc, expected)
+ CompareSummarizedReadCollections(checker, rc, expected)
}