fix race condition

shogo82148 · Apr 26, 2022 · 8e3b25e · 8e3b25e
1 parent fe8f9d8
commit 8e3b25e
Showing 1 changed file with 14 additions and 4 deletions.
diff --git a/xray/plugin_test.go b/xray/plugin_test.go
@@ -6,18 +6,28 @@ import (
 )
 
 func TestAddPlugin(t *testing.T) {
+	org := getPlugins()
+	defer func() {
+		muPlugins.Lock()
+		defer muPlugins.Unlock()
+		plugins = org
+	}()
 	before := len(getPlugins())
 
 	// test of races
 	const n = 10
 	var wg sync.WaitGroup
 	wg.Add(n)
 	for i := 0; i < n; i++ {
-		go AddPlugin(&xrayPlugin{
-			sdkVersion: getVersion(),
-		})
-		go getPlugins()
+		go func() {
+			defer wg.Done()
+			AddPlugin(&xrayPlugin{
+				sdkVersion: getVersion(),
+			})
+			getPlugins()
+		}()
 	}
+	wg.Wait()
 
 	after := len(getPlugins())