修改用例

Author: zouyx

core/state_test.go

@@ -2,17 +2,30 @@ package core
 
 import (
 	"testing"
+	"sync"
+	"github.com/devfeel/dotweb/test"
 )
 
 // 以下为功能测试
 
 func Test_AddRequestCount_1(t *testing.T) {
-	var num uint64 = 1
-	var count uint64
-	for i := 0; i < 100; i++ {
-		count = GlobalState.AddRequestCount(num)
+	var wg sync.WaitGroup
+	wg.Add(2)
+
+	go addRequestCount(&wg,50)
+
+	go addRequestCount(&wg,60)
+
+	wg.Wait()
+
+	test.Equal(t,uint64(110),GlobalState.TotalRequestCount)
+}
+
+func addRequestCount(wg *sync.WaitGroup,count int)  {
+	for i := 0; i < count; i++ {
+		GlobalState.AddRequestCount(1)
 	}
-	t.Log("TotalRequestCount:", count)
+	wg.Add(-1)
 }
 
 func Test_AddRequestCount_2(t *testing.T) {
@@ -26,12 +39,16 @@ func Test_AddRequestCount_2(t *testing.T) {
 }
 
 func Test_AddErrorCount_1(t *testing.T) {
-	var num, count uint64
-	for i := 0; i < 100; i++ {
-		num = 1
-		count = GlobalState.AddErrorCount(num)
-	}
-	t.Log("TotalErrorCount:", count)
+	var wg sync.WaitGroup
+	wg.Add(2)
+
+	go addErrorCount(&wg,50)
+
+	go addErrorCount(&wg,60)
+
+	wg.Wait()
+
+	test.Equal(t,uint64(110),GlobalState.TotalErrorCount)
 }
 
 func Test_AddErrorCount_2(t *testing.T) {
@@ -43,6 +60,12 @@ func Test_AddErrorCount_2(t *testing.T) {
 	t.Log("TotalErrorCount:", count)
 }
 
+func addErrorCount(wg *sync.WaitGroup,count int)  {
+	for i := 0; i < count; i++ {
+		GlobalState.AddErrorCount(1)
+	}
+	wg.Add(-1)
+}
 // 以下是性能测试
 
 //基准测试