mirror of
https://github.com/lightninglabs/aperture.git
synced 2025-12-17 17:14:19 +01:00
3b73ac96cb
To get rid of the loop dependency, we copy the test code that we rely on and fix some imports.
111 lines
2.2 KiB
Go
111 lines
2.2 KiB
Go
package test
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"sync"
|
|
"time"
|
|
|
|
"github.com/btcsuite/btcd/btcec"
|
|
"github.com/lightninglabs/lndclient"
|
|
"github.com/lightningnetwork/lnd/lnrpc/invoicesrpc"
|
|
"github.com/lightningnetwork/lnd/lntypes"
|
|
"github.com/lightningnetwork/lnd/zpay32"
|
|
)
|
|
|
|
type mockInvoices struct {
|
|
lnd *LndMockServices
|
|
wg sync.WaitGroup
|
|
}
|
|
|
|
func (s *mockInvoices) SettleInvoice(ctx context.Context,
|
|
preimage lntypes.Preimage) error {
|
|
|
|
logger.Infof("Settle invoice %v with preimage %v", preimage.Hash(),
|
|
preimage)
|
|
|
|
s.lnd.SettleInvoiceChannel <- preimage
|
|
|
|
return nil
|
|
}
|
|
|
|
func (s *mockInvoices) WaitForFinished() {
|
|
s.wg.Wait()
|
|
}
|
|
|
|
func (s *mockInvoices) CancelInvoice(ctx context.Context,
|
|
hash lntypes.Hash) error {
|
|
|
|
s.lnd.FailInvoiceChannel <- hash
|
|
|
|
return nil
|
|
}
|
|
|
|
func (s *mockInvoices) SubscribeSingleInvoice(ctx context.Context,
|
|
hash lntypes.Hash) (<-chan lndclient.InvoiceUpdate,
|
|
<-chan error, error) {
|
|
|
|
updateChan := make(chan lndclient.InvoiceUpdate, 2)
|
|
errChan := make(chan error)
|
|
|
|
select {
|
|
case s.lnd.SingleInvoiceSubcribeChannel <- &SingleInvoiceSubscription{
|
|
Update: updateChan,
|
|
Err: errChan,
|
|
Hash: hash,
|
|
}:
|
|
case <-ctx.Done():
|
|
return nil, nil, ctx.Err()
|
|
}
|
|
|
|
return updateChan, errChan, nil
|
|
}
|
|
|
|
func (s *mockInvoices) AddHoldInvoice(ctx context.Context,
|
|
in *invoicesrpc.AddInvoiceData) (string, error) {
|
|
|
|
s.lnd.lock.Lock()
|
|
defer s.lnd.lock.Unlock()
|
|
|
|
hash := in.Hash
|
|
|
|
// Create and encode the payment request as a bech32 (zpay32) string.
|
|
creationDate := time.Now()
|
|
|
|
payReq, err := zpay32.NewInvoice(
|
|
s.lnd.ChainParams, *hash, creationDate,
|
|
zpay32.Description(in.Memo),
|
|
zpay32.CLTVExpiry(in.CltvExpiry),
|
|
zpay32.Amount(in.Value),
|
|
)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
privKey, err := btcec.NewPrivateKey(btcec.S256())
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
payReqString, err := payReq.Encode(
|
|
zpay32.MessageSigner{
|
|
SignCompact: func(hash []byte) ([]byte, error) {
|
|
// btcec.SignCompact returns a pubkey-recoverable signature
|
|
sig, err := btcec.SignCompact(
|
|
btcec.S256(), privKey, hash, true,
|
|
)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("can't sign the hash: %v", err)
|
|
}
|
|
|
|
return sig, nil
|
|
},
|
|
},
|
|
)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
return payReqString, nil
|
|
}
|