highperfocused/lnd
1
0
Fork 0
mirror of https://github.com/lightningnetwork/lnd.git synced 2025-08-28 06:32:18 +02:00
Code Issues Packages Projects Releases Wiki Activity

graph: fix flake in unit test

Browse Source
This commit is contained in:
ziggie
2025-02-10 13:11:35 +01:00
parent 6eb8f1f6e3
commit 6373d84baf
2 changed files with 15 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
graph/notifications_test.go
View File

@@ -176,13 +176,21 @@ type mockChain struct {
var _ lnwallet.BlockChainIO = (*mockChain)(nil)
func newMockChain(currentHeight uint32) *mockChain {
return &mockChain{
chain := &mockChain{
bestHeight: int32(currentHeight),
blocks: make(map[chainhash.Hash]*wire.MsgBlock),
utxos: make(map[wire.OutPoint]wire.TxOut),
blockIndex: make(map[uint32]chainhash.Hash),
blockHeightIndex: make(map[chainhash.Hash]uint32),
}
// Initialize the block index with the empty hash for the
// starting height.
startingHash := chainhash.Hash{}
chain.blockIndex[currentHeight] = startingHash
chain.blockHeightIndex[startingHash] = currentHeight
return chain
}
func (m *mockChain) setBestBlock(height int32) {
@@ -196,7 +204,11 @@ func (m *mockChain) GetBestBlock() (*chainhash.Hash, int32, error) {
m.RLock()
defer m.RUnlock()
blockHash := m.blockIndex[uint32(m.bestHeight)]
blockHash, exists := m.blockIndex[uint32(m.bestHeight)]
if !exists {
return nil, 0, fmt.Errorf("block at height %d not found",
m.bestHeight)
}
return &blockHash, m.bestHeight, nil
}