-
Notifications
You must be signed in to change notification settings - Fork 2
/
locker_test.go
107 lines (77 loc) · 2.19 KB
/
locker_test.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
package etcdlock
import (
"context"
"testing"
"time"
"github.com/stretchr/testify/suite"
"google.golang.org/grpc"
)
type LockerSuite struct {
suite.Suite
ctx context.Context
locker *Locker
}
func (s *LockerSuite) SetupSuite() {
s.ctx = context.Background()
locker, err := NewLocker(LockerOptions{
Address: "127.0.0.1:2379",
DialOptions: []grpc.DialOption{grpc.WithInsecure()},
})
s.Nil(err)
s.locker = locker
}
func (s *LockerSuite) TestNewLocker() {
locker, err := NewLocker(LockerOptions{
Address: "127.0.0.1:2379",
DialOptions: []grpc.DialOption{grpc.WithInsecure()},
})
s.Nil(err)
s.Equal(locker.etcdKeyPrefix, defaultEtcdKeyPrefix)
}
func (s *LockerSuite) TestLockNewKey() {
lock, err := s.locker.Lock(s.ctx, "test_lock_new_key", 3*time.Second)
s.Nil(err)
s.Contains(string(lock.keyName), "test_lock_new_key")
}
func (s *LockerSuite) TestLockEmptyKey() {
_, err := s.locker.Lock(s.ctx, "", 3*time.Second)
s.Equal(err.Error(), ErrEmptyKey.Error())
}
func (s *LockerSuite) TestLockLockedKey() {
start := time.Now()
_, err := s.locker.Lock(s.ctx, "test_lock_locked_key", 3*time.Second)
s.Nil(err)
_, err = s.locker.Lock(s.ctx, "test_lock_locked_key", 3*time.Second)
s.Nil(err)
s.True(time.Now().Sub(start) > 3*time.Second)
s.True(time.Now().Sub(start) < 4*time.Second)
}
func (s *LockerSuite) TestIsLockedEmptyKey() {
_, err := s.locker.IsLocked(s.ctx, "")
s.Equal(err.Error(), ErrEmptyKey.Error())
}
func (s *LockerSuite) TestIsLocked() {
isLocked, err := s.locker.IsLocked(s.ctx, "test_is_locked_1")
s.Nil(err)
s.False(isLocked)
_, err = s.locker.Lock(s.ctx, "test_is_locked_1", 3*time.Second)
s.Nil(err)
_, err = s.locker.Lock(s.ctx, "test_is_locked_2", 10*time.Second)
s.Nil(err)
isLocked, err = s.locker.IsLocked(s.ctx, "test_is_locked_1")
s.Nil(err)
s.True(isLocked)
isLocked, err = s.locker.IsLocked(s.ctx, "test_is_locked_1")
s.Nil(err)
s.True(isLocked)
time.Sleep(4 * time.Second)
isLocked, err = s.locker.IsLocked(s.ctx, "test_is_locked_1")
s.Nil(err)
s.False(isLocked)
isLocked, err = s.locker.IsLocked(s.ctx, "test_is_locked_2")
s.Nil(err)
s.True(isLocked)
}
func TestLocker(t *testing.T) {
suite.Run(t, new(LockerSuite))
}