mirror of
https://github.com/kemko/nomad.git
synced 2026-01-01 16:05:42 +03:00
85c30dfd1e
The stdlib testing package now includes this interface, so we can remove our dependency on the external library.
48 lines
810 B
Go
48 lines
810 B
Go
// Copyright (c) HashiCorp, Inc.
|
|
// SPDX-License-Identifier: BUSL-1.1
|
|
|
|
package testutil
|
|
|
|
import (
|
|
"maps"
|
|
"sync"
|
|
"testing"
|
|
)
|
|
|
|
func NewCallCounter() *CallCounter {
|
|
return &CallCounter{
|
|
counts: make(map[string]int),
|
|
}
|
|
}
|
|
|
|
type CallCounter struct {
|
|
lock sync.Mutex
|
|
counts map[string]int
|
|
}
|
|
|
|
func (c *CallCounter) Inc(name string) {
|
|
c.lock.Lock()
|
|
defer c.lock.Unlock()
|
|
c.counts[name]++
|
|
}
|
|
|
|
func (c *CallCounter) Get() map[string]int {
|
|
c.lock.Lock()
|
|
defer c.lock.Unlock()
|
|
return maps.Clone(c.counts)
|
|
}
|
|
|
|
func (c *CallCounter) Reset() {
|
|
c.lock.Lock()
|
|
defer c.lock.Unlock()
|
|
c.counts = make(map[string]int)
|
|
}
|
|
|
|
func (c *CallCounter) AssertCalled(t testing.TB, name string) {
|
|
t.Helper()
|
|
counts := c.Get()
|
|
if _, ok := counts[name]; !ok {
|
|
t.Errorf("'%s' not called; all counts: %v", name, counts)
|
|
}
|
|
}
|