You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
54 lines
1.2 KiB
Go
54 lines
1.2 KiB
Go
package redis
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/tal-tech/go-zero/core/stringx"
|
|
)
|
|
|
|
func TestRedisLock(t *testing.T) {
|
|
runOnRedis(t, func(client *Redis) {
|
|
key := stringx.Rand()
|
|
firstLock := NewRedisLock(client, key)
|
|
firstLock.SetExpire(5)
|
|
firstAcquire, err := firstLock.Acquire()
|
|
assert.Nil(t, err)
|
|
assert.True(t, firstAcquire)
|
|
|
|
secondLock := NewRedisLock(client, key)
|
|
secondLock.SetExpire(5)
|
|
againAcquire, err := secondLock.Acquire()
|
|
assert.Nil(t, err)
|
|
assert.False(t, againAcquire)
|
|
|
|
release, err := firstLock.Release()
|
|
assert.Nil(t, err)
|
|
assert.True(t, release)
|
|
|
|
endAcquire, err := secondLock.Acquire()
|
|
assert.Nil(t, err)
|
|
assert.True(t, endAcquire)
|
|
|
|
endAcquire, err = secondLock.Acquire()
|
|
assert.Nil(t, err)
|
|
assert.True(t, endAcquire)
|
|
|
|
release, err = secondLock.Release()
|
|
assert.Nil(t, err)
|
|
assert.True(t, release)
|
|
|
|
againAcquire, err = firstLock.Acquire()
|
|
assert.Nil(t, err)
|
|
assert.False(t, againAcquire)
|
|
|
|
release, err = secondLock.Release()
|
|
assert.Nil(t, err)
|
|
assert.True(t, release)
|
|
|
|
firstAcquire, err = firstLock.Acquire()
|
|
assert.Nil(t, err)
|
|
assert.True(t, firstAcquire)
|
|
})
|
|
}
|