mirror of
https://github.com/lightninglabs/loop.git
synced 2026-08-16 13:00:37 +02:00
0ee07198f6
Include https://github.com/lightninglabs/lndclient/pull/280 multi: migrate to btcd v2 modules + add WalletKit.SubmitPackage Migrate Loop imports and update Aperture and Taproot Assets to compatible revisions so this commit remains green on its own.
101 lines
2.4 KiB
Go
101 lines
2.4 KiB
Go
package withdraw
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/btcsuite/btcd/btcutil/v2"
|
|
"github.com/btcsuite/btcd/wire/v2"
|
|
"github.com/lightninglabs/loop/loopdb"
|
|
"github.com/lightninglabs/loop/staticaddr/deposit"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
// TestSqlStore tests the basic functionality of the SQLStore.
|
|
func TestSqlStore(t *testing.T) {
|
|
ctxb := context.Background()
|
|
testDb := loopdb.NewTestDB(t)
|
|
defer testDb.Close()
|
|
|
|
depositStore := deposit.NewSqlStore(testDb.BaseDB)
|
|
store := NewSqlStore(loopdb.NewTypedStore[Querier](testDb), depositStore)
|
|
|
|
newID := func() deposit.ID {
|
|
did, err := deposit.GetRandomDepositID()
|
|
require.NoError(t, err)
|
|
|
|
return did
|
|
}
|
|
|
|
d1, d2 := &deposit.Deposit{
|
|
ID: newID(),
|
|
Value: btcutil.Amount(100_000),
|
|
TimeOutSweepPkScript: []byte{
|
|
0x00, 0x14, 0x1a, 0x2b, 0x3c, 0x41,
|
|
},
|
|
},
|
|
&deposit.Deposit{
|
|
ID: newID(),
|
|
Value: btcutil.Amount(200_000),
|
|
TimeOutSweepPkScript: []byte{
|
|
0x00, 0x14, 0x1a, 0x2b, 0x3c, 0x4d,
|
|
},
|
|
}
|
|
|
|
withdrawalTx := &wire.MsgTx{
|
|
Version: 2,
|
|
TxOut: []*wire.TxOut{
|
|
{
|
|
Value: int64(d1.Value + d2.Value - 100),
|
|
PkScript: []byte{
|
|
0x00,
|
|
},
|
|
},
|
|
{
|
|
Value: int64(100),
|
|
PkScript: []byte{
|
|
0x01,
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
err := depositStore.CreateDeposit(ctxb, d1)
|
|
require.NoError(t, err)
|
|
err = depositStore.CreateDeposit(ctxb, d2)
|
|
require.NoError(t, err)
|
|
|
|
err = store.CreateWithdrawal(ctxb, []*deposit.Deposit{d1, d2})
|
|
require.NoError(t, err)
|
|
|
|
withdrawals, err := store.GetAllWithdrawals(ctxb)
|
|
require.NoError(t, err)
|
|
require.Len(t, withdrawals, 1)
|
|
require.NotEmpty(t, withdrawals[0].ID)
|
|
require.EqualValues(
|
|
t, d1.Value+d2.Value, withdrawals[0].TotalDepositAmount,
|
|
)
|
|
require.Len(t, withdrawals[0].Deposits, 2)
|
|
require.EqualValues(
|
|
t, d1.Value, withdrawals[0].Deposits[0].Value,
|
|
)
|
|
require.EqualValues(
|
|
t, d2.Value, withdrawals[0].Deposits[1].Value,
|
|
)
|
|
require.NotEmpty(t, withdrawals[0].InitiationTime)
|
|
|
|
err = store.UpdateWithdrawal(
|
|
ctxb, []*deposit.Deposit{d1, d2}, withdrawalTx, 6, []byte{0x01},
|
|
)
|
|
require.NoError(t, err)
|
|
|
|
withdrawals, err = store.GetAllWithdrawals(ctxb)
|
|
require.NoError(t, err)
|
|
require.Len(t, withdrawals, 1)
|
|
require.NotEmpty(t, withdrawals[0].TxID)
|
|
require.EqualValues(
|
|
t, d1.Value+d2.Value-100, withdrawals[0].WithdrawnAmount,
|
|
)
|
|
require.EqualValues(t, 100, withdrawals[0].ChangeAmount)
|
|
require.EqualValues(t, 6, withdrawals[0].ConfirmationHeight)
|
|
}
|