context_test.go

package watchdog

import (
	"context"
	"testing"
	"time"
)

func TestContext_Lock(t *testing.T) {
	ctx := &ctx{
		ctx:            context.Background(),
		client:         client,
		key:            "test",
		val:            "test1",
		expireDuration: time.Second,
	}
	err := ctx.client.Del(ctx.ctx, ctx.key).Err()
	if err != nil {
		t.Fatal(err)
	}

	ok, err := ctx.Lock()
	if err != nil {
		t.Fatal(err)
	}
	if !ok {
		t.Error("lock failed")
	}

	ctx.val = "test2"
	ok, err = ctx.Lock()
	if err != nil {
		t.Fatal(err)
	}
	if ok {
		t.Error("lock reentrant")
	}
}

func TestContext_Unlock(t *testing.T) {
	ctx := &ctx{
		ctx:            context.Background(),
		client:         client,
		key:            "test",
		val:            "test1",
		expireDuration: time.Second,
	}
	err := ctx.client.Del(ctx.ctx, ctx.key).Err()
	if err != nil {
		t.Fatal(err)
	}

	ok, err := ctx.Lock()
	if err != nil {
		t.Fatal(err)
	}
	if !ok {
		t.Error("lock failed")
	}

	ctx.val = "test2"
	ok, err = ctx.Unlock()
	if err != nil {
		t.Fatal(err)
	}
	if ok {
		t.Log("unlock reentrant")
	}

	ctx.val = "test1"
	ok, err = ctx.Unlock()
	if err != nil {
		t.Fatal(err)
	}
	if !ok {
		t.Error("unlock failed")
	}
}