Prevent data race in test

mileusna · Aug 5, 2021 · 5f6295a · 5f6295a
1 parent 3aecc7a
commit 5f6295a
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/job_test.go b/job_test.go
@@ -102,16 +102,19 @@ func TestRunAll(t *testing.T) {
 
 	ctab := crontab.New()
 
-	if err := ctab.AddJob("* * * * *", func() { testN++ }); err != nil {
+	var wg sync.WaitGroup
+	wg.Add(2)
+
+	if err := ctab.AddJob("* * * * *", func() { testN++; wg.Done() }); err != nil {
 		t.Fatal(err)
 	}
 
-	if err := ctab.AddJob("* * * * *", func(s string) { testS = s }, "param"); err != nil {
+	if err := ctab.AddJob("* * * * *", func(s string) { testS = s; wg.Done() }, "param"); err != nil {
 		t.Fatal(err)
 	}
 
 	ctab.RunAll()
-	time.Sleep(time.Second)
+	wg.Wait()
 
 	if testN != 1 {
 		t.Error("func not executed on RunAll()")