highperfocused/lnd
1
0
Fork 0
mirror of https://github.com/lightningnetwork/lnd.git synced 2025-12-04 18:01:57 +01:00
Code Issues Packages Projects Releases Wiki Activity

routing: add option to force import MC pair history

Browse Source 
This commit adds the `force` flag to the `XImportMissionControl` RPC
which allows skipping rules around the pair import except for what is
mandatory to make values meaningful. This can be useful for when clients
would like to forcibly override MC state in order to manipulate routing
results.
This commit is contained in:
Andras Banki-Horvath
2022-01-19 17:01:09 +01:00
parent 3829385073
commit 23c157a07b
8 changed files with 479 additions and 400 deletions
Download Patch File Download Diff File Expand all files Collapse all files

59
routing/missioncontrol_state_test.go
View File

@@ -5,11 +5,12 @@ import (
"time"
"github.com/lightningnetwork/lnd/routing/route"
"github.com/stretchr/testify/require"
)
// TestMissionControlStateFailureResult tests setting failure results on the
// mission control state.
func TestMissionControlStateFailureResult(t *testing.T) {
// TestMissionControlStateSetLastPairResult tests setting mission control state
// pair results.
func TestMissionControlStateSetLastPairResult(t *testing.T) {
const minFailureRelaxInterval = time.Minute
state := newMissionControlState(minFailureRelaxInterval)
@@ -20,7 +21,9 @@ func TestMissionControlStateFailureResult(t *testing.T) {
)
// Report a 1000 sat failure.
state.setLastPairResult(from, to, timestamp, &pairResult{amt: 1000})
state.setLastPairResult(
from, to, timestamp, &pairResult{amt: 1000}, false,
)
result, _ := state.getLastPairResult(from)
if result[to].FailAmt != 1000 {
t.Fatalf("unexpected fail amount %v", result[to].FailAmt)
@@ -29,7 +32,9 @@ func TestMissionControlStateFailureResult(t *testing.T) {
// Report an 1100 sat failure one hour later. It is expected to
// overwrite the previous failure.
timestamp = timestamp.Add(time.Hour)
state.setLastPairResult(from, to, timestamp, &pairResult{amt: 1100})
state.setLastPairResult(
from, to, timestamp, &pairResult{amt: 1100}, false,
)
result, _ = state.getLastPairResult(from)
if result[to].FailAmt != 1100 {
t.Fatalf("unexpected fail amount %v", result[to].FailAmt)
@@ -39,9 +44,51 @@ func TestMissionControlStateFailureResult(t *testing.T) {
// the failure amount is too soon after the previous failure, the result
// is not applied.
timestamp = timestamp.Add(time.Second)
state.setLastPairResult(from, to, timestamp, &pairResult{amt: 1200})
state.setLastPairResult(
from, to, timestamp, &pairResult{amt: 1200}, false,
)
result, _ = state.getLastPairResult(from)
if result[to].FailAmt != 1100 {
t.Fatalf("unexpected fail amount %v", result[to].FailAmt)
}
// Roll back time 1 second to test forced import.
timestamp = testTime
state.setLastPairResult(
from, to, timestamp, &pairResult{amt: 999}, true,
)
result, _ = state.getLastPairResult(from)
require.Equal(t, 999, int(result[to].FailAmt))
// Report an 1500 sat success.
timestamp = timestamp.Add(time.Second)
state.setLastPairResult(
from, to, timestamp, &pairResult{amt: 1500, success: true}, false,
)
result, _ = state.getLastPairResult(from)
// We don't expect the failtime to change, only the fail amount, we
// expect however change of both the success time and the success amount.
expected := TimedPairResult{
FailTime: timestamp.Add(-time.Second),
FailAmt: 1501,
SuccessTime: timestamp,
SuccessAmt: 1500,
}
require.Equal(t, expected, result[to])
// Again roll back time to test forced import.
state.setLastPairResult(
from, to, testTime, &pairResult{amt: 50, success: true}, true,
)
result, _ = state.getLastPairResult(from)
// We don't expect the failtime to change, only the fail amount, we
// expect however change of both the success time and the success amount.
expected = TimedPairResult{
FailTime: timestamp.Add(-time.Second),
FailAmt: 51,
SuccessTime: testTime,
SuccessAmt: 50,
}
require.Equal(t, expected, result[to])
}