aljaz/lndhub.go
1
1
Fork 0
mirror of https://github.com/getAlby/lndhub.go.git synced 2025-12-19 13:44:53 +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

38
rabbitmq/rabbitmq_test.go
View File

@@ -29,34 +29,39 @@ func TestFinalizedInitializedPayments(t *testing.T) {
client, err := rabbitmq.NewClient(amqpClient)
assert.NoError(t, err)
ch := make(chan amqp.Delivery, 1)
ch := make(chan amqp.Delivery, 2)
amqpClient.EXPECT().
Listen(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).
MaxTimes(1).
Times(1).
Return(ch, nil)
hash := "69e5f0f0590be75e30f671d56afe1d55"
firstHash := "69e5f0f0590be75e30f671d56afe1d55"
secondHash := "ffff0f0590be75e30f671d56afe1d55"
invoices := []models.Invoice{
{
ID: 0,
RHash: hash,
RHash: firstHash,
},
{
ID: 1,
RHash: secondHash,
},
}
lndHubService.EXPECT().
GetAllPendingPayments(gomock.Any()).
MaxTimes(1).
Times(1).
Return(invoices, nil)
lndHubService.EXPECT().
HandleFailedPayment(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).
AnyTimes().
HandleSuccessfulPayment(gomock.Any(), gomock.Eq(&invoices[0]), gomock.Any()).
Times(1).
Return(nil)
lndHubService.EXPECT().
HandleSuccessfulPayment(gomock.Any(), gomock.Any(), gomock.Any()).
AnyTimes().
HandleFailedPayment(gomock.Any(), gomock.Eq(&invoices[1]), gomock.Any(), gomock.Any()).
Times(1).
Return(nil)
lndHubService.EXPECT().
@@ -64,13 +69,24 @@ func TestFinalizedInitializedPayments(t *testing.T) {
AnyTimes().
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()
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 {
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{}