diff --git a/routing/pathfind_test.go b/routing/pathfind_test.go index 06bfa114..0c23bade 100644 --- a/routing/pathfind_test.go +++ b/routing/pathfind_test.go @@ -307,13 +307,13 @@ type testChannelPolicy struct { type testChannelEnd struct { Alias string - testChannelPolicy + *testChannelPolicy } func defaultTestChannelEnd(alias string, capacity btcutil.Amount) *testChannelEnd { return &testChannelEnd{ Alias: alias, - testChannelPolicy: testChannelPolicy{ + testChannelPolicy: &testChannelPolicy{ Expiry: 144, MinHTLC: lnwire.MilliSatoshi(1000), MaxHTLC: lnwire.NewMSatFromSatoshis(capacity), @@ -337,11 +337,11 @@ func symmetricTestChannel(alias1 string, alias2 string, capacity btcutil.Amount, Capacity: capacity, Node1: &testChannelEnd{ Alias: alias1, - testChannelPolicy: *policy, + testChannelPolicy: policy, }, Node2: &testChannelEnd{ Alias: alias2, - testChannelPolicy: *policy, + testChannelPolicy: policy, }, ChannelID: id, } @@ -490,45 +490,48 @@ func createTestGraphFromChannels(testChannels []*testChannel) (*testGraphInstanc return nil, err } - var msgFlags lnwire.ChanUpdateMsgFlags - if testChannel.Node1.MaxHTLC != 0 { - msgFlags = 1 - } - edgePolicy := &channeldb.ChannelEdgePolicy{ - SigBytes: testSig.Serialize(), - MessageFlags: msgFlags, - ChannelFlags: 0, - ChannelID: channelID, - LastUpdate: testTime, - TimeLockDelta: testChannel.Node1.Expiry, - MinHTLC: testChannel.Node1.MinHTLC, - MaxHTLC: testChannel.Node1.MaxHTLC, - FeeBaseMSat: testChannel.Node1.FeeBaseMsat, - FeeProportionalMillionths: testChannel.Node1.FeeRate, - } - if err := graph.UpdateEdgePolicy(edgePolicy); err != nil { - return nil, err + if testChannel.Node1.testChannelPolicy != nil { + var msgFlags lnwire.ChanUpdateMsgFlags + if testChannel.Node1.MaxHTLC != 0 { + msgFlags = 1 + } + edgePolicy := &channeldb.ChannelEdgePolicy{ + SigBytes: testSig.Serialize(), + MessageFlags: msgFlags, + ChannelFlags: 0, + ChannelID: channelID, + LastUpdate: testTime, + TimeLockDelta: testChannel.Node1.Expiry, + MinHTLC: testChannel.Node1.MinHTLC, + MaxHTLC: testChannel.Node1.MaxHTLC, + FeeBaseMSat: testChannel.Node1.FeeBaseMsat, + FeeProportionalMillionths: testChannel.Node1.FeeRate, + } + if err := graph.UpdateEdgePolicy(edgePolicy); err != nil { + return nil, err + } } - msgFlags = 0 - if testChannel.Node2.MaxHTLC != 0 { - msgFlags = 1 - } - edgePolicy = &channeldb.ChannelEdgePolicy{ - SigBytes: testSig.Serialize(), - MessageFlags: msgFlags, - ChannelFlags: lnwire.ChanUpdateDirection, - ChannelID: channelID, - LastUpdate: testTime, - TimeLockDelta: testChannel.Node2.Expiry, - MinHTLC: testChannel.Node2.MinHTLC, - MaxHTLC: testChannel.Node2.MaxHTLC, - FeeBaseMSat: testChannel.Node2.FeeBaseMsat, - FeeProportionalMillionths: testChannel.Node2.FeeRate, - } - - if err := graph.UpdateEdgePolicy(edgePolicy); err != nil { - return nil, err + if testChannel.Node2.testChannelPolicy != nil { + var msgFlags lnwire.ChanUpdateMsgFlags + if testChannel.Node2.MaxHTLC != 0 { + msgFlags = 1 + } + edgePolicy := &channeldb.ChannelEdgePolicy{ + SigBytes: testSig.Serialize(), + MessageFlags: msgFlags, + ChannelFlags: lnwire.ChanUpdateDirection, + ChannelID: channelID, + LastUpdate: testTime, + TimeLockDelta: testChannel.Node2.Expiry, + MinHTLC: testChannel.Node2.MinHTLC, + MaxHTLC: testChannel.Node2.MaxHTLC, + FeeBaseMSat: testChannel.Node2.FeeBaseMsat, + FeeProportionalMillionths: testChannel.Node2.FeeRate, + } + if err := graph.UpdateEdgePolicy(edgePolicy); err != nil { + return nil, err + } } channelID++