routing/test: return pathfinding error

This commit is contained in:
Joost Jager 2020-03-18 10:51:07 +01:00
parent 0f8eb80965
commit e5c7e9a38c
No known key found for this signature in database
GPG Key ID: A61B9D4C393C59C7
2 changed files with 11 additions and 5 deletions

View File

@ -81,7 +81,7 @@ type htlcAttempt struct {
// testPayment launches a test payment and asserts that it is completed after // testPayment launches a test payment and asserts that it is completed after
// the expected number of attempts. // the expected number of attempts.
func (c *integratedRoutingContext) testPayment() []htlcAttempt { func (c *integratedRoutingContext) testPayment() ([]htlcAttempt, error) {
var ( var (
nextPid uint64 nextPid uint64
attempts []htlcAttempt attempts []htlcAttempt
@ -150,7 +150,7 @@ func (c *integratedRoutingContext) testPayment() []htlcAttempt {
c.amt, lnwire.MaxMilliSatoshi, 0, 0, c.amt, lnwire.MaxMilliSatoshi, 0, 0,
) )
if err != nil { if err != nil {
c.t.Fatal(err) return nil, err
} }
// Send out the htlc on the mock graph. // Send out the htlc on the mock graph.
@ -199,7 +199,7 @@ func (c *integratedRoutingContext) testPayment() []htlcAttempt {
c.t.Logf("Payment attempts: %v\n", len(attempts)) c.t.Logf("Payment attempts: %v\n", len(attempts))
return attempts return attempts, nil
} }
// getNodeIndex returns the zero-based index of the given node in the route. // getNodeIndex returns the zero-based index of the given node in the route.

View File

@ -50,7 +50,10 @@ func TestProbabilityExtrapolation(t *testing.T) {
// a specific number of attempts to safe-guard against accidental // a specific number of attempts to safe-guard against accidental
// modifications anywhere in the chain of components that is involved in // modifications anywhere in the chain of components that is involved in
// this test. // this test.
attempts := ctx.testPayment() attempts, err := ctx.testPayment()
if err != nil {
t.Fatalf("payment failed: %v", err)
}
if len(attempts) != 5 { if len(attempts) != 5 {
t.Fatalf("expected 5 attempts, but needed %v", len(attempts)) t.Fatalf("expected 5 attempts, but needed %v", len(attempts))
} }
@ -59,7 +62,10 @@ func TestProbabilityExtrapolation(t *testing.T) {
// of data from other channels), all ten bad channels will be tried // of data from other channels), all ten bad channels will be tried
// first before switching to the paid channel. // first before switching to the paid channel.
ctx.mcCfg.AprioriWeight = 1 ctx.mcCfg.AprioriWeight = 1
attempts = ctx.testPayment() attempts, err = ctx.testPayment()
if err != nil {
t.Fatalf("payment failed: %v", err)
}
if len(attempts) != 11 { if len(attempts) != 11 {
t.Fatalf("expected 11 attempts, but needed %v", len(attempts)) t.Fatalf("expected 11 attempts, but needed %v", len(attempts))
} }