aljaz/lndhub.go
1
1
Fork 0
mirror of https://github.com/getAlby/lndhub.go.git synced 2025-12-20 14:14:47 +01:00
Code Issues Packages Projects Releases Wiki Activity

More fleshed out testing scenario for FinalizePayment routine

Browse Source
This commit is contained in:
Lucas Rouckhout
2023-05-30 10:59:36 +02:00
parent af8e7f2376
commit 752f005257
2 changed files with 29 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
main.go
View File

@@ -165,6 +165,8 @@ func main() {
logger.Fatal(err) logger.Fatal(err)
} }
defer amqpClient.Close()
rabbitmqClient, err = rabbitmq.NewClient(amqpClient, rabbitmqClient, err = rabbitmq.NewClient(amqpClient,
rabbitmq.WithLogger(logger), rabbitmq.WithLogger(logger),
rabbitmq.WithLndInvoiceExchange(c.RabbitMQLndInvoiceExchange), rabbitmq.WithLndInvoiceExchange(c.RabbitMQLndInvoiceExchange),

38
rabbitmq/rabbitmq_test.go
View File

@@ -29,34 +29,39 @@ func TestFinalizedInitializedPayments(t *testing.T) {
client, err := rabbitmq.NewClient(amqpClient) client, err := rabbitmq.NewClient(amqpClient)
assert.NoError(t, err) assert.NoError(t, err)
ch := make(chan amqp.Delivery, 1) ch := make(chan amqp.Delivery, 2)
amqpClient.EXPECT(). amqpClient.EXPECT().
Listen(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()). Listen(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).
MaxTimes(1). Times(1).
Return(ch, nil) Return(ch, nil)
hash := "69e5f0f0590be75e30f671d56afe1d55" firstHash := "69e5f0f0590be75e30f671d56afe1d55"
secondHash := "ffff0f0590be75e30f671d56afe1d55"
invoices := []models.Invoice{ invoices := []models.Invoice{
{ {
ID: 0, ID: 0,
RHash: hash, RHash: firstHash,
},
{
ID: 1,
RHash: secondHash,
}, },
} }
lndHubService.EXPECT(). lndHubService.EXPECT().
GetAllPendingPayments(gomock.Any()). GetAllPendingPayments(gomock.Any()).
MaxTimes(1). Times(1).
Return(invoices, nil) Return(invoices, nil)
lndHubService.EXPECT(). lndHubService.EXPECT().
HandleFailedPayment(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()). HandleSuccessfulPayment(gomock.Any(), gomock.Eq(&invoices[0]), gomock.Any()).
AnyTimes(). Times(1).
Return(nil) Return(nil)
lndHubService.EXPECT(). lndHubService.EXPECT().
HandleSuccessfulPayment(gomock.Any(), gomock.Any(), gomock.Any()). HandleFailedPayment(gomock.Any(), gomock.Eq(&invoices[1]), gomock.Any(), gomock.Any()).
AnyTimes(). Times(1).
Return(nil) Return(nil)
lndHubService.EXPECT(). lndHubService.EXPECT().
@@ -64,13 +69,24 @@ func TestFinalizedInitializedPayments(t *testing.T) {
AnyTimes(). AnyTimes().
Return(models.TransactionEntry{InvoiceID: invoices[0].ID}, nil) Return(models.TransactionEntry{InvoiceID: invoices[0].ID}, nil)
lndHubService.EXPECT().
GetTransactionEntryByInvoiceId(gomock.Any(), gomock.Eq(invoices[1].ID)).
AnyTimes().
Return(models.TransactionEntry{InvoiceID: invoices[1].ID}, nil)
ctx := context.Background() ctx := context.Background()
b, err := json.Marshal(&lnrpc.Payment{PaymentHash: hash, Status: lnrpc.Payment_SUCCEEDED}) successPaymnent, err := json.Marshal(&lnrpc.Payment{PaymentHash: firstHash, Status: lnrpc.Payment_SUCCEEDED})
if err != nil { if err != nil {
t.Error(err) t.Error(err)
} }
ch <- amqp.Delivery{Body: b} failedPayment, err := json.Marshal(&lnrpc.Payment{PaymentHash: secondHash, Status: lnrpc.Payment_FAILED})
if err != nil {
t.Error(err)
}
ch <- amqp.Delivery{Body: successPaymnent}
ch <- amqp.Delivery{Body: failedPayment}
wg := sync.WaitGroup{} wg := sync.WaitGroup{}