mirror of
https://github.com/getAlby/lndhub.go.git
synced 2026-01-07 15:05:37 +01:00
start writing integration test
This commit is contained in:
kiwiidb
186
integration_tests/hodl_invoice_test.go
Normal file
186
integration_tests/hodl_invoice_test.go
Normal file
@@ -0,0 +1,186 @@
|
||||
package integration_tests
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"log"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/getAlby/lndhub.go/common"
|
||||
"github.com/getAlby/lndhub.go/controllers"
|
||||
"github.com/getAlby/lndhub.go/lib"
|
||||
"github.com/getAlby/lndhub.go/lib/responses"
|
||||
"github.com/getAlby/lndhub.go/lib/service"
|
||||
"github.com/getAlby/lndhub.go/lib/tokens"
|
||||
"github.com/go-playground/validator/v10"
|
||||
"github.com/labstack/echo/v4"
|
||||
"github.com/lightningnetwork/lnd/lnrpc"
|
||||
"github.com/stretchr/testify/assert"
|
||||
"github.com/stretchr/testify/suite"
|
||||
)
|
||||
|
||||
type HodlInvoiceSuite struct {
|
||||
TestSuite
|
||||
mlnd *MockLND
|
||||
externalLND *MockLND
|
||||
service *service.LndhubService
|
||||
userLogin ExpectedCreateUserResponseBody
|
||||
userToken string
|
||||
invoiceUpdateSubCancelFn context.CancelFunc
|
||||
hodlLND *LNDMockHodlWrapperAsync
|
||||
}
|
||||
|
||||
func (suite *HodlInvoiceSuite) SetupSuite() {
|
||||
mlnd := newDefaultMockLND()
|
||||
externalLND, err := NewMockLND("1234567890abcdefabcd", 0, make(chan (*lnrpc.Invoice)))
|
||||
if err != nil {
|
||||
log.Fatalf("Error initializing test service: %v", err)
|
||||
}
|
||||
suite.externalLND = externalLND
|
||||
suite.mlnd = mlnd
|
||||
// inject hodl lnd client
|
||||
lndClient, err := NewLNDMockHodlWrapperAsync(mlnd)
|
||||
suite.hodlLND = lndClient
|
||||
if err != nil {
|
||||
log.Fatalf("Error setting up test client: %v", err)
|
||||
}
|
||||
|
||||
svc, err := LndHubTestServiceInit(lndClient)
|
||||
if err != nil {
|
||||
log.Fatalf("Error initializing test service: %v", err)
|
||||
}
|
||||
users, userTokens, err := createUsers(svc, 1)
|
||||
if err != nil {
|
||||
log.Fatalf("Error creating test users: %v", err)
|
||||
}
|
||||
// Subscribe to LND invoice updates in the background
|
||||
// store cancel func to be called in tear down suite
|
||||
ctx, cancel := context.WithCancel(context.Background())
|
||||
suite.invoiceUpdateSubCancelFn = cancel
|
||||
go svc.InvoiceUpdateSubscription(ctx)
|
||||
suite.service = svc
|
||||
e := echo.New()
|
||||
|
||||
e.HTTPErrorHandler = responses.HTTPErrorHandler
|
||||
e.Validator = &lib.CustomValidator{Validator: validator.New()}
|
||||
suite.echo = e
|
||||
assert.Equal(suite.T(), 1, len(users))
|
||||
assert.Equal(suite.T(), 1, len(userTokens))
|
||||
suite.userLogin = users[0]
|
||||
suite.userToken = userTokens[0]
|
||||
suite.echo.Use(tokens.Middleware([]byte(suite.service.Config.JWTSecret)))
|
||||
suite.echo.GET("/balance", controllers.NewBalanceController(suite.service).Balance)
|
||||
suite.echo.POST("/addinvoice", controllers.NewAddInvoiceController(suite.service).AddInvoice)
|
||||
suite.echo.POST("/payinvoice", controllers.NewPayInvoiceController(suite.service).PayInvoice)
|
||||
}
|
||||
|
||||
func (suite *HodlInvoiceSuite) TestHodlInvoiceSuccess() {
|
||||
userFundingSats := 1000
|
||||
externalSatRequested := 500
|
||||
// fund user account
|
||||
invoiceResponse := suite.createAddInvoiceReq(userFundingSats, "integration test external payment user", suite.userToken)
|
||||
err := suite.mlnd.mockPaidInvoice(invoiceResponse, 0, false, nil)
|
||||
assert.NoError(suite.T(), err)
|
||||
|
||||
// wait a bit for the callback event to hit
|
||||
time.Sleep(10 * time.Millisecond)
|
||||
|
||||
// create external invoice
|
||||
externalInvoice := lnrpc.Invoice{
|
||||
Memo: "integration tests: external pay from user",
|
||||
Value: int64(externalSatRequested),
|
||||
}
|
||||
invoice, err := suite.externalLND.AddInvoice(context.Background(), &externalInvoice)
|
||||
assert.NoError(suite.T(), err)
|
||||
// pay external from user, req will be canceled after 2 sec
|
||||
go suite.createPayInvoiceReqWithCancel(invoice.PaymentRequest, suite.userToken)
|
||||
|
||||
// wait for request to fail
|
||||
time.Sleep(5 * time.Second)
|
||||
|
||||
//TODO
|
||||
//start payment tracking loop
|
||||
//send settle invoice with lnrpc.payment
|
||||
|
||||
// check to see that balance was reduced
|
||||
userId := getUserIdFromToken(suite.userToken)
|
||||
userBalance, err := suite.service.CurrentUserBalance(context.Background(), userId)
|
||||
if err != nil {
|
||||
fmt.Printf("Error when getting balance %v\n", err.Error())
|
||||
}
|
||||
assert.Equal(suite.T(), int64(userFundingSats-externalSatRequested), userBalance)
|
||||
//todo: check that invoice was updated as completed
|
||||
|
||||
}
|
||||
|
||||
func (suite *HodlInvoiceSuite) TestHodlInvoiceFailure() {
|
||||
userFundingSats := 1000
|
||||
externalSatRequested := 500
|
||||
// fund user account
|
||||
invoiceResponse := suite.createAddInvoiceReq(userFundingSats, "integration test external payment user", suite.userToken)
|
||||
err := suite.mlnd.mockPaidInvoice(invoiceResponse, 0, false, nil)
|
||||
assert.NoError(suite.T(), err)
|
||||
|
||||
// wait a bit for the callback event to hit
|
||||
time.Sleep(10 * time.Millisecond)
|
||||
|
||||
// create external invoice
|
||||
externalInvoice := lnrpc.Invoice{
|
||||
Memo: "integration tests: external pay from user",
|
||||
Value: int64(externalSatRequested),
|
||||
}
|
||||
invoice, err := suite.externalLND.AddInvoice(context.Background(), &externalInvoice)
|
||||
assert.NoError(suite.T(), err)
|
||||
// pay external from user, req will be canceled after 2 sec
|
||||
go suite.createPayInvoiceReqWithCancel(invoice.PaymentRequest, suite.userToken)
|
||||
|
||||
// wait for request to fail
|
||||
time.Sleep(5 * time.Second)
|
||||
|
||||
// check to see that balance was reduced
|
||||
userId := getUserIdFromToken(suite.userToken)
|
||||
userBalance, err := suite.service.CurrentUserBalance(context.Background(), userId)
|
||||
if err != nil {
|
||||
fmt.Printf("Error when getting balance %v\n", err.Error())
|
||||
}
|
||||
assert.Equal(suite.T(), int64(userFundingSats-externalSatRequested), userBalance)
|
||||
|
||||
//TODO
|
||||
//start payment tracking loop
|
||||
//send cancel invoice with lnrpc.payment
|
||||
|
||||
// check that balance was reverted and invoice is in error state
|
||||
userBalance, err = suite.service.CurrentUserBalance(context.Background(), userId)
|
||||
if err != nil {
|
||||
fmt.Printf("Error when getting balance %v\n", err.Error())
|
||||
}
|
||||
assert.Equal(suite.T(), int64(userFundingSats), userBalance)
|
||||
|
||||
invoices, err := suite.service.InvoicesFor(context.Background(), userId, common.InvoiceTypeOutgoing)
|
||||
if err != nil {
|
||||
fmt.Printf("Error when getting invoices %v\n", err.Error())
|
||||
}
|
||||
assert.Equal(suite.T(), 1, len(invoices))
|
||||
assert.Equal(suite.T(), common.InvoiceStateError, invoices[0].State)
|
||||
assert.Equal(suite.T(), SendPaymentMockError, invoices[0].ErrorMessage)
|
||||
|
||||
transactonEntries, err := suite.service.TransactionEntriesFor(context.Background(), userId)
|
||||
if err != nil {
|
||||
fmt.Printf("Error when getting transaction entries %v\n", err.Error())
|
||||
}
|
||||
// check if there are 3 transaction entries, with reversed credit and debit account ids
|
||||
assert.Equal(suite.T(), 3, len(transactonEntries))
|
||||
assert.Equal(suite.T(), transactonEntries[1].CreditAccountID, transactonEntries[2].DebitAccountID)
|
||||
assert.Equal(suite.T(), transactonEntries[1].DebitAccountID, transactonEntries[2].CreditAccountID)
|
||||
assert.Equal(suite.T(), transactonEntries[1].Amount, int64(externalSatRequested))
|
||||
assert.Equal(suite.T(), transactonEntries[2].Amount, int64(externalSatRequested))
|
||||
}
|
||||
|
||||
func (suite *HodlInvoiceSuite) TearDownSuite() {
|
||||
suite.invoiceUpdateSubCancelFn()
|
||||
}
|
||||
|
||||
func TestHodlInvoiceSuite(t *testing.T) {
|
||||
suite.Run(t, new(HodlInvoiceSuite))
|
||||
}
|
||||
@@ -2,50 +2,56 @@ package integration_tests
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
|
||||
"github.com/getAlby/lndhub.go/lnd"
|
||||
"github.com/lightningnetwork/lnd/lnrpc"
|
||||
"github.com/lightningnetwork/lnd/lnrpc/routerrpc"
|
||||
"google.golang.org/grpc"
|
||||
)
|
||||
|
||||
type LNDMockHodlWrapper struct {
|
||||
lnd.LightningClientWrapper
|
||||
}
|
||||
|
||||
func NewLNDMockHodlWrapper(lnd lnd.LightningClientWrapper) (result *LNDMockWrapper, err error) {
|
||||
return &LNDMockWrapper{
|
||||
lnd,
|
||||
}, nil
|
||||
}
|
||||
|
||||
func (wrapper *LNDMockHodlWrapper) SendPaymentSync(ctx context.Context, req *lnrpc.SendRequest, options ...grpc.CallOption) (*lnrpc.SendResponse, error) {
|
||||
return nil, errors.New(SendPaymentMockError)
|
||||
}
|
||||
|
||||
// mock where send payment sync failure is controlled by channel
|
||||
// even though send payment method is still sync, suffix "Async" here is used to show intention of using this mock
|
||||
var paymentResultChannel = make(chan bool, 1)
|
||||
|
||||
type LNDMockHodlWrapperAsync struct {
|
||||
hps *HodlPaymentSubscriber
|
||||
lnd.LightningClientWrapper
|
||||
}
|
||||
|
||||
func NewLNDMockHodlWrapperAsync(lnd lnd.LightningClientWrapper) (result *LNDMockWrapperAsync, err error) {
|
||||
return &LNDMockWrapperAsync{
|
||||
lnd,
|
||||
type HodlPaymentSubscriber struct {
|
||||
ch chan (lnrpc.Payment)
|
||||
}
|
||||
|
||||
// wait for channel, then return
|
||||
func (hps *HodlPaymentSubscriber) Recv() (lnrpc.Payment, error) {
|
||||
return <-hps.ch, nil
|
||||
}
|
||||
|
||||
func NewLNDMockHodlWrapperAsync(lnd lnd.LightningClientWrapper) (result *LNDMockHodlWrapperAsync, err error) {
|
||||
return &LNDMockHodlWrapperAsync{
|
||||
hps: &HodlPaymentSubscriber{
|
||||
ch: make(chan lnrpc.Payment),
|
||||
},
|
||||
LightningClientWrapper: lnd,
|
||||
}, nil
|
||||
}
|
||||
|
||||
func (wrapper *LNDMockHodlWrapperAsync) SubscribePayment(ctx context.Context, req *routerrpc.TrackPaymentRequest, options ...grpc.CallOption) (lnd.SubscribePaymentWrapper, error) {
|
||||
return nil, nil
|
||||
}
|
||||
|
||||
func (wrapper *LNDMockHodlWrapperAsync) SendPaymentSync(ctx context.Context, req *lnrpc.SendRequest, options ...grpc.CallOption) (*lnrpc.SendResponse, error) {
|
||||
//block indefinetely
|
||||
//because we don't want this function to ever return something here
|
||||
//the payments should be processed asynchronously by the payment tracker
|
||||
select {}
|
||||
}
|
||||
|
||||
func (wrapper *LNDMockHodlWrapperAsync) SettlePayment(success bool) {
|
||||
paymentResultChannel <- success
|
||||
func (wrapper *LNDMockHodlWrapperAsync) SettlePayment(payment lnrpc.Payment) {
|
||||
wrapper.hps.ch <- payment
|
||||
}
|
||||
|
||||
//TODO: payment tracker implemantation: read from channel, return to receive method
|
||||
//write test that completes payment
|
||||
//write test that fails payment
|
||||
|
||||
Reference in New Issue
Block a user