mirrors/lnd
1
0
Fork 0
mirror of https://github.com/lightningnetwork/lnd.git synced 2025-03-04 01:36:24 +01:00
Code Issues Projects Releases Wiki Activity

routing: add quit channel to mockChainView

Browse source
This commit is contained in:
Olaoluwa Osuntokun 2017-11-06 15:57:19 -08:00
parent 4f32ee94ea
commit fea8cbf920
No known key found for this signature in database
GPG key ID: 964EA263DD637C21
1 changed files with 13 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
routing/notifications_test.go
View file

@ -213,6 +213,8 @@ type mockChainView struct {
staleBlocks chan *chainview.FilteredBlock
filter map[wire.OutPoint]struct{}
quit chan struct{}
}
// A compile time check to ensure mockChainView implements the
@ -224,6 +226,7 @@ func newMockChainView() *mockChainView {
newBlocks: make(chan *chainview.FilteredBlock, 10),
staleBlocks: make(chan *chainview.FilteredBlock, 10),
filter: make(map[wire.OutPoint]struct{}),
quit: make(chan struct{}),
}
}
@ -244,10 +247,14 @@ func (m *mockChainView) notifyBlock(hash chainhash.Hash, height uint32,
m.RLock()
defer m.RUnlock()
m.newBlocks <- &chainview.FilteredBlock{
select {
case m.newBlocks <- &chainview.FilteredBlock{
Hash: hash,
Height: height,
Transactions: txns,
}:
case <-m.quit:
return
}
}
@ -257,10 +264,14 @@ func (m *mockChainView) notifyStaleBlock(hash chainhash.Hash, height uint32,
m.RLock()
defer m.RUnlock()
m.staleBlocks <- &chainview.FilteredBlock{
select {
case m.staleBlocks <- &chainview.FilteredBlock{
Hash: hash,
Height: height,
Transactions: txns,
}:
case <-m.quit:
return
}
}