gitea/modules/globallock/globallock_test.go

// Copyright 2024 The Gitea Authors. All rights reserved.
// SPDX-License-Identifier: MIT

package globallock

import (
	"context"
	"os"
	"sync"
	"testing"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

func TestLockAndDo(t *testing.T) {
	t.Run("redis", func(t *testing.T) {
		url := "redis://127.0.0.1:6379/0"
		if os.Getenv("CI") == "" {
			// Make it possible to run tests against a local redis instance
			url = os.Getenv("TEST_REDIS_URL")
			if url == "" {
				t.Skip("TEST_REDIS_URL not set and not running in CI")
				return
			}
		}

		oldDefaultLocker := defaultLocker
		oldInitFunc := initFunc
		defer func() {
			defaultLocker = oldDefaultLocker
			initFunc = oldInitFunc
			if defaultLocker == nil {
				initOnce = sync.Once{}
			}
		}()

		initOnce = sync.Once{}
		initFunc = func() {
			defaultLocker = NewRedisLocker(url)
		}

		testLockAndDo(t)
		require.NoError(t, defaultLocker.(*redisLocker).Close())
	})
	t.Run("memory", func(t *testing.T) {
		oldDefaultLocker := defaultLocker
		oldInitFunc := initFunc
		defer func() {
			defaultLocker = oldDefaultLocker
			initFunc = oldInitFunc
			if defaultLocker == nil {
				initOnce = sync.Once{}
			}
		}()

		initOnce = sync.Once{}
		initFunc = func() {
			defaultLocker = NewMemoryLocker()
		}

		testLockAndDo(t)
	})
}

func testLockAndDo(t *testing.T) {
	const concurrency = 1000

	ctx := context.Background()
	count := 0
	wg := sync.WaitGroup{}
	wg.Add(concurrency)
	for i := 0; i < concurrency; i++ {
		go func() {
			defer wg.Done()
			err := LockAndDo(ctx, "test", func(ctx context.Context) error {
				count++

				// It's impossible to acquire the lock inner the function
				ok, err := TryLockAndDo(ctx, "test", func(ctx context.Context) error {
					assert.Fail(t, "should not acquire the lock")
					return nil
				})
				assert.False(t, ok)
				assert.NoError(t, err)

				return nil
			})
			require.NoError(t, err)
		}()
	}

	wg.Wait()

	assert.Equal(t, concurrency, count)
}