itest: estimate on-chain fee with selected inputs
This commit is contained in:
2 files changed
+240
No files matched your search
@@ -759,6 +759,14 @@ var allTestCases = []*lntest.TestCase{
|
|||||||
Name: "estimate fee",
|
Name: "estimate fee",
|
||||||
TestFunc: testEstimateFee,
|
TestFunc: testEstimateFee,
|
||||||
},
|
},
|
||||||
|
{
|
||||||
|
Name: "estimate on chain fee with selected inputs",
|
||||||
|
TestFunc: testEstimateOnChainFeeWithSelectedInputs,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Name: "estimate on chain fee auto selected inputs",
|
||||||
|
TestFunc: testEstimateOnChainFeeAutoSelectedInputs,
|
||||||
|
},
|
||||||
}
|
}
|
||||||
|
|
||||||
// appendPrefixed is used to add a prefix to each test name in the subtests
|
// appendPrefixed is used to add a prefix to each test name in the subtests
|
||||||
|
|||||||
@@ -0,0 +1,232 @@
|
|||||||
|
package itest
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
|
||||||
|
"github.com/btcsuite/btcd/btcutil"
|
||||||
|
"github.com/lightningnetwork/lnd/lnrpc"
|
||||||
|
"github.com/lightningnetwork/lnd/lnrpc/walletrpc"
|
||||||
|
"github.com/lightningnetwork/lnd/lntest"
|
||||||
|
"github.com/stretchr/testify/require"
|
||||||
|
)
|
||||||
|
|
||||||
|
// testEstimateOnChainFeeWithSelectedInputs tests that the EstimateFee RPC
|
||||||
|
// with specific input selection produces an accurate fee estimate that matches
|
||||||
|
// the actual fee when sending coins with the same inputs.
|
||||||
|
func testEstimateOnChainFeeWithSelectedInputs(ht *lntest.HarnessTest) {
|
||||||
|
// Create a new node for this test.
|
||||||
|
alice := ht.NewNode("Alice", nil)
|
||||||
|
|
||||||
|
// Fund Alice with multiple UTXOs of different amounts to give us
|
||||||
|
// several inputs to choose from.
|
||||||
|
const (
|
||||||
|
utxo1Amount = btcutil.Amount(500_000)
|
||||||
|
utxo2Amount = btcutil.Amount(300_000)
|
||||||
|
utxo3Amount = btcutil.Amount(200_000)
|
||||||
|
targetConf = 2
|
||||||
|
)
|
||||||
|
|
||||||
|
ht.FundCoins(utxo1Amount, alice)
|
||||||
|
ht.FundCoins(utxo2Amount, alice)
|
||||||
|
ht.FundCoins(utxo3Amount, alice)
|
||||||
|
|
||||||
|
// List Alice's UTXOs to get their outpoints.
|
||||||
|
utxosResp := alice.RPC.ListUnspent(&walletrpc.ListUnspentRequest{
|
||||||
|
MinConfs: 1,
|
||||||
|
MaxConfs: 1000,
|
||||||
|
})
|
||||||
|
require.GreaterOrEqual(ht, len(utxosResp.Utxos), 3,
|
||||||
|
"expected at least 3 UTXOs")
|
||||||
|
|
||||||
|
// Create a lookup map to find outpoints by amount.
|
||||||
|
utxoLookup := make(map[int64]*lnrpc.OutPoint)
|
||||||
|
for _, utxo := range utxosResp.Utxos {
|
||||||
|
utxoLookup[utxo.AmountSat] = utxo.Outpoint
|
||||||
|
}
|
||||||
|
|
||||||
|
// Select the first two UTXOs for our test.
|
||||||
|
selectedInputs := []*lnrpc.OutPoint{
|
||||||
|
utxoLookup[int64(utxo1Amount)],
|
||||||
|
utxoLookup[int64(utxo2Amount)],
|
||||||
|
}
|
||||||
|
require.NotNil(ht, selectedInputs[0], "first UTXO not found")
|
||||||
|
require.NotNil(ht, selectedInputs[1], "second UTXO not found")
|
||||||
|
|
||||||
|
// Generate a destination address for the transaction.
|
||||||
|
destAddrResp := alice.RPC.NewAddress(&lnrpc.NewAddressRequest{
|
||||||
|
Type: lnrpc.AddressType_WITNESS_PUBKEY_HASH,
|
||||||
|
})
|
||||||
|
|
||||||
|
// Amount to send (less than the sum of selected UTXOs to allow for
|
||||||
|
// fees and change).
|
||||||
|
const sendAmount = int64(400_000)
|
||||||
|
|
||||||
|
// Create an address-to-amount map for the transaction.
|
||||||
|
addrToAmount := map[string]int64{
|
||||||
|
destAddrResp.Address: sendAmount,
|
||||||
|
}
|
||||||
|
|
||||||
|
// Call EstimateFee with the selected inputs.
|
||||||
|
estimateReq := &lnrpc.EstimateFeeRequest{
|
||||||
|
AddrToAmount: addrToAmount,
|
||||||
|
TargetConf: targetConf,
|
||||||
|
Inputs: selectedInputs,
|
||||||
|
}
|
||||||
|
estimateResp, err := alice.RPC.LN.EstimateFee(ht.Context(), estimateReq)
|
||||||
|
require.NoError(ht, err, "EstimateFee failed")
|
||||||
|
|
||||||
|
ht.Logf("Fee estimate: %d sats, fee rate: %d sat/vbyte",
|
||||||
|
estimateResp.FeeSat, estimateResp.SatPerVbyte)
|
||||||
|
|
||||||
|
// Verify that the estimate response includes the inputs we specified.
|
||||||
|
require.Len(ht, estimateResp.Inputs, 2,
|
||||||
|
"expected 2 inputs in estimate response")
|
||||||
|
|
||||||
|
// Create a map of input outpoints from the estimate for an easy lookup.
|
||||||
|
estimateInputs := make(map[string]bool)
|
||||||
|
for _, input := range estimateResp.Inputs {
|
||||||
|
key := fmt.Sprintf("%s:%d", input.TxidStr, input.OutputIndex)
|
||||||
|
estimateInputs[key] = true
|
||||||
|
}
|
||||||
|
|
||||||
|
// Verify each selected outpoint is in the estimate.
|
||||||
|
for _, outpoint := range selectedInputs {
|
||||||
|
key := fmt.Sprintf("%s:%d", outpoint.TxidStr,
|
||||||
|
outpoint.OutputIndex)
|
||||||
|
|
||||||
|
require.True(ht, estimateInputs[key],
|
||||||
|
"outpoint %s not found in estimate inputs", key)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Now actually send coins using the same parameters.
|
||||||
|
sendReq := &lnrpc.SendCoinsRequest{
|
||||||
|
Addr: destAddrResp.Address,
|
||||||
|
Amount: sendAmount,
|
||||||
|
TargetConf: targetConf,
|
||||||
|
Outpoints: selectedInputs,
|
||||||
|
}
|
||||||
|
sendResp := alice.RPC.SendCoins(sendReq)
|
||||||
|
txid := sendResp.Txid
|
||||||
|
|
||||||
|
ht.Logf("Transaction sent with txid: %s", txid)
|
||||||
|
|
||||||
|
// Get the transaction details to extract the actual fee paid.
|
||||||
|
txDetails := alice.RPC.GetTransactions(
|
||||||
|
&lnrpc.GetTransactionsRequest{
|
||||||
|
StartHeight: 0,
|
||||||
|
EndHeight: -1,
|
||||||
|
},
|
||||||
|
)
|
||||||
|
|
||||||
|
// Find our transaction in the list.
|
||||||
|
var actualFee int64
|
||||||
|
var found bool
|
||||||
|
for _, tx := range txDetails.Transactions {
|
||||||
|
if tx.TxHash == txid {
|
||||||
|
actualFee = tx.TotalFees
|
||||||
|
found = true
|
||||||
|
ht.Logf("Actual fee paid: %d sats", actualFee)
|
||||||
|
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
require.True(ht, found, "sent transaction not found")
|
||||||
|
|
||||||
|
require.EqualValues(
|
||||||
|
ht, estimateResp.FeeSat, actualFee, "fee estimate does not "+
|
||||||
|
"match actual fee",
|
||||||
|
)
|
||||||
|
|
||||||
|
// Mine the SendCoinsRequest.
|
||||||
|
ht.MineBlocksAndAssertNumTxes(1, 1)
|
||||||
|
}
|
||||||
|
|
||||||
|
// testEstimateOnChainFeeAutoSelectedInputs tests that the EstimateFee RPC
|
||||||
|
// without input selection allows the wallet to auto-select inputs, and that
|
||||||
|
// using those selected inputs in SendCoins produces a matching fee.
|
||||||
|
func testEstimateOnChainFeeAutoSelectedInputs(ht *lntest.HarnessTest) {
|
||||||
|
// Create a new node for this test.
|
||||||
|
alice := ht.NewNode("Alice", nil)
|
||||||
|
|
||||||
|
// Fund Alice with multiple UTXOs.
|
||||||
|
const (
|
||||||
|
utxo1Amount = btcutil.Amount(500_000)
|
||||||
|
utxo2Amount = btcutil.Amount(300_000)
|
||||||
|
utxo3Amount = btcutil.Amount(200_000)
|
||||||
|
targetConf = 2
|
||||||
|
)
|
||||||
|
|
||||||
|
ht.FundCoins(utxo1Amount, alice)
|
||||||
|
ht.FundCoins(utxo2Amount, alice)
|
||||||
|
ht.FundCoins(utxo3Amount, alice)
|
||||||
|
|
||||||
|
// Generate a destination address.
|
||||||
|
destAddrResp := alice.RPC.NewAddress(
|
||||||
|
&lnrpc.NewAddressRequest{
|
||||||
|
Type: lnrpc.AddressType_WITNESS_PUBKEY_HASH,
|
||||||
|
},
|
||||||
|
)
|
||||||
|
|
||||||
|
const sendAmount = int64(100_000)
|
||||||
|
addrToAmount := map[string]int64{
|
||||||
|
destAddrResp.Address: sendAmount,
|
||||||
|
}
|
||||||
|
|
||||||
|
// Estimate fee without specifying inputs (wallet auto-selects).
|
||||||
|
estimateReq := &lnrpc.EstimateFeeRequest{
|
||||||
|
AddrToAmount: addrToAmount,
|
||||||
|
TargetConf: targetConf,
|
||||||
|
}
|
||||||
|
estimateResp, err := alice.RPC.LN.EstimateFee(ht.Context(), estimateReq)
|
||||||
|
require.NoError(ht, err, "EstimateFee failed")
|
||||||
|
|
||||||
|
ht.Logf("Fee estimate (auto-select): %d sats, "+
|
||||||
|
"fee rate: %d sat/vbyte, inputs selected: %d",
|
||||||
|
estimateResp.FeeSat, estimateResp.SatPerVbyte,
|
||||||
|
len(estimateResp.Inputs))
|
||||||
|
|
||||||
|
// The estimate should have selected some inputs.
|
||||||
|
require.NotEmpty(ht, estimateResp.Inputs,
|
||||||
|
"estimate should have selected inputs")
|
||||||
|
|
||||||
|
// Send using the inputs that were selected by the estimate.
|
||||||
|
sendReq := &lnrpc.SendCoinsRequest{
|
||||||
|
Addr: destAddrResp.Address,
|
||||||
|
Amount: sendAmount,
|
||||||
|
TargetConf: targetConf,
|
||||||
|
Outpoints: estimateResp.Inputs,
|
||||||
|
}
|
||||||
|
sendResp := alice.RPC.SendCoins(sendReq)
|
||||||
|
txid := sendResp.Txid
|
||||||
|
|
||||||
|
ht.Logf("Transaction sent with txid: %s", txid)
|
||||||
|
|
||||||
|
// Get the actual fee.
|
||||||
|
txDetails := alice.RPC.GetTransactions(
|
||||||
|
&lnrpc.GetTransactionsRequest{
|
||||||
|
StartHeight: 0,
|
||||||
|
EndHeight: -1,
|
||||||
|
},
|
||||||
|
)
|
||||||
|
|
||||||
|
var actualFee int64
|
||||||
|
var found bool
|
||||||
|
for _, tx := range txDetails.Transactions {
|
||||||
|
if tx.TxHash == txid {
|
||||||
|
actualFee = tx.TotalFees
|
||||||
|
found = true
|
||||||
|
ht.Logf("Actual fee paid: %d sats", actualFee)
|
||||||
|
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
require.True(ht, found, "sent transaction not found")
|
||||||
|
|
||||||
|
require.EqualValues(
|
||||||
|
ht, estimateResp.FeeSat, actualFee, "fee estimate does not "+
|
||||||
|
"match actual fee",
|
||||||
|
)
|
||||||
|
|
||||||
|
// Mine the SendCoinsRequest.
|
||||||
|
ht.MineBlocksAndAssertNumTxes(1, 1)
|
||||||
|
}
|
||||||
Reference in new issue
Block a user