109 lines
2.4 KiB
Go
109 lines
2.4 KiB
Go
package routing
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/lightningnetwork/lnd/lnwire"
|
|
"github.com/lightningnetwork/lnd/routing/route"
|
|
)
|
|
|
|
var (
|
|
mcTestNode = route.Vertex{}
|
|
mcTestEdge = EdgeLocator{
|
|
ChannelID: 123,
|
|
}
|
|
mcTestTime = time.Date(2018, time.January, 9, 14, 00, 00, 0, time.UTC)
|
|
)
|
|
|
|
type mcTestContext struct {
|
|
t *testing.T
|
|
mc *MissionControl
|
|
now time.Time
|
|
}
|
|
|
|
func createMcTestContext(t *testing.T) *mcTestContext {
|
|
ctx := &mcTestContext{
|
|
t: t,
|
|
now: mcTestTime,
|
|
}
|
|
|
|
mc := NewMissionControl(
|
|
nil, nil, nil, &MissionControlConfig{
|
|
PenaltyHalfLife: 30 * time.Minute,
|
|
AprioriHopProbability: 0.8,
|
|
},
|
|
)
|
|
|
|
mc.now = func() time.Time { return ctx.now }
|
|
ctx.mc = mc
|
|
|
|
return ctx
|
|
}
|
|
|
|
// Assert that mission control returns a probability for an edge.
|
|
func (ctx *mcTestContext) expectP(amt lnwire.MilliSatoshi,
|
|
expected float64) {
|
|
|
|
ctx.t.Helper()
|
|
|
|
p := ctx.mc.getEdgeProbability(mcTestNode, mcTestEdge, amt)
|
|
if p != expected {
|
|
ctx.t.Fatalf("unexpected probability %v", p)
|
|
}
|
|
}
|
|
|
|
// TestMissionControl tests mission control probability estimation.
|
|
func TestMissionControl(t *testing.T) {
|
|
ctx := createMcTestContext(t)
|
|
|
|
ctx.now = testTime
|
|
|
|
testTime := time.Date(2018, time.January, 9, 14, 00, 00, 0, time.UTC)
|
|
|
|
testNode := route.Vertex{}
|
|
testEdge := edge{
|
|
channel: 123,
|
|
}
|
|
|
|
// Initial probability is expected to be 1.
|
|
ctx.expectP(1000, 0.8)
|
|
|
|
// Expect probability to be zero after reporting the edge as failed.
|
|
ctx.mc.reportEdgeFailure(testEdge, 1000)
|
|
ctx.expectP(1000, 0)
|
|
|
|
// As we reported with a min penalization amt, a lower amt than reported
|
|
// should be unaffected.
|
|
ctx.expectP(500, 0.8)
|
|
|
|
// Edge decay started.
|
|
ctx.now = testTime.Add(30 * time.Minute)
|
|
ctx.expectP(1000, 0.4)
|
|
|
|
// Edge fails again, this time without a min penalization amt. The edge
|
|
// should be penalized regardless of amount.
|
|
ctx.mc.reportEdgeFailure(testEdge, 0)
|
|
ctx.expectP(1000, 0)
|
|
ctx.expectP(500, 0)
|
|
|
|
// Edge decay started.
|
|
ctx.now = testTime.Add(60 * time.Minute)
|
|
ctx.expectP(1000, 0.4)
|
|
|
|
// A node level failure should bring probability of every channel back
|
|
// to zero.
|
|
ctx.mc.reportVertexFailure(testNode)
|
|
ctx.expectP(1000, 0)
|
|
|
|
// Check whether history snapshot looks sane.
|
|
history := ctx.mc.GetHistorySnapshot()
|
|
if len(history.Nodes) != 1 {
|
|
t.Fatal("unexpected number of nodes")
|
|
}
|
|
|
|
if len(history.Nodes[0].Channels) != 1 {
|
|
t.Fatal("unexpected number of channels")
|
|
}
|
|
}
|