Update trylock_test.go

subchen · May 29, 2018 · 318e27f · 318e27f
1 parent 6e9f3dd
commit 318e27f
Showing 1 changed file with 23 additions and 0 deletions.
diff --git a/trylock_test.go b/trylock_test.go
@@ -1,6 +1,7 @@
 package trylock
 
 import (
+	"sync/atomic"
 	"testing"
 	"time"
 )
@@ -143,3 +144,25 @@ func TestMutexLockUnLockInvalid(t *testing.T) {
 	}()
 	mu.RUnlock()
 }
+
+func TestMutexLockBroadcast(t *testing.T) {
+	mu := New()
+	mu.Lock()
+
+	done := int32(0)
+	for i := 0; i < 3; i++ {
+		go func() {
+			mu.RLock()
+			atomic.AddInt32(&done, 1)
+			mu.RUnlock()
+		}()
+	}
+
+	mu.Unlock()
+
+	time.Sleep(10 * time.Millisecond)
+
+	if done != 3 {
+		t.Fatal("Broadcast is failed")
+	}
+}