From 1f19ed71aae328e4d70e128bd39a3abb811ac000 Mon Sep 17 00:00:00 2001
From: zhangbiao <zhangbiao@channelsoft.com>
Date: Fri, 14 Jun 2019 10:20:15 +0800
Subject: [PATCH] update mutex test

---
 g/internal/mutex/mutex_z_unit_test.go | 87 ++++++++++++++++++++++++---
 1 file changed, 77 insertions(+), 10 deletions(-)

diff --git a/g/internal/mutex/mutex_z_unit_test.go b/g/internal/mutex/mutex_z_unit_test.go
index 905d6c0d2..cb2567fab 100644
--- a/g/internal/mutex/mutex_z_unit_test.go
+++ b/g/internal/mutex/mutex_z_unit_test.go
@@ -7,26 +7,93 @@
 package mutex_test
 
 import (
+	"github.com/gogf/gf/g/container/garray"
 	"github.com/gogf/gf/g/internal/mutex"
 	"github.com/gogf/gf/g/test/gtest"
 	"testing"
+	"time"
 )
 
 func TestMutex(t *testing.T) {
 	gtest.Case(t, func() {
 		lock := mutex.New()
-		lock.Lock()
-		lock.Unlock()
 		gtest.Assert(lock.IsSafe(), true)
 
-		safeLock1 := mutex.New(false)
-		safeLock1.Lock()
-		safeLock1.Unlock()
-		gtest.Assert(safeLock1.IsSafe(), true)
+		lock = mutex.New(false)
+		gtest.Assert(lock.IsSafe(), true)
 
-		unsafeLock1 := mutex.New(true)
-		unsafeLock1.Lock()
-		unsafeLock1.Unlock()
-		gtest.Assert(unsafeLock1.IsSafe(), false)
+		lock = mutex.New(false, false)
+		gtest.Assert(lock.IsSafe(), true)
+
+		lock = mutex.New(true, false)
+		gtest.Assert(lock.IsSafe(), false)
+
+		lock = mutex.New(true, true)
+		gtest.Assert(lock.IsSafe(), false)
+
+		lock = mutex.New(true)
+		gtest.Assert(lock.IsSafe(), false)
+	})
+}
+
+func TestSafeMutex(t *testing.T) {
+	gtest.Case(t, func() {
+		safeLock := mutex.New(false)
+		array := garray.New()
+
+		go func() {
+			safeLock.Lock()
+			array.Append(1)
+			time.Sleep(100 * time.Millisecond)
+			array.Append(1)
+			safeLock.Unlock()
+		}()
+		go func() {
+			time.Sleep(10 * time.Millisecond)
+			safeLock.Lock()
+			array.Append(1)
+			time.Sleep(200 * time.Millisecond)
+			array.Append(1)
+			safeLock.Unlock()
+		}()
+		time.Sleep(50 * time.Millisecond)
+		gtest.Assert(array.Len(), 1)
+		time.Sleep(80 * time.Millisecond)
+		gtest.Assert(array.Len(), 3)
+		time.Sleep(100 * time.Millisecond)
+		gtest.Assert(array.Len(), 3)
+		time.Sleep(100 * time.Millisecond)
+		gtest.Assert(array.Len(), 4)
+	})
+}
+
+func TestUnsafeMutex(t *testing.T) {
+	gtest.Case(t, func() {
+		unsafeLock := mutex.New(true)
+		array := garray.New()
+
+		go func() {
+			unsafeLock.Lock()
+			array.Append(1)
+			time.Sleep(100 * time.Millisecond)
+			array.Append(1)
+			unsafeLock.Unlock()
+		}()
+		go func() {
+			time.Sleep(10 * time.Millisecond)
+			unsafeLock.Lock()
+			array.Append(1)
+			time.Sleep(200 * time.Millisecond)
+			array.Append(1)
+			unsafeLock.Unlock()
+		}()
+		time.Sleep(50 * time.Millisecond)
+		gtest.Assert(array.Len(), 2)
+		time.Sleep(100 * time.Millisecond)
+		gtest.Assert(array.Len(), 3)
+		time.Sleep(50 * time.Millisecond)
+		gtest.Assert(array.Len(), 3)
+		time.Sleep(100 * time.Millisecond)
+		gtest.Assert(array.Len(), 4)
 	})
 }