mirror of
https://github.com/lightningnetwork/lnd.git
synced 2025-07-28 13:52:55 +02:00
itest+lntest: migrate lntemp
to lntest
This commit performs the takeover that `lntemp` is now promoted to be `lntest`, and the scaffolding is now removed as all the refactoring is finished!
This commit is contained in:
279
lntest/rpc/wallet_kit.go
Normal file
279
lntest/rpc/wallet_kit.go
Normal file
@@ -0,0 +1,279 @@
|
||||
package rpc
|
||||
|
||||
import (
|
||||
"context"
|
||||
|
||||
"github.com/lightningnetwork/lnd/lnrpc/signrpc"
|
||||
"github.com/lightningnetwork/lnd/lnrpc/walletrpc"
|
||||
"github.com/stretchr/testify/require"
|
||||
)
|
||||
|
||||
// =====================
|
||||
// WalletKitClient related RPCs.
|
||||
// =====================
|
||||
|
||||
type (
|
||||
SignReq *walletrpc.SignMessageWithAddrResponse
|
||||
VerifyResp *walletrpc.VerifyMessageWithAddrResponse
|
||||
)
|
||||
|
||||
// FinalizePsbt makes a RPC call to node's ListUnspent and asserts.
|
||||
func (h *HarnessRPC) ListUnspent(
|
||||
req *walletrpc.ListUnspentRequest) *walletrpc.ListUnspentResponse {
|
||||
|
||||
ctxt, cancel := context.WithTimeout(h.runCtx, DefaultTimeout)
|
||||
defer cancel()
|
||||
|
||||
resp, err := h.WalletKit.ListUnspent(ctxt, req)
|
||||
h.NoError(err, "ListUnspent")
|
||||
|
||||
return resp
|
||||
}
|
||||
|
||||
// DeriveKey makes a RPC call to the DeriveKey and asserts.
|
||||
func (h *HarnessRPC) DeriveKey(kl *signrpc.KeyLocator) *signrpc.KeyDescriptor {
|
||||
ctxt, cancel := context.WithTimeout(h.runCtx, DefaultTimeout)
|
||||
defer cancel()
|
||||
|
||||
key, err := h.WalletKit.DeriveKey(ctxt, kl)
|
||||
h.NoError(err, "DeriveKey")
|
||||
|
||||
return key
|
||||
}
|
||||
|
||||
// SendOutputs makes a RPC call to the node's WalletKitClient and asserts.
|
||||
func (h *HarnessRPC) SendOutputs(
|
||||
req *walletrpc.SendOutputsRequest) *walletrpc.SendOutputsResponse {
|
||||
|
||||
ctxt, cancel := context.WithTimeout(h.runCtx, DefaultTimeout)
|
||||
defer cancel()
|
||||
|
||||
resp, err := h.WalletKit.SendOutputs(ctxt, req)
|
||||
h.NoError(err, "SendOutputs")
|
||||
|
||||
return resp
|
||||
}
|
||||
|
||||
// FundPsbt makes a RPC call to node's FundPsbt and asserts.
|
||||
func (h *HarnessRPC) FundPsbt(
|
||||
req *walletrpc.FundPsbtRequest) *walletrpc.FundPsbtResponse {
|
||||
|
||||
ctxt, cancel := context.WithTimeout(h.runCtx, DefaultTimeout)
|
||||
defer cancel()
|
||||
|
||||
resp, err := h.WalletKit.FundPsbt(ctxt, req)
|
||||
h.NoError(err, "FundPsbt")
|
||||
|
||||
return resp
|
||||
}
|
||||
|
||||
// FinalizePsbt makes a RPC call to node's FinalizePsbt and asserts.
|
||||
func (h *HarnessRPC) FinalizePsbt(
|
||||
req *walletrpc.FinalizePsbtRequest) *walletrpc.FinalizePsbtResponse {
|
||||
|
||||
ctxt, cancel := context.WithTimeout(h.runCtx, DefaultTimeout)
|
||||
defer cancel()
|
||||
|
||||
resp, err := h.WalletKit.FinalizePsbt(ctxt, req)
|
||||
h.NoError(err, "FinalizePsbt")
|
||||
|
||||
return resp
|
||||
}
|
||||
|
||||
// LabelTransactionAssertErr makes a RPC call to the node's LabelTransaction
|
||||
// and asserts an error is returned. It then returns the error.
|
||||
func (h *HarnessRPC) LabelTransactionAssertErr(
|
||||
req *walletrpc.LabelTransactionRequest) error {
|
||||
|
||||
ctxt, cancel := context.WithTimeout(h.runCtx, DefaultTimeout)
|
||||
defer cancel()
|
||||
|
||||
_, err := h.WalletKit.LabelTransaction(ctxt, req)
|
||||
require.Error(h, err, "expected error returned")
|
||||
|
||||
return err
|
||||
}
|
||||
|
||||
// LabelTransaction makes a RPC call to the node's LabelTransaction
|
||||
// and asserts no error is returned.
|
||||
func (h *HarnessRPC) LabelTransaction(req *walletrpc.LabelTransactionRequest) {
|
||||
ctxt, cancel := context.WithTimeout(h.runCtx, DefaultTimeout)
|
||||
defer cancel()
|
||||
|
||||
_, err := h.WalletKit.LabelTransaction(ctxt, req)
|
||||
h.NoError(err, "LabelTransaction")
|
||||
}
|
||||
|
||||
// DeriveNextKey makes a RPC call to the DeriveNextKey and asserts.
|
||||
func (h *HarnessRPC) DeriveNextKey(
|
||||
req *walletrpc.KeyReq) *signrpc.KeyDescriptor {
|
||||
|
||||
ctxt, cancel := context.WithTimeout(h.runCtx, DefaultTimeout)
|
||||
defer cancel()
|
||||
|
||||
key, err := h.WalletKit.DeriveNextKey(ctxt, req)
|
||||
h.NoError(err, "DeriveNextKey")
|
||||
|
||||
return key
|
||||
}
|
||||
|
||||
// ListAddresses makes a RPC call to the ListAddresses and asserts.
|
||||
func (h *HarnessRPC) ListAddresses(
|
||||
req *walletrpc.ListAddressesRequest) *walletrpc.ListAddressesResponse {
|
||||
|
||||
ctxt, cancel := context.WithTimeout(h.runCtx, DefaultTimeout)
|
||||
defer cancel()
|
||||
|
||||
key, err := h.WalletKit.ListAddresses(ctxt, req)
|
||||
h.NoError(err, "ListAddresses")
|
||||
|
||||
return key
|
||||
}
|
||||
|
||||
// SignMessageWithAddr makes a RPC call to the SignMessageWithAddr and asserts.
|
||||
func (h *HarnessRPC) SignMessageWithAddr(
|
||||
req *walletrpc.SignMessageWithAddrRequest) SignReq {
|
||||
|
||||
ctxt, cancel := context.WithTimeout(h.runCtx, DefaultTimeout)
|
||||
defer cancel()
|
||||
|
||||
key, err := h.WalletKit.SignMessageWithAddr(ctxt, req)
|
||||
h.NoError(err, "SignMessageWithAddr")
|
||||
|
||||
return key
|
||||
}
|
||||
|
||||
// VerifyMessageWithAddr makes a RPC call to
|
||||
// the VerifyMessageWithAddr and asserts.
|
||||
func (h *HarnessRPC) VerifyMessageWithAddr(
|
||||
req *walletrpc.VerifyMessageWithAddrRequest) VerifyResp {
|
||||
|
||||
ctxt, cancel := context.WithTimeout(h.runCtx, DefaultTimeout)
|
||||
defer cancel()
|
||||
|
||||
key, err := h.WalletKit.VerifyMessageWithAddr(ctxt, req)
|
||||
h.NoError(err, "VerifyMessageWithAddr")
|
||||
|
||||
return key
|
||||
}
|
||||
|
||||
// ListSweeps makes a ListSweeps RPC call to the node's WalletKit client.
|
||||
func (h *HarnessRPC) ListSweeps(verbose bool) *walletrpc.ListSweepsResponse {
|
||||
ctxt, cancel := context.WithTimeout(h.runCtx, DefaultTimeout)
|
||||
defer cancel()
|
||||
|
||||
req := &walletrpc.ListSweepsRequest{Verbose: verbose}
|
||||
resp, err := h.WalletKit.ListSweeps(ctxt, req)
|
||||
h.NoError(err, "ListSweeps")
|
||||
|
||||
return resp
|
||||
}
|
||||
|
||||
// PendingSweeps makes a RPC call to the node's WalletKitClient and asserts.
|
||||
func (h *HarnessRPC) PendingSweeps() *walletrpc.PendingSweepsResponse {
|
||||
ctxt, cancel := context.WithTimeout(h.runCtx, DefaultTimeout)
|
||||
defer cancel()
|
||||
|
||||
req := &walletrpc.PendingSweepsRequest{}
|
||||
|
||||
resp, err := h.WalletKit.PendingSweeps(ctxt, req)
|
||||
h.NoError(err, "PendingSweeps")
|
||||
|
||||
return resp
|
||||
}
|
||||
|
||||
// PublishTransaction makes an RPC call to the node's WalletKitClient and
|
||||
// asserts.
|
||||
func (h *HarnessRPC) PublishTransaction(
|
||||
req *walletrpc.Transaction) *walletrpc.PublishResponse {
|
||||
|
||||
ctxt, cancel := context.WithTimeout(h.runCtx, DefaultTimeout)
|
||||
defer cancel()
|
||||
|
||||
resp, err := h.WalletKit.PublishTransaction(ctxt, req)
|
||||
h.NoError(err, "PublishTransaction")
|
||||
|
||||
return resp
|
||||
}
|
||||
|
||||
// BumpFee makes a RPC call to the node's WalletKitClient and asserts.
|
||||
func (h *HarnessRPC) BumpFee(
|
||||
req *walletrpc.BumpFeeRequest) *walletrpc.BumpFeeResponse {
|
||||
|
||||
ctxt, cancel := context.WithTimeout(h.runCtx, DefaultTimeout)
|
||||
defer cancel()
|
||||
|
||||
resp, err := h.WalletKit.BumpFee(ctxt, req)
|
||||
h.NoError(err, "BumpFee")
|
||||
|
||||
return resp
|
||||
}
|
||||
|
||||
// ListAccounts makes a RPC call to the node's WalletKitClient and asserts.
|
||||
func (h *HarnessRPC) ListAccounts(
|
||||
req *walletrpc.ListAccountsRequest) *walletrpc.ListAccountsResponse {
|
||||
|
||||
ctxt, cancel := context.WithTimeout(h.runCtx, DefaultTimeout)
|
||||
defer cancel()
|
||||
|
||||
resp, err := h.WalletKit.ListAccounts(ctxt, req)
|
||||
h.NoError(err, "ListAccounts")
|
||||
|
||||
return resp
|
||||
}
|
||||
|
||||
// ImportAccount makes a RPC call to the node's WalletKitClient and asserts.
|
||||
func (h *HarnessRPC) ImportAccount(
|
||||
req *walletrpc.ImportAccountRequest) *walletrpc.ImportAccountResponse {
|
||||
|
||||
ctxt, cancel := context.WithTimeout(h.runCtx, DefaultTimeout)
|
||||
defer cancel()
|
||||
|
||||
resp, err := h.WalletKit.ImportAccount(ctxt, req)
|
||||
h.NoError(err, "ImportAccount")
|
||||
|
||||
return resp
|
||||
}
|
||||
|
||||
// ImportPublicKey makes a RPC call to the node's WalletKitClient and asserts.
|
||||
//
|
||||
//nolint:lll
|
||||
func (h *HarnessRPC) ImportPublicKey(
|
||||
req *walletrpc.ImportPublicKeyRequest) *walletrpc.ImportPublicKeyResponse {
|
||||
|
||||
ctxt, cancel := context.WithTimeout(h.runCtx, DefaultTimeout)
|
||||
defer cancel()
|
||||
|
||||
resp, err := h.WalletKit.ImportPublicKey(ctxt, req)
|
||||
h.NoError(err, "ImportPublicKey")
|
||||
|
||||
return resp
|
||||
}
|
||||
|
||||
// SignPsbt makes a RPC call to the node's WalletKitClient and asserts.
|
||||
func (h *HarnessRPC) SignPsbt(
|
||||
req *walletrpc.SignPsbtRequest) *walletrpc.SignPsbtResponse {
|
||||
|
||||
ctxt, cancel := context.WithTimeout(h.runCtx, DefaultTimeout)
|
||||
defer cancel()
|
||||
|
||||
resp, err := h.WalletKit.SignPsbt(ctxt, req)
|
||||
h.NoError(err, "SignPsbt")
|
||||
|
||||
return resp
|
||||
}
|
||||
|
||||
// ImportTapscript makes a RPC call to the node's WalletKitClient and asserts.
|
||||
//
|
||||
//nolint:lll
|
||||
func (h *HarnessRPC) ImportTapscript(
|
||||
req *walletrpc.ImportTapscriptRequest) *walletrpc.ImportTapscriptResponse {
|
||||
|
||||
ctxt, cancel := context.WithTimeout(h.runCtx, DefaultTimeout)
|
||||
defer cancel()
|
||||
|
||||
resp, err := h.WalletKit.ImportTapscript(ctxt, req)
|
||||
h.NoError(err, "ImportTapscript")
|
||||
|
||||
return resp
|
||||
}
|
Reference in New Issue
Block a user