JCS-pub/common/pkgs/distlock/lockprovider/ipfs_lock_test.go

package lockprovider

import (
	"testing"

	. "github.com/smartystreets/goconvey/convey"
	"gitlink.org.cn/cloudream/common/pkgs/distlock"
)

func Test_IPFSLock(t *testing.T) {
	cases := []struct {
		title     string
		initLocks []distlock.Lock
		doLock    distlock.Lock
		wantOK    bool
	}{
		{
			title: "同节点，同一个Buzy锁",
			initLocks: []distlock.Lock{
				{
					Path: []string{IPFSLockPathPrefix, "hub1"},
					Name: IPFSBuzyLock,
				},
			},
			doLock: distlock.Lock{
				Path: []string{IPFSLockPathPrefix, "hub1"},
				Name: IPFSBuzyLock,
			},
			wantOK: true,
		},
		{
			title: "同节点，同一个GC锁",
			initLocks: []distlock.Lock{
				{
					Path: []string{IPFSLockPathPrefix, "hub1"},
					Name: IPFSGCLock,
				},
			},
			doLock: distlock.Lock{
				Path: []string{IPFSLockPathPrefix, "hub1"},
				Name: IPFSGCLock,
			},
			wantOK: true,
		},
		{
			title: "同时设置Buzy和GC",
			initLocks: []distlock.Lock{
				{
					Path:   []string{IPFSLockPathPrefix, "hub1"},
					Name:   IPFSBuzyLock,
					Target: *NewStringLockTarget(),
				},
			},
			doLock: distlock.Lock{
				Path:   []string{IPFSLockPathPrefix, "hub1"},
				Name:   IPFSGCLock,
				Target: *NewStringLockTarget(),
			},
			wantOK: false,
		},
	}

	for _, ca := range cases {
		Convey(ca.title, t, func() {
			ipfsLock := NewIPFSLock()

			for _, l := range ca.initLocks {
				ipfsLock.Lock("req1", l)
			}

			err := ipfsLock.CanLock(ca.doLock)
			if ca.wantOK {
				So(err, ShouldBeNil)
			} else {
				So(err, ShouldNotBeNil)
			}
		})
	}

	Convey("解锁", t, func() {
		ipfsLock := NewIPFSLock()

		lock := distlock.Lock{
			Path: []string{IPFSLockPathPrefix, "hub1"},
			Name: IPFSBuzyLock,
		}

		ipfsLock.Lock("req1", lock)

		err := ipfsLock.CanLock(lock)
		So(err, ShouldBeNil)

		ipfsLock.Unlock("req1", lock)

		lock = distlock.Lock{
			Path: []string{IPFSLockPathPrefix, "hub1"},
			Name: IPFSGCLock,
		}
		err = ipfsLock.CanLock(lock)
		So(err, ShouldBeNil)
	})

}