68 lines
1.0 KiB
Go
68 lines
1.0 KiB
Go
package mutexkv
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestMutexKVLock(t *testing.T) {
|
|
mkv := NewMutexKV()
|
|
|
|
mkv.Lock("foo")
|
|
|
|
doneCh := make(chan struct{})
|
|
|
|
go func() {
|
|
mkv.Lock("foo")
|
|
close(doneCh)
|
|
}()
|
|
|
|
select {
|
|
case <-doneCh:
|
|
t.Fatal("Second lock was able to be taken. This shouldn't happen.")
|
|
case <-time.After(50 * time.Millisecond):
|
|
// pass
|
|
}
|
|
}
|
|
|
|
func TestMutexKVUnlock(t *testing.T) {
|
|
mkv := NewMutexKV()
|
|
|
|
mkv.Lock("foo")
|
|
mkv.Unlock("foo")
|
|
|
|
doneCh := make(chan struct{})
|
|
|
|
go func() {
|
|
mkv.Lock("foo")
|
|
close(doneCh)
|
|
}()
|
|
|
|
select {
|
|
case <-doneCh:
|
|
// pass
|
|
case <-time.After(50 * time.Millisecond):
|
|
t.Fatal("Second lock blocked after unlock. This shouldn't happen.")
|
|
}
|
|
}
|
|
|
|
func TestMutexKVDifferentKeys(t *testing.T) {
|
|
mkv := NewMutexKV()
|
|
|
|
mkv.Lock("foo")
|
|
|
|
doneCh := make(chan struct{})
|
|
|
|
go func() {
|
|
mkv.Lock("bar")
|
|
close(doneCh)
|
|
}()
|
|
|
|
select {
|
|
case <-doneCh:
|
|
// pass
|
|
case <-time.After(50 * time.Millisecond):
|
|
t.Fatal("Second lock on a different key blocked. This shouldn't happen.")
|
|
}
|
|
}
|