Mirroristas/kaspad
2
0
Fork 0
mirror of https://github.com/kaspanet/kaspad.git synced 2025-07-06 04:42:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

blockchain: Fix race condition in notifications_test

Browse Source
This commit is contained in:
Jim Posen 2017-08-14 20:48:32 -07:00 committed by Dave Collins
parent b71d6c3010
commit ca9baf77ec
1 changed files with 15 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
blockchain/notifications_test.go
View File

@ -6,7 +6,6 @@ package blockchain_test
import ( import (
"testing" "testing"
"time"
"github.com/btcsuite/btcd/blockchain" "github.com/btcsuite/btcd/blockchain"
"github.com/btcsuite/btcd/chaincfg" "github.com/btcsuite/btcd/chaincfg"
@ -27,24 +26,26 @@ func TestNotifications(t *testing.T) {
} }
defer teardownFunc() defer teardownFunc()
notifications := make(chan blockchain.Notification) notificationCount := 0
chain.Subscribe(func(notification *blockchain.Notification) {
go func() { callback := func(notification *blockchain.Notification) {
notifications <- *notification if notification.Type == blockchain.NTBlockAccepted {
}() notificationCount++
}) }
}
// Register callback 3 times then assert it is called three times
chain.Subscribe(callback)
chain.Subscribe(callback)
chain.Subscribe(callback)
_, _, err = chain.ProcessBlock(blocks[1], blockchain.BFNone) _, _, err = chain.ProcessBlock(blocks[1], blockchain.BFNone)
if err != nil { if err != nil {
t.Fatalf("ProcessBlock fail on block 1: %v\n", err) t.Fatalf("ProcessBlock fail on block 1: %v\n", err)
} }
select { if notificationCount != 3 {
case notification := <-notifications: t.Fatalf("Expected notification callback to be executed 3 times, found %d",
if notification.Type != blockchain.NTBlockAccepted { notificationCount)
t.Errorf("Expected NTBlockAccepted notification, got %v", notification.Type)
}
case <-time.After(time.Second):
t.Error("Expected blockchain notification callback to fire")
} }
} }