diff --git a/routing/integrated_routing_context_test.go b/routing/integrated_routing_context_test.go index d421c62fc..418b380ca 100644 --- a/routing/integrated_routing_context_test.go +++ b/routing/integrated_routing_context_test.go @@ -81,7 +81,7 @@ type htlcAttempt struct { // testPayment launches a test payment and asserts that it is completed after // the expected number of attempts. -func (c *integratedRoutingContext) testPayment() []htlcAttempt { +func (c *integratedRoutingContext) testPayment() ([]htlcAttempt, error) { var ( nextPid uint64 attempts []htlcAttempt @@ -150,7 +150,7 @@ func (c *integratedRoutingContext) testPayment() []htlcAttempt { c.amt, lnwire.MaxMilliSatoshi, 0, 0, ) if err != nil { - c.t.Fatal(err) + return nil, err } // 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)) - return attempts + return attempts, nil } // getNodeIndex returns the zero-based index of the given node in the route. diff --git a/routing/integrated_routing_test.go b/routing/integrated_routing_test.go index 0362f208c..1e77ce810 100644 --- a/routing/integrated_routing_test.go +++ b/routing/integrated_routing_test.go @@ -50,7 +50,10 @@ func TestProbabilityExtrapolation(t *testing.T) { // a specific number of attempts to safe-guard against accidental // modifications anywhere in the chain of components that is involved in // this test. - attempts := ctx.testPayment() + attempts, err := ctx.testPayment() + if err != nil { + t.Fatalf("payment failed: %v", err) + } if len(attempts) != 5 { 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 // first before switching to the paid channel. ctx.mcCfg.AprioriWeight = 1 - attempts = ctx.testPayment() + attempts, err = ctx.testPayment() + if err != nil { + t.Fatalf("payment failed: %v", err) + } if len(attempts) != 11 { t.Fatalf("expected 11 attempts, but needed %v", len(attempts)) }