1 files changed, 42 insertions, 0 deletions
diff --git a/src/go/collectors/go.d.plugin/modules/vsphere/scrape/throttled_caller_test.go b/src/go/collectors/go.d.plugin/modules/vsphere/scrape/throttled_caller_test.go
new file mode 100644
index 000000000..545ed1603
--- /dev/null
+++ b/src/go/collectors/go.d.plugin/modules/vsphere/scrape/throttled_caller_test.go
@@ -0,0 +1,42 @@
+// SPDX-License-Identifier: GPL-3.0-or-later
+
+package scrape
+
+import (
+	"sync"
+	"sync/atomic"
+	"testing"
+	"time"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func Test_throttledCaller(t *testing.T) {
+	var current int64
+	var max int64
+	var total int64
+	var mux sync.Mutex
+	limit := 5
+	n := 10000
+	tc := newThrottledCaller(limit)
+
+	for i := 0; i < n; i++ {
+		job := func() {
+			atomic.AddInt64(&total, 1)
+			atomic.AddInt64(&current, 1)
+			time.Sleep(100 * time.Microsecond)
+
+			mux.Lock()
+			defer mux.Unlock()
+			if atomic.LoadInt64(&current) > max {
+				max = atomic.LoadInt64(&current)
+			}
+			atomic.AddInt64(&current, -1)
+		}
+		tc.call(job)
+	}
+	tc.wait()
+
+	assert.Equal(t, int64(n), total)
+	assert.Equal(t, max, int64(limit))
+}