X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/aed4d13a2fd3a97679c512eab2b4a9e5e24df112..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 3dc8e37dff..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 - BlockToReplication map[blockdigest.BlockDigest]int - BlockToCollectionUuids map[blockdigest.BlockDigest][]string -} - -func CompareSummarizedReadCollections(c *C, - summarized ReadCollections, - expected ExpectedSummary) { - - c.Assert(summarized.OwnerToCollectionSize, DeepEquals, - expected.OwnerToCollectionSize) - - c.Assert(summarized.BlockToReplication, DeepEquals, - expected.BlockToReplication) - - summarizedBlockToCollectionUuids := - make(map[blockdigest.BlockDigest]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.BlockDigest]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.MakeTestBlockDigest(1) - blockDigest2 := blockdigest.MakeTestBlockDigest(2) - - expected := ExpectedSummary{ - OwnerToCollectionSize: map[string]int{c.OwnerUuid: c.TotalSize}, - BlockToReplication: map[blockdigest.BlockDigest]int{blockDigest1: 5, blockDigest2: 5}, - BlockToCollectionUuids: map[blockdigest.BlockDigest][]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.MakeTestBlockDigest(1) - blockDigest2 := blockdigest.MakeTestBlockDigest(2) - blockDigest3 := blockdigest.MakeTestBlockDigest(3) - - expected := ExpectedSummary{ - OwnerToCollectionSize: map[string]int{ - c0.OwnerUuid: c0.TotalSize, - c1.OwnerUuid: c1.TotalSize, - }, - BlockToReplication: map[blockdigest.BlockDigest]int{ - blockDigest1: 5, - blockDigest2: 8, - blockDigest3: 8, - }, - BlockToCollectionUuids: map[blockdigest.BlockDigest][]string{ - blockDigest1: []string{c0.Uuid}, - blockDigest2: []string{c0.Uuid, c1.Uuid}, - blockDigest3: []string{c1.Uuid}, - }, - } - - CompareSummarizedReadCollections(checker, rc, expected) -}