test: integration tests for premium

This commit is to modify and add integration tests for the addition of premium.
ElementsProject · Dec 23, 2023 · 8870169 · 8870169
1 parent 1ec38e5
commit 8870169
Show file tree

Hide file tree

Showing 10 changed files with 1,564 additions and 984 deletions.
diff --git a/peerswaprpc/server.go b/peerswaprpc/server.go
@@ -144,7 +144,7 @@ func (p *PeerswapServer) SwapOut(ctx context.Context, request *SwapOutRequest) (
 		return nil, fmt.Errorf("peer is not connected")
 	}
 
-	swapOut, err := p.swaps.SwapOut(peerId, request.Asset, shortId.String(), pk, request.SwapAmount)
+	swapOut, err := p.swaps.SwapOut(peerId, request.Asset, shortId.String(), pk, request.SwapAmount, request.GetPremiumLimit())
 	if err != nil {
 		return nil, err
 	}
@@ -268,7 +268,7 @@ func (p *PeerswapServer) SwapIn(ctx context.Context, request *SwapInRequest) (*S
 		return nil, fmt.Errorf("peer is not connected")
 	}
 
-	swapIn, err := p.swaps.SwapIn(peerId, request.Asset, shortId.String(), pk, request.SwapAmount)
+	swapIn, err := p.swaps.SwapIn(peerId, request.Asset, shortId.String(), pk, request.SwapAmount, request.GetPremiumLimit())
 	if err != nil {
 		return nil, err
 	}
@@ -376,10 +376,12 @@ func (p *PeerswapServer) ListPeers(ctx context.Context, request *ListPeersReques
 			}
 
 			peerSwapPeers = append(peerSwapPeers, &PeerSwapPeer{
-				NodeId:          v.PubKey,
-				SwapsAllowed:    poll.PeerAllowed,
-				SupportedAssets: poll.Assets,
-				Channels:        getPeerSwapChannels(v.PubKey, channelRes.Channels),
+				NodeId:                v.PubKey,
+				SwapsAllowed:          poll.PeerAllowed,
+				SwapInPremiumRatePpm:  poll.SwapInPremiumRatePPM,
+				SwapOutPremiumRatePpm: poll.SwapOutPremiumRatePPM,
+				SupportedAssets:       poll.Assets,
+				Channels:              getPeerSwapChannels(v.PubKey, channelRes.Channels),
 				AsSender: &SwapStats{
 					SwapsOut: SenderSwapsOut,
 					SwapsIn:  SenderSwapsIn,
@@ -571,6 +573,7 @@ func PrettyprintFromServiceSwap(swap *swap.SwapStateMachine) *PrettyPrintSwap {
 		ClaimTxId:       swap.Data.ClaimTxId,
 		CancelMessage:   swap.Data.GetCancelMessage(),
 		LndChanId:       lnd_chan_id,
+		PremiumAmount:   swap.Data.GetPremium(),
 	}
 }
 

diff --git a/policy/policy_test.go b/policy/policy_test.go
@@ -84,12 +84,14 @@ func Test_Reload(t *testing.T) {
 	policy, err := create(strings.NewReader(conf))
 	assert.NoError(t, err)
 	assert.EqualValues(t, &Policy{
-		ReserveOnchainMsat: defaultReserveOnchainMsat,
-		PeerAllowlist:      []string{peer1, peer2},
-		SuspiciousPeerList: defaultSuspiciousPeerList,
-		AcceptAllPeers:     accept,
-		MinSwapAmountMsat:  defaultMinSwapAmountMsat,
-		AllowNewSwaps:      defaultAllowNewSwaps,
+		ReserveOnchainMsat:    defaultReserveOnchainMsat,
+		PeerAllowlist:         []string{peer1, peer2},
+		SuspiciousPeerList:    defaultSuspiciousPeerList,
+		AcceptAllPeers:        accept,
+		MinSwapAmountMsat:     defaultMinSwapAmountMsat,
+		AllowNewSwaps:         defaultAllowNewSwaps,
+		SwapInPremiumRatePPM:  defaultSwapInPremiumRatePPM,
+		SwapOutPremiumRatePPM: defaultSwapOutPremiumRatePPM,
 	}, policy)
 
 	newPeer := "new_peer"
@@ -98,12 +100,14 @@ func Test_Reload(t *testing.T) {
 	err = policy.reload(strings.NewReader(newConf))
 	assert.NoError(t, err)
 	assert.EqualValues(t, &Policy{
-		ReserveOnchainMsat: defaultReserveOnchainMsat,
-		PeerAllowlist:      []string{newPeer},
-		SuspiciousPeerList: []string{newPeer},
-		AcceptAllPeers:     defaultAcceptAllPeers,
-		MinSwapAmountMsat:  defaultMinSwapAmountMsat,
-		AllowNewSwaps:      defaultAllowNewSwaps,
+		ReserveOnchainMsat:    defaultReserveOnchainMsat,
+		PeerAllowlist:         []string{newPeer},
+		SuspiciousPeerList:    []string{newPeer},
+		AcceptAllPeers:        defaultAcceptAllPeers,
+		MinSwapAmountMsat:     defaultMinSwapAmountMsat,
+		AllowNewSwaps:         defaultAllowNewSwaps,
+		SwapInPremiumRatePPM:  defaultSwapInPremiumRatePPM,
+		SwapOutPremiumRatePPM: defaultSwapOutPremiumRatePPM,
 	}, policy)
 }
 
@@ -121,12 +125,14 @@ func Test_Reload_NoOverrideOnError(t *testing.T) {
 	policy, err := create(strings.NewReader(conf))
 	assert.NoError(t, err)
 	assert.EqualValues(t, &Policy{
-		ReserveOnchainMsat: defaultReserveOnchainMsat,
-		PeerAllowlist:      []string{peer1, peer2},
-		SuspiciousPeerList: defaultSuspiciousPeerList,
-		AcceptAllPeers:     accept,
-		MinSwapAmountMsat:  defaultMinSwapAmountMsat,
-		AllowNewSwaps:      defaultAllowNewSwaps,
+		ReserveOnchainMsat:    defaultReserveOnchainMsat,
+		PeerAllowlist:         []string{peer1, peer2},
+		SuspiciousPeerList:    defaultSuspiciousPeerList,
+		AcceptAllPeers:        accept,
+		MinSwapAmountMsat:     defaultMinSwapAmountMsat,
+		AllowNewSwaps:         defaultAllowNewSwaps,
+		SwapInPremiumRatePPM:  defaultSwapInPremiumRatePPM,
+		SwapOutPremiumRatePPM: defaultSwapOutPremiumRatePPM,
 	}, policy)
 
 	// copy policy

diff --git a/swap/service_test.go b/swap/service_test.go
@@ -257,7 +257,7 @@ func TestMessageFromUnexpectedPeer(t *testing.T) {
 	if err != nil {
 		t.Fatal(err)
 	}
-	aliceSwap, err := aliceSwapService.SwapOut(peer, "btc", channelId, initiator, amount, 0)
+	aliceSwap, err := aliceSwapService.SwapOut(peer, "btc", channelId, initiator, amount, 1000)
 	if err != nil {
 		t.Fatalf(" error swapping oput %v: ", err)
 	}