mirror of
https://github.com/lightninglabs/loop.git
synced 2026-08-17 13:06:51 +02:00
991830336f
Associate every deposit with the static address parameters that created it. This lets restored deposits recover the correct script and signing keys instead of assuming the legacy root address.
98 lines
2.4 KiB
Go
98 lines
2.4 KiB
Go
package deposit
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"testing"
|
|
|
|
"github.com/btcsuite/btcd/wire"
|
|
"github.com/jackc/pgx/v5"
|
|
"github.com/lightninglabs/loop/fsm"
|
|
"github.com/lightninglabs/loop/loopdb/sqlc"
|
|
"github.com/lightninglabs/loop/staticaddr/script"
|
|
"github.com/lightningnetwork/lnd/lntypes"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestCreateDepositRejectsUnpersistedAddress(t *testing.T) {
|
|
store := NewSqlStore(nil)
|
|
deposit := &Deposit{
|
|
AddressParams: &script.Parameters{},
|
|
}
|
|
|
|
err := store.CreateDeposit(context.Background(), deposit)
|
|
require.ErrorContains(t, err, "static address ID must be set")
|
|
}
|
|
|
|
func TestToDeposit(t *testing.T) {
|
|
depositID, err := GetRandomDepositID()
|
|
require.NoError(t, err)
|
|
|
|
swapHash, err := lntypes.MakeHash(dummyHashBytes())
|
|
require.NoError(t, err)
|
|
|
|
tx := wire.NewMsgTx(2)
|
|
txHash := tx.TxHash()
|
|
|
|
tests := []struct {
|
|
name string
|
|
row sqlc.AllDepositsRow
|
|
lastUpdate sqlc.DepositUpdate
|
|
expectErr bool
|
|
}{
|
|
{
|
|
name: "fully valid data",
|
|
row: sqlc.AllDepositsRow{
|
|
DepositID: depositID[:],
|
|
TxHash: txHash[:],
|
|
Amount: 100000000,
|
|
ConfirmationHeight: 123456,
|
|
SwapHash: swapHash[:],
|
|
},
|
|
lastUpdate: sqlc.DepositUpdate{
|
|
UpdateState: "completed",
|
|
},
|
|
expectErr: false,
|
|
},
|
|
{
|
|
name: "fully valid data",
|
|
row: sqlc.AllDepositsRow{
|
|
DepositID: depositID[:],
|
|
TxHash: txHash[:],
|
|
Amount: 100000000,
|
|
ConfirmationHeight: 123456,
|
|
},
|
|
lastUpdate: sqlc.DepositUpdate{
|
|
UpdateState: "completed",
|
|
},
|
|
expectErr: false,
|
|
},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
t.Run(test.name, func(t *testing.T) {
|
|
result, err := ToDeposit(test.row, test.lastUpdate)
|
|
if test.expectErr {
|
|
require.Error(t, err)
|
|
require.Nil(t, result)
|
|
} else {
|
|
require.NoError(t, err)
|
|
require.NotNil(t, result)
|
|
require.Equal(t, fsm.StateType(test.lastUpdate.UpdateState), result.state)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func dummyHashBytes() []byte {
|
|
return []byte{0x04, 0x05, 0x06, 0x07, 0x08, 0x09, 0x0a, 0x0b,
|
|
0x0c, 0x0d, 0x0e, 0x0f, 0x10, 0x11, 0x12, 0x13, 0x14, 0x15,
|
|
0x16, 0x17, 0x18, 0x19, 0x1a, 0x1b, 0x1c, 0x1d, 0x1e, 0x1f,
|
|
0x20, 0x21, 0x22, 0x23}
|
|
}
|
|
|
|
// TestErrNoRows ensures that pgx.ErrNoRows is a wrapped sql.ErrNoRows, so we
|
|
// don't have to check against both of them.
|
|
func TestErrNoRows(t *testing.T) {
|
|
require.ErrorIs(t, pgx.ErrNoRows, sql.ErrNoRows)
|
|
}
|