locker

locker/source_locker_test.go

@@ -1,19 +1,39 @@
-package locker
+package locker_test
 
 import (
 	"sync"
 	"testing"
 	"time"
+
+	"github.com/charlienet/go-mixed/locker"
+	"github.com/stretchr/testify/assert"
 )
 
 var sourcekey = "u-0001"
 
 func TestTryLock(t *testing.T) {
+	l := locker.NewSourceLocker()
+	l.Lock("aa")
 
+	assert.False(t, l.TryLock("aa"))
+	assert.True(t, l.TryLock("bb"))
+
+	defer l.Unlock("aa")
+}
+
+func TestM(t *testing.T) {
+	l := locker.NewSourceLocker()
+
+	for i := 0; i < 10000000; i++ {
+		l.Lock("aaa")
+		l.Unlock("aaa")
+	}
+
+	t.Logf("%+v", l)
 }
 
 func TestSourceLocker(t *testing.T) {
-	l := NewSourceLocker()
+	l := locker.NewSourceLocker()
 
 	c := 5
 	n := 0
@@ -41,7 +61,7 @@ func TestSourceTryLock(t *testing.T) {
 	wg := new(sync.WaitGroup)
 	wg.Add(c)
 
-	l := NewSourceLocker()
+	l := locker.NewSourceLocker()
 
 	for i := 0; i < c; i++ {
 		go func() {
@@ -61,7 +81,7 @@ func TestSourceTryLock(t *testing.T) {
 }
 
 func BenchmarkSourceLocker(b *testing.B) {
-	l := NewSourceLocker()
+	l := locker.NewSourceLocker()
 
 	b.RunParallel(func(p *testing.PB) {
 		for p.Next() {