Chore: Fix flaky serverlock integration test (#42633)

This commit is contained in:
Marcus Efraimsson
2021-12-02 12:53:21 +01:00
committed by GitHub
parent 4080bcbd6c
commit 54fa7b57fe

View File

@ -8,7 +8,7 @@ import (
"testing"
"time"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
)
func TestServerLok(t *testing.T) {
@ -16,21 +16,22 @@ func TestServerLok(t *testing.T) {
counter := 0
fn := func(context.Context) { counter++ }
atInterval := time.Second * 1
atInterval := time.Hour
ctx := context.Background()
//this time `fn` should be executed
assert.Nil(t, sl.LockAndExecute(ctx, "test-operation", atInterval, fn))
require.Nil(t, sl.LockAndExecute(ctx, "test-operation", atInterval, fn))
require.Equal(t, 1, counter)
//this should not execute `fn`
assert.Nil(t, sl.LockAndExecute(ctx, "test-operation", atInterval, fn))
assert.Nil(t, sl.LockAndExecute(ctx, "test-operation", atInterval, fn))
require.Nil(t, sl.LockAndExecute(ctx, "test-operation", atInterval, fn))
require.Nil(t, sl.LockAndExecute(ctx, "test-operation", atInterval, fn))
require.Equal(t, 1, counter)
// wait 2 second.
<-time.After(time.Second * 2)
atInterval = time.Millisecond
// now `fn` should be executed again
err := sl.LockAndExecute(ctx, "test-operation", atInterval, fn)
assert.Nil(t, err)
assert.Equal(t, counter, 2)
require.Nil(t, err)
require.Equal(t, 2, counter)
}