You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
40 lines
840 B
Go
40 lines
840 B
Go
package queue
|
|
|
|
import (
|
|
"fmt"
|
|
"math"
|
|
"strconv"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestMultiQueuePusher(t *testing.T) {
|
|
const numPushers = 100
|
|
var pushers []Pusher
|
|
var mockedPushers []*mockedPusher
|
|
for i := 0; i < numPushers; i++ {
|
|
p := &mockedPusher{
|
|
name: "pusher:" + strconv.Itoa(i),
|
|
}
|
|
pushers = append(pushers, p)
|
|
mockedPushers = append(mockedPushers, p)
|
|
}
|
|
|
|
pusher := NewMultiQueuePusher(pushers)
|
|
assert.True(t, len(pusher.Name()) > 0)
|
|
|
|
for i := 0; i < 1000; i++ {
|
|
_ = pusher.Push("item")
|
|
}
|
|
|
|
var counts []int
|
|
for _, p := range mockedPushers {
|
|
counts = append(counts, p.count)
|
|
}
|
|
mean := calcMean(counts)
|
|
variance := calcVariance(mean, counts)
|
|
assert.True(t, math.Abs(mean-1000*(1-failProba)) < 10)
|
|
assert.True(t, variance < 100, fmt.Sprintf("too big variance - %.2f", variance))
|
|
}
|