37 lines
704 B
Go
37 lines
704 B
Go
|
package controllers
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestWorker_Add(t *testing.T) {
|
||
|
|
ctx, cancel := context.WithCancel(context.Background())
|
||
|
|
defer cancel()
|
||
|
|
|
||
|
|
w := newWorker(ctx, 2, func(st *ScaleTarget) {})
|
||
|
|
require.True(t, w.Add(&ScaleTarget{}))
|
||
|
|
require.True(t, w.Add(&ScaleTarget{}))
|
||
|
|
require.False(t, w.Add(&ScaleTarget{}))
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestWorker_Work(t *testing.T) {
|
||
|
|
ctx, cancel := context.WithCancel(context.Background())
|
||
|
|
defer cancel()
|
||
|
|
|
||
|
|
var count int
|
||
|
|
|
||
|
|
w := newWorker(ctx, 1, func(st *ScaleTarget) {
|
||
|
|
count++
|
||
|
|
cancel()
|
||
|
|
})
|
||
|
|
require.True(t, w.Add(&ScaleTarget{}))
|
||
|
|
require.False(t, w.Add(&ScaleTarget{}))
|
||
|
|
|
||
|
|
<-w.Done()
|
||
|
|
|
||
|
|
require.Equal(t, count, 1)
|
||
|
|
}
|