diff --git a/liquidity/autoloop_testcontext_test.go b/liquidity/autoloop_testcontext_test.go index 3ba0fc4..81430a4 100644 --- a/liquidity/autoloop_testcontext_test.go +++ b/liquidity/autoloop_testcontext_test.go @@ -135,7 +135,7 @@ func newAutoloopTestCtx(t *testing.T, parameters Parameters, cfg := &Config{ AutoloopTicker: ticker.NewForce(DefaultAutoloopTicker), - Restrictions: func(_ context.Context, swapType swap.Type) (*Restrictions, + Restrictions: func(_ context.Context, swapType swap.Type, initiator string) (*Restrictions, error) { if swapType == swap.TypeOut { diff --git a/liquidity/liquidity_test.go b/liquidity/liquidity_test.go index 4066d07..f3fae1e 100644 --- a/liquidity/liquidity_test.go +++ b/liquidity/liquidity_test.go @@ -147,7 +147,7 @@ func newTestConfig() (*Config, *test.LndMockServices) { ) return &Config{ - Restrictions: func(_ context.Context, _ swap.Type) (*Restrictions, + Restrictions: func(_ context.Context, _ swap.Type, initiator string) (*Restrictions, error) { return testRestrictions, nil @@ -1318,7 +1318,8 @@ type mockServer struct { } // Restrictions mocks a call to the server to get swap size restrictions. -func (m *mockServer) Restrictions(ctx context.Context, swapType swap.Type) ( +func (m *mockServer) Restrictions(ctx context.Context, swapType swap.Type, + initiator string) ( *Restrictions, error) { args := m.Called(ctx, swapType) diff --git a/liquidity/loopin_builder_test.go b/liquidity/loopin_builder_test.go index 3a3d16f..6d0af18 100644 --- a/liquidity/loopin_builder_test.go +++ b/liquidity/loopin_builder_test.go @@ -113,6 +113,7 @@ func TestLoopinBuildSwap(t *testing.T) { Amount: swapAmt, LastHop: &peer1, HtlcConfTarget: htlcConfTarget, + Initiator: autoloopSwapInitiator, } errPrecondition = status.Error(codes.FailedPrecondition, "failed") diff --git a/server_mock_test.go b/server_mock_test.go index 82f3c30..daef960 100644 --- a/server_mock_test.go +++ b/server_mock_test.go @@ -102,7 +102,7 @@ func (s *serverMock) NewLoopOutSwap(_ context.Context, swapHash lntypes.Hash, }, nil } -func (s *serverMock) GetLoopOutTerms(ctx context.Context) ( +func (s *serverMock) GetLoopOutTerms(ctx context.Context, initiator string) ( *LoopOutTerms, error) { return &LoopOutTerms{ @@ -114,7 +114,7 @@ func (s *serverMock) GetLoopOutTerms(ctx context.Context) ( } func (s *serverMock) GetLoopOutQuote(ctx context.Context, amt btcutil.Amount, - expiry int32, _ time.Time) (*LoopOutQuote, error) { + expiry int32, _ time.Time, initiator string) (*LoopOutQuote, error) { dest := [33]byte{1, 2, 3} @@ -209,7 +209,7 @@ func (s *serverMock) assertSwapCanceled(t *testing.T, details *outCancelDetails) require.Equal(t, details, <-s.cancelSwap) } -func (s *serverMock) GetLoopInTerms(ctx context.Context) ( +func (s *serverMock) GetLoopInTerms(ctx context.Context, initiator string) ( *LoopInTerms, error) { return &LoopInTerms{ @@ -219,7 +219,7 @@ func (s *serverMock) GetLoopInTerms(ctx context.Context) ( } func (s *serverMock) GetLoopInQuote(context.Context, btcutil.Amount, - route.Vertex, *route.Vertex, [][]zpay32.HopHint) (*LoopInQuote, error) { + route.Vertex, *route.Vertex, [][]zpay32.HopHint, string) (*LoopInQuote, error) { return &LoopInQuote{ SwapFee: testSwapFee,