From 60a22ff09df85ce018c6c760c7e910c9f167434b Mon Sep 17 00:00:00 2001
From: Joost Jager <joost.jager@gmail.com>
Date: Tue, 8 Sep 2020 15:18:10 +0200
Subject: [PATCH] routing/test: parametrize test amount
---
routing/pathfind_test.go | 19 ++++++++++++++-----
1 file changed, 14 insertions(+), 5 deletions(-)
diff --git a/routing/pathfind_test.go b/routing/pathfind_test.go
index 6ba63aeb7..76c0e68e2 100644
--- a/routing/pathfind_test.go
+++ b/routing/pathfind_test.go
@@ -2461,6 +2461,7 @@ func TestProbabilityRouting(t *testing.T) {
p10, p11, p20 float64
minProbability float64
expectedChan uint64
+ amount btcutil.Amount
}{
// Test two variations with probabilities that should multiply
// to the same total route probability. In both cases the three
@@ -2474,12 +2475,14 @@ func TestProbabilityRouting(t *testing.T) {
p10: 0.8, p11: 0.5, p20: 0.7,
minProbability: 0.1,
expectedChan: 10,
+ amount: 100,
},
{
name: "three hop 2",
p10: 0.5, p11: 0.8, p20: 0.7,
minProbability: 0.1,
expectedChan: 10,
+ amount: 100,
},
// If the probability of the two hop route is increased, its
@@ -2492,6 +2495,7 @@ func TestProbabilityRouting(t *testing.T) {
p10: 0.5, p11: 0.8, p20: 0.85,
minProbability: 0.1,
expectedChan: 20,
+ amount: 100,
},
// If the same probabilities are used with a probability lower bound of
@@ -2502,6 +2506,7 @@ func TestProbabilityRouting(t *testing.T) {
p10: 0.8, p11: 0.5, p20: 0.7,
minProbability: 0.5,
expectedChan: 20,
+ amount: 100,
},
// With a probability limit above the probability of both routes, we
@@ -2512,21 +2517,24 @@ func TestProbabilityRouting(t *testing.T) {
p10: 0.8, p11: 0.5, p20: 0.7,
minProbability: 0.8,
expectedChan: 0,
+ amount: 100,
},
}
for _, tc := range testCases {
+ tc := tc
+
t.Run(tc.name, func(t *testing.T) {
testProbabilityRouting(
- t, tc.p10, tc.p11, tc.p20,
+ t, tc.amount, tc.p10, tc.p11, tc.p20,
tc.minProbability, tc.expectedChan,
)
})
}
}
-func testProbabilityRouting(t *testing.T, p10, p11, p20, minProbability float64,
- expectedChan uint64) {
+func testProbabilityRouting(t *testing.T, paymentAmt btcutil.Amount,
+ p10, p11, p20, minProbability float64, expectedChan uint64) {
t.Parallel()
@@ -2558,7 +2566,6 @@ func testProbabilityRouting(t *testing.T, p10, p11, p20, minProbability float64,
alias := ctx.testGraphInstance.aliasMap
- paymentAmt := lnwire.NewMSatFromSatoshis(100)
target := ctx.testGraphInstance.aliasMap["target"]
// Configure a probability source with the test parameters.
@@ -2586,7 +2593,9 @@ func testProbabilityRouting(t *testing.T, p10, p11, p20, minProbability float64,
MinProbability: minProbability,
}
- path, err := ctx.findPath(target, paymentAmt)
+ path, err := ctx.findPath(
+ target, lnwire.NewMSatFromSatoshis(paymentAmt),
+ )
if expectedChan == 0 {
if err != errNoPathFound {
t.Fatalf("expected no path found, but got %v", err)