Archives/loop
2
0
Fork 0
mirror of https://github.com/lightninglabs/loop synced 2024-11-08 01:10:29 +00:00
Code Issues Releases Wiki Activity

sweepbatcher: use method AddSweep in test

Browse Source
This commit is contained in:
Boris Nagaev 2024-05-13 21:16:06 -03:00
parent 951f98147d
commit b5b17991a5
No known key found for this signature in database
1 changed files with 20 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
sweepbatcher/sweep_batcher_test.go
View File

@ -98,14 +98,14 @@ func TestSweepBatcherBatchCreation(t *testing.T) {
store.AssertLoopOutStored()
// Deliver sweep request to batcher.
batcher.sweepReqs <- sweepReq1
require.NoError(t, batcher.AddSweep(&sweepReq1))
// Since a batch was created we check that it registered for its primary
// sweep's spend.
<-lnd.RegisterSpendChannel
// Insert the same swap twice, this should be a noop.
batcher.sweepReqs <- sweepReq1
require.NoError(t, batcher.AddSweep(&sweepReq1))
// Once batcher receives sweep request it will eventually spin up a
// batch.
@ -137,7 +137,7 @@ func TestSweepBatcherBatchCreation(t *testing.T) {
require.NoError(t, err)
store.AssertLoopOutStored()
batcher.sweepReqs <- sweepReq2
require.NoError(t, batcher.AddSweep(&sweepReq2))
// Batcher should not create a second batch as timeout distance is small
// enough.
@ -169,7 +169,7 @@ func TestSweepBatcherBatchCreation(t *testing.T) {
require.NoError(t, err)
store.AssertLoopOutStored()
batcher.sweepReqs <- sweepReq3
require.NoError(t, batcher.AddSweep(&sweepReq3))
// Batcher should create a second batch as timeout distance is greater
// than the threshold
@ -251,7 +251,7 @@ func TestSweepBatcherSimpleLifecycle(t *testing.T) {
store.AssertLoopOutStored()
// Deliver sweep request to batcher.
batcher.sweepReqs <- sweepReq1
require.NoError(t, batcher.AddSweep(&sweepReq1))
// Eventually request will be consumed and a new batch will spin up.
require.Eventually(t, func() bool {
@ -435,15 +435,15 @@ func TestSweepBatcherSweepReentry(t *testing.T) {
store.AssertLoopOutStored()
// Feed the sweeps to the batcher.
batcher.sweepReqs <- sweepReq1
require.NoError(t, batcher.AddSweep(&sweepReq1))
// After inserting the primary (first) sweep, a spend monitor should be
// registered.
<-lnd.RegisterSpendChannel
batcher.sweepReqs <- sweepReq2
require.NoError(t, batcher.AddSweep(&sweepReq2))
batcher.sweepReqs <- sweepReq3
require.NoError(t, batcher.AddSweep(&sweepReq3))
// Batcher should create a batch for the sweeps.
require.Eventually(t, func() bool {
@ -593,7 +593,7 @@ func TestSweepBatcherNonWalletAddr(t *testing.T) {
store.AssertLoopOutStored()
// Deliver sweep request to batcher.
batcher.sweepReqs <- sweepReq1
require.NoError(t, batcher.AddSweep(&sweepReq1))
// Once batcher receives sweep request it will eventually spin up a
// batch.
@ -606,7 +606,7 @@ func TestSweepBatcherNonWalletAddr(t *testing.T) {
<-lnd.RegisterSpendChannel
// Insert the same swap twice, this should be a noop.
batcher.sweepReqs <- sweepReq1
require.NoError(t, batcher.AddSweep(&sweepReq1))
// Create a second sweep request that has a timeout distance less than
// our configured threshold.
@ -633,7 +633,7 @@ func TestSweepBatcherNonWalletAddr(t *testing.T) {
require.NoError(t, err)
store.AssertLoopOutStored()
batcher.sweepReqs <- sweepReq2
require.NoError(t, batcher.AddSweep(&sweepReq2))
// Batcher should create a second batch as first batch is a non wallet
// addr batch.
@ -670,7 +670,7 @@ func TestSweepBatcherNonWalletAddr(t *testing.T) {
require.NoError(t, err)
store.AssertLoopOutStored()
batcher.sweepReqs <- sweepReq3
require.NoError(t, batcher.AddSweep(&sweepReq3))
// Batcher should create a new batch as timeout distance is greater than
// the threshold
@ -879,7 +879,7 @@ func TestSweepBatcherComposite(t *testing.T) {
store.AssertLoopOutStored()
// Deliver sweep request to batcher.
batcher.sweepReqs <- sweepReq1
require.NoError(t, batcher.AddSweep(&sweepReq1))
// Once batcher receives sweep request it will eventually spin up a
// batch.
@ -892,9 +892,9 @@ func TestSweepBatcherComposite(t *testing.T) {
<-lnd.RegisterSpendChannel
// Insert the same swap twice, this should be a noop.
batcher.sweepReqs <- sweepReq1
require.NoError(t, batcher.AddSweep(&sweepReq1))
batcher.sweepReqs <- sweepReq2
require.NoError(t, batcher.AddSweep(&sweepReq2))
// Batcher should not create a second batch as timeout distance is small
// enough.
@ -902,7 +902,7 @@ func TestSweepBatcherComposite(t *testing.T) {
return len(batcher.batches) == 1
}, test.Timeout, eventuallyCheckFrequency)
batcher.sweepReqs <- sweepReq3
require.NoError(t, batcher.AddSweep(&sweepReq3))
// Batcher should create a second batch as this sweep pays to a non
// wallet address.
@ -914,7 +914,7 @@ func TestSweepBatcherComposite(t *testing.T) {
// sweep's spend.
<-lnd.RegisterSpendChannel
batcher.sweepReqs <- sweepReq4
require.NoError(t, batcher.AddSweep(&sweepReq4))
// Batcher should create a third batch as timeout distance is greater
// than the threshold.
@ -926,7 +926,7 @@ func TestSweepBatcherComposite(t *testing.T) {
// sweep's spend.
<-lnd.RegisterSpendChannel
batcher.sweepReqs <- sweepReq5
require.NoError(t, batcher.AddSweep(&sweepReq5))
// Batcher should not create a fourth batch as timeout distance is small
// enough for it to join the last batch.
@ -934,7 +934,7 @@ func TestSweepBatcherComposite(t *testing.T) {
return len(batcher.batches) == 3
}, test.Timeout, eventuallyCheckFrequency)
batcher.sweepReqs <- sweepReq6
require.NoError(t, batcher.AddSweep(&sweepReq6))
// Batcher should create a fourth batch as this sweep pays to a non
// wallet address.
@ -1084,7 +1084,7 @@ func TestRestoringEmptyBatch(t *testing.T) {
store.AssertLoopOutStored()
// Deliver sweep request to batcher.
batcher.sweepReqs <- sweepReq
require.NoError(t, batcher.AddSweep(&sweepReq))
// Since a batch was created we check that it registered for its primary
// sweep's spend.