X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/a6c7c8db01d37534622763dd385019e9dad17181..c1e7f148bf3340300ae2f41d1ba7588cdfbb3b42:/sdk/go/dispatch/throttle_test.go diff --git a/sdk/go/dispatch/throttle_test.go b/sdk/go/dispatch/throttle_test.go index d91513ed40..d12659645e 100644 --- a/sdk/go/dispatch/throttle_test.go +++ b/sdk/go/dispatch/throttle_test.go @@ -1,3 +1,7 @@ +// Copyright (C) The Arvados Authors. All rights reserved. +// +// SPDX-License-Identifier: Apache-2.0 + package dispatch import ( @@ -18,21 +22,20 @@ type ThrottleTestSuite struct{} func (*ThrottleTestSuite) TestThrottle(c *check.C) { uuid := "zzzzz-zzzzz-zzzzzzzzzzzzzzz" + t0 := throttle{} + c.Check(t0.Check(uuid), check.Equals, true) + c.Check(t0.Check(uuid), check.Equals, true) - t := throttle{} - c.Check(t.Check(uuid), check.Equals, true) - c.Check(t.Check(uuid), check.Equals, true) - - t = throttle{hold: time.Nanosecond} - c.Check(t.Check(uuid), check.Equals, true) + tNs := throttle{hold: time.Nanosecond} + c.Check(tNs.Check(uuid), check.Equals, true) time.Sleep(time.Microsecond) - c.Check(t.Check(uuid), check.Equals, true) - - t = throttle{hold: time.Minute} - c.Check(t.Check(uuid), check.Equals, true) - c.Check(t.Check(uuid), check.Equals, false) - c.Check(t.Check(uuid), check.Equals, false) - t.seen[uuid].last = time.Now().Add(-time.Hour) - c.Check(t.Check(uuid), check.Equals, true) - c.Check(t.Check(uuid), check.Equals, false) + c.Check(tNs.Check(uuid), check.Equals, true) + + tMin := throttle{hold: time.Minute} + c.Check(tMin.Check(uuid), check.Equals, true) + c.Check(tMin.Check(uuid), check.Equals, false) + c.Check(tMin.Check(uuid), check.Equals, false) + tMin.seen[uuid].last = time.Now().Add(-time.Hour) + c.Check(tMin.Check(uuid), check.Equals, true) + c.Check(tMin.Check(uuid), check.Equals, false) }