highperfocused/lnd
1
0
Fork 0
mirror of https://github.com/lightningnetwork/lnd.git synced 2025-07-03 12:03:37 +02:00
Code Issues Packages Projects Releases Wiki Activity

autopilot: add testDBGraph type

Browse Source 
Introduce a new type for testing code so the main databaseChannelGraph
type does not need to make various write calls to the
`graphdb.ChannelGraph` but the new testDBGraph type still can for tests.
This commit is contained in:
Elle Mouton
2025-02-05 12:13:28 +02:00
parent 1184c9eaf8
commit 3e5d807773
1 changed files with 12 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
autopilot/prefattach_test.go
View File

@ -30,6 +30,11 @@ type testGraph interface {
addRandNode() (*btcec.PublicKey, error) addRandNode() (*btcec.PublicKey, error)
} }
type testDBGraph struct {
db *graphdb.ChannelGraph
databaseChannelGraph
}
func newDiskChanGraph(t *testing.T) (testGraph, error) { func newDiskChanGraph(t *testing.T) (testGraph, error) {
backend, err := kvdb.GetBoltBackend(&kvdb.BoltBackendConfig{ backend, err := kvdb.GetBoltBackend(&kvdb.BoltBackendConfig{
DBPath: t.TempDir(), DBPath: t.TempDir(),
@ -44,12 +49,15 @@ func newDiskChanGraph(t *testing.T) (testGraph, error) {
graphDB, err := graphdb.NewChannelGraph(backend) graphDB, err := graphdb.NewChannelGraph(backend)
require.NoError(t, err) require.NoError(t, err)
return &databaseChannelGraph{ return &testDBGraph{
db: graphDB, db: graphDB,
databaseChannelGraph: databaseChannelGraph{
db: graphDB,
},
}, nil }, nil
} }
var _ testGraph = (*databaseChannelGraph)(nil) var _ testGraph = (*testDBGraph)(nil)
func newMemChanGraph(_ *testing.T) (testGraph, error) { func newMemChanGraph(_ *testing.T) (testGraph, error) {
return newMemChannelGraph(), nil return newMemChannelGraph(), nil
@ -378,7 +386,7 @@ func TestPrefAttachmentSelectSkipNodes(t *testing.T) {
// addRandChannel creates a new channel two target nodes. This function is // addRandChannel creates a new channel two target nodes. This function is
// meant to aide in the generation of random graphs for use within test cases // meant to aide in the generation of random graphs for use within test cases
// the exercise the autopilot package. // the exercise the autopilot package.
func (d *databaseChannelGraph) addRandChannel(node1, node2 *btcec.PublicKey, func (d *testDBGraph) addRandChannel(node1, node2 *btcec.PublicKey,
capacity btcutil.Amount) (*ChannelEdge, *ChannelEdge, error) { capacity btcutil.Amount) (*ChannelEdge, *ChannelEdge, error) {
fetchNode := func(pub *btcec.PublicKey) (*models.LightningNode, error) { fetchNode := func(pub *btcec.PublicKey) (*models.LightningNode, error) {
@ -522,7 +530,7 @@ func (d *databaseChannelGraph) addRandChannel(node1, node2 *btcec.PublicKey,
nil nil
} }
func (d *databaseChannelGraph) addRandNode() (*btcec.PublicKey, error) { func (d *testDBGraph) addRandNode() (*btcec.PublicKey, error) {
nodeKey, err := randKey() nodeKey, err := randKey()
if err != nil { if err != nil {
return nil, err return nil, err