fix race in testcase

sanjayarun07 · May 4, 2018 · 0179a73 · 0179a73
1 parent 8b52715
commit 0179a73
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/pow/cpuminer/miner_test.go b/pow/cpuminer/miner_test.go
@@ -30,6 +30,8 @@ import (
 
 	"time"
 
+	"sync"
+
 	"github.com/thunderdb/ThunderDB/crypto/hash"
 )
 
@@ -52,10 +54,14 @@ func TestCPUMiner_HashBlock(t *testing.T) {
 	var (
 		err error
 	)
+	var wg sync.WaitGroup
+	wg.Add(1)
 	go func() {
 		err = miner.CalculateBlockNonce(block, *big.NewInt(0), diffWanted)
+		wg.Done()
 	}()
 	nonceFromCh := <-nonceCh
+	wg.Wait()
 
 	hash := hash.DoubleHashH(append(data, nonceFromCh.Nonce.Bytes()...))
 	if err != nil || nonceFromCh.Difficulty < diffWanted || hash.Difficulty() < diffWanted {