mirror of
https://github.com/kaspanet/kaspad.git
synced 2025-10-14 00:59:33 +00:00
5fb220c38a
* [DEV-105] use utxodiff in mempool instead of utxoview * [DEV-105] get rid of utxoview * [DEV-105] fix tests to use utxoset * [DEV-105] remove utxoview type * [DEV-105] move DagSetup to test_utils.go * [DEV-105] add comments and add blockdag/test_utils_test.go * [DEV-105] add restoreInputs arg to removeTransaction * [DEV-105] give more descriptive names to vars * [DEV-115] close txChan outside of HandleNewBlock * [DEV-105] rename DagSetup -> DAGSetup
52 lines
1.3 KiB
Go
52 lines
1.3 KiB
Go
// Copyright (c) 2017 The btcsuite developers
|
|
// Use of this source code is governed by an ISC
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package blockdag
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/daglabs/btcd/dagconfig"
|
|
)
|
|
|
|
// TestNotifications ensures that notification callbacks are fired on events.
|
|
func TestNotifications(t *testing.T) {
|
|
blocks, err := loadBlocks("blk_0_to_4.dat")
|
|
if err != nil {
|
|
t.Fatalf("Error loading file: %v\n", err)
|
|
}
|
|
|
|
// Create a new database and chain instance to run tests against.
|
|
chain, teardownFunc, err := DAGSetup("notifications",
|
|
&dagconfig.MainNetParams)
|
|
if err != nil {
|
|
t.Fatalf("Failed to setup chain instance: %v", err)
|
|
}
|
|
defer teardownFunc()
|
|
|
|
notificationCount := 0
|
|
callback := func(notification *Notification) {
|
|
if notification.Type == NTBlockAccepted {
|
|
notificationCount++
|
|
}
|
|
}
|
|
|
|
// Register callback multiple times then assert it is called that many
|
|
// times.
|
|
const numSubscribers = 3
|
|
for i := 0; i < numSubscribers; i++ {
|
|
chain.Subscribe(callback)
|
|
}
|
|
|
|
_, err = chain.ProcessBlock(blocks[1], BFNone)
|
|
if err != nil {
|
|
t.Fatalf("ProcessBlock fail on block 1: %v\n", err)
|
|
}
|
|
|
|
if notificationCount != numSubscribers {
|
|
t.Fatalf("Expected notification callback to be executed %d "+
|
|
"times, found %d", numSubscribers, notificationCount)
|
|
}
|
|
}
|