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

fix test with sleep

Browse Source
This commit is contained in:
kiwiidb
2023-06-20 13:45:15 +02:00
parent f99b9e9824
commit bec0544a55
1 changed files with 27 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files

73
rabbitmq/rabbitmq_test.go
View File

@@ -35,18 +35,18 @@ func TestFinalizedInitializedPayments(t *testing.T) {
Times(1).
Return(ch, nil)
firstHash := "69e5f0f0590be75e30f671d56afe1d55"
secondHash := "ffff0f0590be75e30f671d56afe1d55"
firstHash := "69e5f0f0590be75e30f671d56afe1d55"
secondHash := "ffff0f0590be75e30f671d56afe1d55"
invoices := []models.Invoice{
{
ID: 0,
RHash: firstHash,
ID: 0,
RHash: firstHash,
},
{
ID: 1,
RHash: secondHash,
},
{
ID: 1,
RHash: secondHash,
},
}
lndHubService.EXPECT().
@@ -75,48 +75,29 @@ func TestFinalizedInitializedPayments(t *testing.T) {
Return(models.TransactionEntry{InvoiceID: invoices[1].ID}, nil)
ctx := context.Background()
successPayment, err := json.Marshal(&lnrpc.Payment{PaymentHash: firstHash, Status: lnrpc.Payment_SUCCEEDED})
if err != nil {
t.Error(err)
}
successPayment, err := json.Marshal(&lnrpc.Payment{PaymentHash: firstHash, Status: lnrpc.Payment_SUCCEEDED})
if err != nil {
t.Error(err)
}
failedPayment, err := json.Marshal(&lnrpc.Payment{PaymentHash: secondHash, Status: lnrpc.Payment_FAILED})
if err != nil {
t.Error(err)
}
failedPayment, err := json.Marshal(&lnrpc.Payment{PaymentHash: secondHash, Status: lnrpc.Payment_FAILED})
if err != nil {
t.Error(err)
}
ch <- amqp.Delivery{Body: successPayment}
ch <- amqp.Delivery{Body: failedPayment}
ch <- amqp.Delivery{Body: successPayment}
ch <- amqp.Delivery{Body: failedPayment}
wg := sync.WaitGroup{}
wg := sync.WaitGroup{}
wg.Add(1)
go func() {
err = client.FinalizeInitializedPayments(ctx, lndHubService)
wg.Add(1)
go func() {
err = client.FinalizeInitializedPayments(ctx, lndHubService)
assert.NoError(t, err)
wg.Done()
}()
assert.NoError(t, err)
wg.Done()
}()
waitTimeout(&wg, time.Second * 3, t)
}
// waitTimeout waits for the waitgroup for the specified max timeout.
// Returns true if waiting timed out.
func waitTimeout(wg *sync.WaitGroup, timeout time.Duration, t *testing.T) bool {
c := make(chan struct{})
go func() {
defer close(c)
wg.Wait()
}()
select {
case <-c:
return false // completed normally
case <-time.After(timeout):
t.Errorf("Waiting on waitgroup timed out during test")
return true // timed out
}
//wait a bit for payments to be processed
time.Sleep(time.Second)
}