highperfocused/lnd
1
0
Fork 0
mirror of https://github.com/lightningnetwork/lnd.git synced 2025-05-24 10:40:10 +02:00
Code Issues Packages Projects Releases Wiki Activity

channeldb/test: refactor TestGraphCacheForEachNodeChannel

Browse Source
This commit is contained in:
Joost Jager 2023-11-19 11:09:21 +01:00
parent 9d4251c18d
commit d97e7d30fb
1 changed files with 16 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
channeldb/graph_test.go
View File

@ -3934,19 +3934,24 @@ func TestGraphCacheForEachNodeChannel(t *testing.T) {
// Add the channel, but only insert a single edge into the graph. // Add the channel, but only insert a single edge into the graph.
require.NoError(t, graph.AddChannelEdge(edgeInfo)) require.NoError(t, graph.AddChannelEdge(edgeInfo))
// We should be able to accumulate the single channel added, even getSingleChannel := func() *DirectedChannel {
// though we have a nil edge policy here. var ch *DirectedChannel
var numChans int
err = graph.ForEachNodeDirectedChannel(nil, node1.PubKeyBytes, err = graph.ForEachNodeDirectedChannel(nil, node1.PubKeyBytes,
func(_ *DirectedChannel) error { func(c *DirectedChannel) error {
numChans++ require.Nil(t, ch)
ch = c
return nil return nil
}, },
) )
require.NoError(t, err) require.NoError(t, err)
require.Equal(t, numChans, 1) return ch
}
// We should be able to accumulate the single channel added, even
// though we have a nil edge policy here.
require.NotNil(t, getSingleChannel())
} }
// TestGraphLoading asserts that the cache is properly reconstructed after a // TestGraphLoading asserts that the cache is properly reconstructed after a