mirror of
https://github.com/grafana/grafana.git
synced 2024-12-02 13:39:19 -06:00
65 lines
1.7 KiB
Go
65 lines
1.7 KiB
Go
package serverlock
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestIntegrationServerLock_LockAndExecute(t *testing.T) {
|
|
if testing.Short() {
|
|
t.Skip("skipping integration test")
|
|
}
|
|
sl := createTestableServerLock(t)
|
|
|
|
counter := 0
|
|
fn := func(context.Context) { counter++ }
|
|
atInterval := time.Hour
|
|
ctx := context.Background()
|
|
|
|
//this time `fn` should be executed
|
|
require.Nil(t, sl.LockAndExecute(ctx, "test-operation", atInterval, fn))
|
|
require.Equal(t, 1, counter)
|
|
|
|
//this should not execute `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)
|
|
|
|
atInterval = time.Millisecond
|
|
|
|
// now `fn` should be executed again
|
|
err := sl.LockAndExecute(ctx, "test-operation", atInterval, fn)
|
|
require.Nil(t, err)
|
|
require.Equal(t, 2, counter)
|
|
}
|
|
|
|
func TestIntegrationServerLock_LockExecuteAndRelease(t *testing.T) {
|
|
if testing.Short() {
|
|
t.Skip("skipping integration test")
|
|
}
|
|
sl := createTestableServerLock(t)
|
|
|
|
counter := 0
|
|
fn := func(context.Context) { counter++ }
|
|
atInterval := time.Hour
|
|
ctx := context.Background()
|
|
|
|
//
|
|
err := sl.LockExecuteAndRelease(ctx, "test-operation", atInterval, fn)
|
|
require.NoError(t, err)
|
|
require.Equal(t, 1, counter)
|
|
|
|
// the function will be executed again, as everytime the lock is released
|
|
err = sl.LockExecuteAndRelease(ctx, "test-operation", atInterval, fn)
|
|
require.NoError(t, err)
|
|
err = sl.LockExecuteAndRelease(ctx, "test-operation", atInterval, fn)
|
|
require.NoError(t, err)
|
|
err = sl.LockExecuteAndRelease(ctx, "test-operation", atInterval, fn)
|
|
require.NoError(t, err)
|
|
|
|
require.Equal(t, 4, counter)
|
|
}
|