X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/5e27876fa4d3faf3b973282bfb4f152c02345bdc..47a5c66948792602f657e5e1157bfd172d23c546:/services/datamanager/collection/collection_test.go diff --git a/services/datamanager/collection/collection_test.go b/services/datamanager/collection/collection_test.go deleted file mode 100644 index 1669bb7784..0000000000 --- a/services/datamanager/collection/collection_test.go +++ /dev/null @@ -1,123 +0,0 @@ -package collection - -import ( - "git.curoverse.com/arvados.git/sdk/go/blockdigest" - . "gopkg.in/check.v1" - "testing" -) - -// Gocheck boilerplate -func Test(t *testing.T) { - TestingT(t) -} - -type MySuite struct{} - -var _ = Suite(&MySuite{}) - -// This captures the result we expect from -// ReadCollections.Summarize(). Because CollectionUuidToIndex is -// indeterminate, we replace BlockToCollectionIndices with -// BlockToCollectionUuids. -type ExpectedSummary struct { - OwnerToCollectionSize map[string]int - BlockToDesiredReplication map[blockdigest.DigestWithSize]int - BlockToCollectionUuids map[blockdigest.DigestWithSize][]string -} - -func CompareSummarizedReadCollections(c *C, - summarized ReadCollections, - expected ExpectedSummary) { - - c.Assert(summarized.OwnerToCollectionSize, DeepEquals, - expected.OwnerToCollectionSize) - - c.Assert(summarized.BlockToDesiredReplication, DeepEquals, - expected.BlockToDesiredReplication) - - summarizedBlockToCollectionUuids := - make(map[blockdigest.DigestWithSize]map[string]struct{}) - for digest, indices := range summarized.BlockToCollectionIndices { - uuidSet := make(map[string]struct{}) - summarizedBlockToCollectionUuids[digest] = uuidSet - for _, index := range indices { - uuidSet[summarized.CollectionIndexToUuid[index]] = struct{}{} - } - } - - expectedBlockToCollectionUuids := - make(map[blockdigest.DigestWithSize]map[string]struct{}) - for digest, uuidSlice := range expected.BlockToCollectionUuids { - uuidSet := make(map[string]struct{}) - expectedBlockToCollectionUuids[digest] = uuidSet - for _, uuid := range uuidSlice { - uuidSet[uuid] = struct{}{} - } - } - - c.Assert(summarizedBlockToCollectionUuids, DeepEquals, - expectedBlockToCollectionUuids) -} - -func (s *MySuite) TestSummarizeSimple(checker *C) { - rc := MakeTestReadCollections([]TestCollectionSpec{TestCollectionSpec{ - ReplicationLevel: 5, - Blocks: []int{1, 2}, - }}) - - rc.Summarize(nil) - - c := rc.UuidToCollection["col0"] - - blockDigest1 := blockdigest.MakeTestDigestWithSize(1) - blockDigest2 := blockdigest.MakeTestDigestWithSize(2) - - expected := ExpectedSummary{ - OwnerToCollectionSize: map[string]int{c.OwnerUuid: c.TotalSize}, - BlockToDesiredReplication: map[blockdigest.DigestWithSize]int{blockDigest1: 5, blockDigest2: 5}, - BlockToCollectionUuids: map[blockdigest.DigestWithSize][]string{blockDigest1: []string{c.Uuid}, blockDigest2: []string{c.Uuid}}, - } - - CompareSummarizedReadCollections(checker, rc, expected) -} - -func (s *MySuite) TestSummarizeOverlapping(checker *C) { - rc := MakeTestReadCollections([]TestCollectionSpec{ - TestCollectionSpec{ - ReplicationLevel: 5, - Blocks: []int{1, 2}, - }, - TestCollectionSpec{ - ReplicationLevel: 8, - Blocks: []int{2, 3}, - }, - }) - - rc.Summarize(nil) - - c0 := rc.UuidToCollection["col0"] - c1 := rc.UuidToCollection["col1"] - - blockDigest1 := blockdigest.MakeTestDigestWithSize(1) - blockDigest2 := blockdigest.MakeTestDigestWithSize(2) - blockDigest3 := blockdigest.MakeTestDigestWithSize(3) - - expected := ExpectedSummary{ - OwnerToCollectionSize: map[string]int{ - c0.OwnerUuid: c0.TotalSize, - c1.OwnerUuid: c1.TotalSize, - }, - BlockToDesiredReplication: map[blockdigest.DigestWithSize]int{ - blockDigest1: 5, - blockDigest2: 8, - blockDigest3: 8, - }, - BlockToCollectionUuids: map[blockdigest.DigestWithSize][]string{ - blockDigest1: []string{c0.Uuid}, - blockDigest2: []string{c0.Uuid, c1.Uuid}, - blockDigest3: []string{c1.Uuid}, - }, - } - - CompareSummarizedReadCollections(checker, rc, expected) -}