Wait goroutines to finish when testing race conditions

CHANGELOG.md

@@ -4,6 +4,10 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](http://keepachangelog.com/en/1.0.0/)
 and this project adheres to [Semantic Versioning](http://semver.org/spec/v2.0.0.html).
 
+## [0.4.4] - 2018-03-01
+### Fixed
+- `TestRace` now waits all goroutines to finish.
+
 ## [0.4.3] - 2018-03-01
 ### Fixed
 - Bounds out of range error when trying to get a node with placeholder.
@@ -108,6 +112,7 @@ and this project adheres to [Semantic Versioning](http://semver.org/spec/v2.0.0.
 - This package's source code, including examples and tests.
 - Go dep files.
 
+[0.4.4]: https://github.com/gbrlsnchs/radix/compare/v0.4.3...v0.4.4
 [0.4.3]: https://github.com/gbrlsnchs/radix/compare/v0.4.2...v0.4.3
 [0.4.2]: https://github.com/gbrlsnchs/radix/compare/v0.4.1...v0.4.2
 [0.4.1]: https://github.com/gbrlsnchs/radix/compare/v0.4.0...v0.4.1

tree_test.go

@@ -2,8 +2,8 @@ package radix_test
 
 import (
 	"strconv"
+	"sync"
 	"testing"
-	"time"
 
 	. "github.com/gbrlsnchs/radix"
 
@@ -350,11 +350,14 @@ func TestRace(t *testing.T) {
 	}
 	tree := New("TestRace")
 	tree.Safe = true
+	var wg sync.WaitGroup
+
+	wg.Add(len(list) * 2)
 
 	for i, n := range list {
 		go func(i int, n string) {
 			tree.Add(n, i)
-			time.Sleep(time.Second * 3)
+			wg.Done()
 		}(i, n)
 	}
 
@@ -366,8 +369,9 @@ func TestRace(t *testing.T) {
 
 			tree.Del(n)
 			tree.Add(n, v)
-
-			time.Sleep(time.Second * 3)
+			wg.Done()
 		}(n)
 	}
+
+	wg.Wait()
 }