Function TestBasic

internal/kmutex/kmutex_test.go:31–94

(t *testing.T)

)

func TestBasic(t *testing.T) {
	t.Parallel()

	km := newKeyMutex()
	ctx := context.Background()

	km.Lock(ctx, "c1")
	km.Lock(ctx, "c2")

	assert.Equal(t, len(km.locks), 2)
	assert.Equal(t, km.locks["c1"].ref, 1)
	assert.Equal(t, km.locks["c2"].ref, 1)

	checkWaitFn := func(key string, num int) {
		retries := 100
		waitLock := false

		for range retries {
			// prevent from data-race
			km.mu.Lock()
			ref := km.locks[key].ref
			km.mu.Unlock()

			if ref == num {
				waitLock = true
				break
			}
			time.Sleep(time.Duration(randutil.Int63n(100)) * time.Millisecond)
		}
		assert.Equal(t, waitLock, true)
	}

	// should acquire successfully after release
	{
		waitCh := make(chan struct{})
		go func() {
			defer close(waitCh)

			km.Lock(ctx, "c1")
		}()
		checkWaitFn("c1", 2)

		km.Unlock("c1")

		<-waitCh
		assert.Equal(t, km.locks["c1"].ref, 1)
	}

	// failed to acquire if context cancel
	{
		var errCh = make(chan error, 1)

		ctx, cancel := context.WithCancel(context.Background())
		go func() {
			errCh <- km.Lock(ctx, "c1")
		}()

		checkWaitFn("c1", 2)