mirror of
https://github.com/lightningequipment/circuitbreaker.git
synced 2026-08-13 12:33:02 +02:00
101 lines
2.2 KiB
Go
101 lines
2.2 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/lightningnetwork/lnd/lnrpc/routerrpc"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestProcess(t *testing.T) {
|
|
p := newProcess()
|
|
|
|
cfg := &config{
|
|
groupConfig: groupConfig{
|
|
MaxPendingHtlcs: 2,
|
|
},
|
|
BaseSatPerHr: 1,
|
|
RatePpmPerHr: 5,
|
|
ReportingInterval: time.Minute * 1,
|
|
}
|
|
|
|
client := newLndclientMock()
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
|
|
exit := make(chan error)
|
|
go func() {
|
|
exit <- p.run(ctx, client, cfg)
|
|
}()
|
|
|
|
key := &routerrpc.CircuitKey{
|
|
ChanId: 2,
|
|
HtlcId: 5,
|
|
}
|
|
client.htlcInterceptorRequests <- &routerrpc.ForwardHtlcInterceptRequest{
|
|
IncomingCircuitKey: key,
|
|
}
|
|
|
|
resp := <-client.htlcInterceptorResponses
|
|
require.Equal(t, routerrpc.ResolveHoldForwardAction_RESUME, resp.Action)
|
|
|
|
client.htlcEvents <- &routerrpc.HtlcEvent{
|
|
EventType: routerrpc.HtlcEvent_FORWARD,
|
|
IncomingChannelId: key.ChanId,
|
|
IncomingHtlcId: key.HtlcId,
|
|
Event: &routerrpc.HtlcEvent_SettleEvent{},
|
|
}
|
|
|
|
time.Sleep(time.Second)
|
|
|
|
cancel()
|
|
require.NoError(t, <-exit)
|
|
}
|
|
|
|
func TestRateLimit(t *testing.T) {
|
|
p := newProcess()
|
|
|
|
cfg := &config{
|
|
groupConfig: groupConfig{
|
|
HtlcMinInterval: time.Minute,
|
|
HtlcBurstSize: 2,
|
|
},
|
|
}
|
|
|
|
client := newLndclientMock()
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
|
|
exit := make(chan error)
|
|
go func() {
|
|
exit <- p.run(ctx, client, cfg)
|
|
}()
|
|
|
|
key := &routerrpc.CircuitKey{
|
|
ChanId: 2,
|
|
HtlcId: 5,
|
|
}
|
|
interceptReq := &routerrpc.ForwardHtlcInterceptRequest{
|
|
IncomingCircuitKey: key,
|
|
}
|
|
|
|
// First htlc accepted.
|
|
client.htlcInterceptorRequests <- interceptReq
|
|
resp := <-client.htlcInterceptorResponses
|
|
require.Equal(t, routerrpc.ResolveHoldForwardAction_RESUME, resp.Action)
|
|
|
|
// Second htlc right after is also accepted because of burst size 2.
|
|
client.htlcInterceptorRequests <- interceptReq
|
|
resp = <-client.htlcInterceptorResponses
|
|
require.Equal(t, routerrpc.ResolveHoldForwardAction_RESUME, resp.Action)
|
|
|
|
// Third htlc again right after should be rejected.
|
|
client.htlcInterceptorRequests <- interceptReq
|
|
resp = <-client.htlcInterceptorResponses
|
|
require.Equal(t, routerrpc.ResolveHoldForwardAction_FAIL, resp.Action)
|
|
|
|
cancel()
|
|
require.NoError(t, <-exit)
|
|
}
|