65 lines
1.7 KiB
Go
65 lines
1.7 KiB
Go
|
package cassandra
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"context"
|
|||
|
|
"testing"
|
|||
|
|
"time"
|
|||
|
|
|
|||
|
|
"github.com/stretchr/testify/assert"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
type LockTest struct {
|
|||
|
|
ID string `db:"id" partition_key:"true"`
|
|||
|
|
Holder string `db:"holder"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (l *LockTest) TableName() string { return "lock_test" }
|
|||
|
|
|
|||
|
|
func TestTryLockAndUnLock(t *testing.T) {
|
|||
|
|
ctx := context.Background()
|
|||
|
|
|
|||
|
|
assert.NoError(t, cassandraDBTest.EnsureTable(`
|
|||
|
|
CREATE TABLE IF NOT EXISTS my_keyspace.lock_test (
|
|||
|
|
id TEXT PRIMARY KEY,
|
|||
|
|
holder TEXT
|
|||
|
|
);
|
|||
|
|
`))
|
|||
|
|
|
|||
|
|
lockID := "lock-123"
|
|||
|
|
holder := "daniel"
|
|||
|
|
|
|||
|
|
lockDoc := &LockTest{ID: lockID, Holder: holder}
|
|||
|
|
|
|||
|
|
t.Run("acquire lock - success", func(t *testing.T) {
|
|||
|
|
err := cassandraDBTest.TryLock(ctx, lockDoc, "my_keyspace")
|
|||
|
|
assert.NoError(t, err, "TryLock 應該成功")
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
t.Run("acquire lock again - fail", func(t *testing.T) {
|
|||
|
|
err := cassandraDBTest.TryLock(ctx, lockDoc, "my_keyspace")
|
|||
|
|
assert.Error(t, err, "重複上鎖應該失敗")
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
t.Run("unlock", func(t *testing.T) {
|
|||
|
|
err := cassandraDBTest.UnLock(ctx, lockDoc, "my_keyspace")
|
|||
|
|
assert.NoError(t, err, "UnLock 應該成功")
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
t.Run("lock with TTL", func(t *testing.T) {
|
|||
|
|
lockWithTTL := &LockTest{ID: "lock-ttl", Holder: "jack"}
|
|||
|
|
err := cassandraDBTest.TryLock(ctx, lockWithTTL, "my_keyspace", WithLockTTL(2*time.Second))
|
|||
|
|
assert.NoError(t, err)
|
|||
|
|
// 兩秒後嘗試再次上鎖應該成功（TTL 過期）
|
|||
|
|
time.Sleep(3 * time.Second)
|
|||
|
|
err = cassandraDBTest.TryLock(ctx, lockWithTTL, "my_keyspace")
|
|||
|
|
assert.NoError(t, err)
|
|||
|
|
_ = cassandraDBTest.UnLock(ctx, lockWithTTL, "my_keyspace")
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
t.Run("unlock not exist", func(t *testing.T) {
|
|||
|
|
nonExist := &LockTest{ID: "not-exist", Holder: "nobody"}
|
|||
|
|
err := cassandraDBTest.UnLock(ctx, nonExist, "my_keyspace")
|
|||
|
|
assert.Error(t, err, "unlock 不存在的鎖應該失敗")
|
|||
|
|
})
|
|||
|
|
}
|