package test import ( "context" "errors" "sync" "time" "github.com/btcsuite/btcd/btcutil" "github.com/btcsuite/btcd/chaincfg" "github.com/btcsuite/btcd/chaincfg/chainhash" "github.com/btcsuite/btcd/wire" "github.com/btcsuite/btcwallet/wtxmgr" "github.com/lightninglabs/lndclient" "github.com/lightningnetwork/lnd/keychain" "github.com/lightningnetwork/lnd/lnrpc/walletrpc" "github.com/lightningnetwork/lnd/lnwallet" "github.com/lightningnetwork/lnd/lnwallet/chainfee" ) // DefaultMockFee is the default value we use for fee estimates when no values // are set for specific conf targets. var DefaultMockFee = chainfee.SatPerKWeight(10000) type mockWalletKit struct { lnd *LndMockServices keyIndex int32 feeEstimateLock sync.Mutex feeEstimates map[int32]chainfee.SatPerKWeight } var _ lndclient.WalletKitClient = (*mockWalletKit)(nil) func (m *mockWalletKit) ListUnspent(ctx context.Context, minConfs, maxConfs int32) ([]*lnwallet.Utxo, error) { return nil, nil } func (m *mockWalletKit) LeaseOutput(ctx context.Context, lockID wtxmgr.LockID, op wire.OutPoint, duration time.Duration) (time.Time, error) { return time.Now(), nil } func (m *mockWalletKit) ReleaseOutput(ctx context.Context, lockID wtxmgr.LockID, op wire.OutPoint) error { return nil } func (m *mockWalletKit) DeriveNextKey(ctx context.Context, family int32) ( *keychain.KeyDescriptor, error) { index := m.keyIndex _, pubKey := CreateKey(index) m.keyIndex++ return &keychain.KeyDescriptor{ KeyLocator: keychain.KeyLocator{ Family: keychain.KeyFamily(family), Index: uint32(index), }, PubKey: pubKey, }, nil } func (m *mockWalletKit) DeriveKey(ctx context.Context, in *keychain.KeyLocator) ( *keychain.KeyDescriptor, error) { _, pubKey := CreateKey(int32(in.Index)) return &keychain.KeyDescriptor{ KeyLocator: *in, PubKey: pubKey, }, nil } func (m *mockWalletKit) NextAddr(ctx context.Context) (btcutil.Address, error) { addr, err := btcutil.NewAddressWitnessPubKeyHash( make([]byte, 20), &chaincfg.TestNet3Params, ) if err != nil { return nil, err } return addr, nil } func (m *mockWalletKit) PublishTransaction(ctx context.Context, tx *wire.MsgTx, _ string) error { m.lnd.AddTx(tx) m.lnd.TxPublishChannel <- tx return nil } func (m *mockWalletKit) SendOutputs(ctx context.Context, outputs []*wire.TxOut, feeRate chainfee.SatPerKWeight, _ string) (*wire.MsgTx, error) { var inputTxHash chainhash.Hash tx := wire.MsgTx{} tx.AddTxIn(&wire.TxIn{ PreviousOutPoint: wire.OutPoint{ Hash: inputTxHash, Index: 0, }, }) for _, out := range outputs { tx.AddTxOut(&wire.TxOut{ PkScript: out.PkScript, Value: out.Value, }) } m.lnd.AddTx(&tx) m.lnd.SendOutputsChannel <- tx return &tx, nil } func (m *mockWalletKit) setFeeEstimate(confTarget int32, fee chainfee.SatPerKWeight) { m.feeEstimateLock.Lock() defer m.feeEstimateLock.Unlock() m.feeEstimates[confTarget] = fee } func (m *mockWalletKit) EstimateFee(ctx context.Context, confTarget int32) ( chainfee.SatPerKWeight, error) { m.feeEstimateLock.Lock() defer m.feeEstimateLock.Unlock() if confTarget <= 1 { return 0, errors.New("conf target must be greater than 1") } feeEstimate, ok := m.feeEstimates[confTarget] if !ok { return DefaultMockFee, nil } return feeEstimate, nil } // ListSweeps returns a list of the sweep transaction ids known to our node. func (m *mockWalletKit) ListSweeps(_ context.Context) ([]string, error) { return m.lnd.Sweeps, nil } // BumpFee attempts to bump the fee of a transaction by spending one of // its outputs at the given fee rate. This essentially results in a // child-pays-for-parent (CPFP) scenario. If the given output has been // used in a previous BumpFee call, then a transaction replacing the // previous is broadcast, resulting in a replace-by-fee (RBF) scenario. func (m *mockWalletKit) BumpFee(context.Context, wire.OutPoint, chainfee.SatPerKWeight) error { return nil } // ListAccounts retrieves all accounts belonging to the wallet by default. // Optional name and addressType can be provided to filter through all of the // wallet accounts and return only those matching. func (m *mockWalletKit) ListAccounts(context.Context, string, walletrpc.AddressType) ([]*walletrpc.Account, error) { return nil, nil }