X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/1304e044aa87a65145bf8b6d4bc141586556c0ed..0a6c326074bdba18a13428f4580a313a6b5d7687:/sdk/go/keepclient/root_sorter_test.go?ds=sidebyside diff --git a/sdk/go/keepclient/root_sorter_test.go b/sdk/go/keepclient/root_sorter_test.go index 0d45e908ed..a6fbaeded3 100644 --- a/sdk/go/keepclient/root_sorter_test.go +++ b/sdk/go/keepclient/root_sorter_test.go @@ -1,7 +1,10 @@ +// Copyright (C) The Arvados Authors. All rights reserved. +// +// SPDX-License-Identifier: Apache-2.0 + package keepclient import ( - "crypto/md5" "fmt" . "gopkg.in/check.v1" "strconv" @@ -9,28 +12,25 @@ import ( ) type RootSorterSuite struct{} + var _ = Suite(&RootSorterSuite{}) -func FakeSvcRoot(i uint64) (string) { +func FakeSvcRoot(i uint64) string { return fmt.Sprintf("https://%x.svc/", i) } -func FakeSvcUuid(i uint64) (string) { +func FakeSvcUUID(i uint64) string { return fmt.Sprintf("zzzzz-bi6l4-%015x", i) } -func FakeServiceRoots(n uint64) (map[string]string) { +func FakeServiceRoots(n uint64) map[string]string { sr := map[string]string{} - for i := uint64(0); i < n; i ++ { - sr[FakeSvcUuid(i)] = FakeSvcRoot(i) + for i := uint64(0); i < n; i++ { + sr[FakeSvcUUID(i)] = FakeSvcRoot(i) } return sr } -func Md5String(data string) (string) { - return fmt.Sprintf("%032x", md5.Sum([]byte(data))) -} - func (*RootSorterSuite) EmptyRoots(c *C) { rs := NewRootSorter(map[string]string{}, Md5String("foo")) c.Check(rs.GetSortedRoots(), Equals, []string{}) @@ -43,19 +43,21 @@ func (*RootSorterSuite) JustOneRoot(c *C) { func (*RootSorterSuite) ReferenceSet(c *C) { fakeroots := FakeServiceRoots(16) - expected_orders := []string{ + // These reference probe orders are explained further in + // ../../python/tests/test_keep_client.py: + expectedOrders := []string{ "3eab2d5fc9681074", "097dba52e648f1c3", "c5b4e023f8a7d691", "9d81c02e76a3bf54", } - for h, expected_order := range expected_orders { + for h, expectedOrder := range expectedOrders { hash := Md5String(fmt.Sprintf("%064x", h)) roots := NewRootSorter(fakeroots, hash).GetSortedRoots() - for i, svc_id_s := range strings.Split(expected_order, "") { - svc_id, err := strconv.ParseUint(svc_id_s, 16, 64) + for i, svcIDs := range strings.Split(expectedOrder, "") { + svcID, err := strconv.ParseUint(svcIDs, 16, 64) c.Assert(err, Equals, nil) - c.Check(roots[i], Equals, FakeSvcRoot(svc_id)) + c.Check(roots[i], Equals, FakeSvcRoot(svcID)) } } }