X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/1535c8d141b07f246a501d288a0d3a903ce41a56..469521bb7ea65a68f04a7595d3b6b10264026db6:/sdk/go/httpserver/request_limiter_test.go diff --git a/sdk/go/httpserver/request_limiter_test.go b/sdk/go/httpserver/request_limiter_test.go index a8cc806962..9258fbfa58 100644 --- a/sdk/go/httpserver/request_limiter_test.go +++ b/sdk/go/httpserver/request_limiter_test.go @@ -1,3 +1,7 @@ +// Copyright (C) The Arvados Authors. All rights reserved. +// +// SPDX-License-Identifier: Apache-2.0 + package httpserver import ( @@ -18,7 +22,7 @@ func (h *testHandler) ServeHTTP(resp http.ResponseWriter, req *http.Request) { <-h.okToProceed } -func newTestHandler(maxReqs int) *testHandler { +func newTestHandler() *testHandler { return &testHandler{ inHandler: make(chan struct{}), okToProceed: make(chan struct{}), @@ -26,8 +30,8 @@ func newTestHandler(maxReqs int) *testHandler { } func TestRequestLimiter1(t *testing.T) { - h := newTestHandler(10) - l := NewRequestLimiter(1, h) + h := newTestHandler() + l := NewRequestLimiter(1, h, nil) var wg sync.WaitGroup resps := make([]*httptest.ResponseRecorder, 10) for i := 0; i < 10; i++ { @@ -86,8 +90,8 @@ func TestRequestLimiter1(t *testing.T) { } func TestRequestLimiter10(t *testing.T) { - h := newTestHandler(10) - l := NewRequestLimiter(10, h) + h := newTestHandler() + l := NewRequestLimiter(10, h, nil) var wg sync.WaitGroup for i := 0; i < 10; i++ { wg.Add(1)