update: 单例模式测试用例改造为并发以验证并发安全

Author: razertory

03_singleton/singleton_test.go

@@ -1,11 +1,26 @@
 package singleton
 
-import "testing"
+import (
+	"sync"
+	"testing"
+)
+
+const count = 100
 
 func TestSingleton(t *testing.T) {
-	ins1 := GetInstance()
-	ins2 := GetInstance()
-	if ins1 != ins2 {
-		t.Fatal("instance is not equal")
+	wg := sync.WaitGroup{}
+	wg.Add(count)
+	instances := [count]*Singleton{}
+	for i := 0; i < count; i++ {
+		go func(index int) {
+			instances[index] = GetInstance()
+			wg.Done()
+		}(i)
+	}
+	wg.Wait()
+	for i := 1; i < count; i++ {
+		if instances[i] != instances[i-1] {
+			t.Fatal("instance is not equal")
+		}
 	}
 }