Skip to content

Commit

Permalink
test mint keyset rotations in wallet
Browse files Browse the repository at this point in the history
  • Loading branch information
@elnosh
elnosh committed Nov 18, 2024
1 parent 5dc5087 commit 8a36ddf
Show file tree
Hide file tree
Showing 3 changed files with 153 additions and 26 deletions.
55 changes: 38 additions & 17 deletions testutils/utils.go
View file
Original file line number Diff line number Diff line change
Expand Up @@ -187,9 +187,10 @@ func FundCashuWallet(ctx context.Context, wallet *wallet.Wallet, lnd *btcdocker.
return nil
}

func mintConfig(
func MintConfig(
lnd *btcdocker.Lnd,
port string,
derivationPathIdx uint32,
dbpath string,
dbMigrationPath string,
inputFeePpk uint,
Expand All @@ -198,6 +199,38 @@ func mintConfig(
if err := os.MkdirAll(dbpath, 0750); err != nil {
return nil, err
}

var lightningClient lightning.Client
if lnd != nil {
var err error
lightningClient, err = LndClient(lnd, dbpath)
if err != nil {
return nil, err
}
} else {
lightningClient = &lightning.FakeBackend{}
}

timeout := time.Second * 2
mintConfig := &mint.Config{
DerivationPathIdx: derivationPathIdx,
Port: port,
MintPath: dbpath,
DBMigrationPath: dbMigrationPath,
InputFeePpk: inputFeePpk,
Limits: limits,
LightningClient: lightningClient,
LogLevel: mint.Disable,
MeltTimeout: &timeout,
}

return mintConfig, nil
}

func LndClient(lnd *btcdocker.Lnd, dbpath string) (*lightning.LndClient, error) {
if err := os.MkdirAll(dbpath, 0750); err != nil {
return nil, err
}
nodeDir := lnd.LndDir

macaroonPath := filepath.Join(dbpath, "/admin.macaroon")
Expand Down Expand Up @@ -239,20 +272,7 @@ func mintConfig(
return nil, fmt.Errorf("error setting LND client: %v", err)
}

timeout := time.Second * 2
mintConfig := &mint.Config{
DerivationPathIdx: 0,
Port: port,
MintPath: dbpath,
DBMigrationPath: dbMigrationPath,
InputFeePpk: inputFeePpk,
Limits: limits,
LightningClient: lndClient,
LogLevel: mint.Disable,
MeltTimeout: &timeout,
}

return mintConfig, nil
return lndClient, nil
}

func CreateTestMint(
Expand All @@ -262,7 +282,7 @@ func CreateTestMint(
inputFeePpk uint,
limits mint.MintLimits,
) (*mint.Mint, error) {
config, err := mintConfig(lnd, "", dbpath, dbMigrationPath, inputFeePpk, limits)
config, err := MintConfig(lnd, "", 0, dbpath, dbMigrationPath, inputFeePpk, limits)
if err != nil {
return nil, err
}
Expand All @@ -277,11 +297,12 @@ func CreateTestMint(
func CreateTestMintServer(
lnd *btcdocker.Lnd,
port string,
derivationPathIdx uint32,
dbpath string,
dbMigrationPath string,
inputFeePpk uint,
) (*mint.MintServer, error) {
config, err := mintConfig(lnd, port, dbpath, dbMigrationPath, inputFeePpk, mint.MintLimits{})
config, err := MintConfig(lnd, port, derivationPathIdx, dbpath, dbMigrationPath, inputFeePpk, mint.MintLimits{})
if err != nil {
return nil, err
}
Expand Down
6 changes: 5 additions & 1 deletion wallet/wallet.go
View file
Original file line number Diff line number Diff line change
Expand Up @@ -473,7 +473,11 @@ func (w *Wallet) Receive(token cashu.Token, swapToTrusted bool) (uint64, error)
}

if swapToTrusted {
mint := &walletMint{mintURL: tokenMint, activeKeyset: *keyset}
inactiveKeysets, err := GetMintInactiveKeysets(tokenMint, w.unit)
if err != nil {
return 0, err
}
mint := &walletMint{mintURL: tokenMint, activeKeyset: *keyset, inactiveKeysets: inactiveKeysets}
amountSwapped, err := w.swapToTrusted(proofsToSwap, mint)
if err != nil {
return 0, fmt.Errorf("error swapping token to trusted mint: %v", err)
Expand Down
118 changes: 110 additions & 8 deletions wallet/wallet_integration_test.go
View file
Original file line number Diff line number Diff line change
Expand Up @@ -12,13 +12,15 @@ import (
"path/filepath"
"slices"
"testing"
"time"

"github.com/btcsuite/btcd/btcec/v2"
btcdocker "github.com/elnosh/btc-docker-test"
"github.com/elnosh/gonuts/cashu"
"github.com/elnosh/gonuts/cashu/nuts/nut05"
"github.com/elnosh/gonuts/cashu/nuts/nut11"
"github.com/elnosh/gonuts/cashu/nuts/nut12"
"github.com/elnosh/gonuts/mint"
"github.com/elnosh/gonuts/testutils"
"github.com/elnosh/gonuts/wallet"
"github.com/lightningnetwork/lnd/lnrpc"
Expand Down Expand Up @@ -85,7 +87,7 @@ func testMain(m *testing.M) int {
}

testMintPath := filepath.Join(".", "testmint1")
testMint, err := testutils.CreateTestMintServer(lnd1, "3338", testMintPath, dbMigrationPath, 0)
testMint, err := testutils.CreateTestMintServer(lnd1, "3338", 0, testMintPath, dbMigrationPath, 0)
if err != nil {
log.Println(err)
return 1
Expand All @@ -98,7 +100,7 @@ func testMain(m *testing.M) int {
}()

mintPath := filepath.Join(".", "testmintwithfees")
mintWithFees, err := testutils.CreateTestMintServer(lnd1, "8888", mintPath, dbMigrationPath, 100)
mintWithFees, err := testutils.CreateTestMintServer(lnd1, "8888", 0, mintPath, dbMigrationPath, 100)
if err != nil {
log.Println(err)
return 1
Expand Down Expand Up @@ -154,7 +156,6 @@ func TestMintTokens(t *testing.T) {
if err != nil {
t.Fatalf("got unexpected error: %v", err)
}

if mintedAmount != mintAmount {
t.Fatalf("expected proofs amount of '%v' but got '%v' instead", mintAmount, mintedAmount)
}
Expand Down Expand Up @@ -260,7 +261,7 @@ func TestReceive(t *testing.T) {
}

testMintPath := filepath.Join(".", "testmint2")
testMint, err := testutils.CreateTestMintServer(lnd2, "3339", testMintPath, dbMigrationPath, 0)
testMint, err := testutils.CreateTestMintServer(lnd2, "3339", 0, testMintPath, dbMigrationPath, 0)
if err != nil {
t.Fatal(err)
}
Expand Down Expand Up @@ -459,7 +460,7 @@ func TestMelt(t *testing.T) {
func TestMintSwap(t *testing.T) {
mint2URL := "http://127.0.0.1:8081"
testMintPath := filepath.Join(".", "testmint2")
testMint, err := testutils.CreateTestMintServer(lnd2, "8081", testMintPath, dbMigrationPath, 0)
testMint, err := testutils.CreateTestMintServer(lnd2, "8081", 0, testMintPath, dbMigrationPath, 0)
if err != nil {
t.Fatal(err)
}
Expand Down Expand Up @@ -808,6 +809,107 @@ func TestPendingProofs(t *testing.T) {
}
}

// Test wallet operations work after mint rotates to new keyset
func TestKeysetRotations(t *testing.T) {
mintURL := "http://127.0.0.1:8082"
testMintPath := filepath.Join(".", "testmintkeysetrotation")
var keysetDerivationIdx uint32 = 0
testMint, err := testutils.CreateTestMintServer(lnd1, "8082", keysetDerivationIdx, testMintPath, dbMigrationPath, 0)
if err != nil {
t.Fatal(err)
}
defer func() {
os.RemoveAll(testMintPath)
}()
go func() {
if err := testMint.Start(); err != nil {
t.Fatal(err)
}
}()

var bumpKeyset = func(mint *mint.MintServer) *mint.MintServer {
testMint.Shutdown()
keysetDerivationIdx++
testMint, err := testutils.CreateTestMintServer(lnd1, "8082", keysetDerivationIdx, testMintPath, dbMigrationPath, 0)
if err != nil {
t.Fatal(err)
}
return testMint
}

testWalletPath := filepath.Join(".", "/testkeysetrotationwallet")
testWallet, err := testutils.CreateTestWallet(testWalletPath, mintURL)
if err != nil {
t.Fatal(err)
}
defer func() {
os.RemoveAll(testWalletPath)
}()

testWalletPath2 := filepath.Join(".", "/testkeysetrotationwallet2")
testWallet2, err := testutils.CreateTestWallet(testWalletPath2, mintURL)
if err != nil {
t.Fatal(err)
}
defer func() {
os.RemoveAll(testWalletPath2)
}()

var mintAmount uint64 = 30000
mintRes, err := testWallet.RequestMint(mintAmount, testWallet.CurrentMint())
if err != nil {
t.Fatalf("error requesting mint: %v", err)
}
sendPaymentRequest := lnrpc.SendRequest{
PaymentRequest: mintRes.Request,
}
response, _ := lnd2.Client.SendPaymentSync(ctx, &sendPaymentRequest)
if len(response.PaymentError) > 0 {
t.Fatalf("error paying invoice: %v", response.PaymentError)
}

testMint = bumpKeyset(testMint)
go func() {
if err := testMint.Start(); err != nil {
t.Fatal(err)
}
}()
time.Sleep(time.Millisecond * 500)

_, err = testWallet.MintTokens(mintRes.Quote)
if err != nil {
t.Fatalf("got unexpected error minting tokens: %v", err)
}

testMint = bumpKeyset(testMint)
go func() {
if err := testMint.Start(); err != nil {
t.Fatal(err)
}
}()
time.Sleep(time.Millisecond * 500)

activeKeyset, _ := wallet.GetMintActiveKeyset(mintURL, cashu.Sat)
// SendToPubkey would require a swap so new proofs should have id from new keyset
lockedProofs, err := testWallet.SendToPubkey(210, mintURL, testWallet.GetReceivePubkey(), nil, false)
if err != nil {
t.Fatalf("unexpected getting locked proofs: %v", err)
}
if lockedProofs[0].Id != activeKeyset.Id {
t.Fatalf("expected proofs with id '%v' but got '%v'", activeKeyset.Id, lockedProofs[0].Id)
}
token, _ := cashu.NewTokenV4(lockedProofs, mintURL, cashu.Sat, false)

testMint = bumpKeyset(testMint)
go func() {
if err := testMint.Start(); err != nil {
t.Fatal(err)
}
}()
time.Sleep(time.Millisecond * 500)
_, err = testWallet2.Receive(token, false)
}

func TestWalletRestore(t *testing.T) {
mintURL := "http://127.0.0.1:3338"

Expand Down Expand Up @@ -905,7 +1007,7 @@ func testWalletRestore(

func TestHTLC(t *testing.T) {
htlcMintPath := filepath.Join(".", "htlcmint1")
htlcMint, err := testutils.CreateTestMintServer(lnd1, "8080", htlcMintPath, dbMigrationPath, 0)
htlcMint, err := testutils.CreateTestMintServer(lnd1, "8080", 0, htlcMintPath, dbMigrationPath, 0)
if err != nil {
t.Fatal(err)
}
Expand Down Expand Up @@ -981,7 +1083,7 @@ func TestHTLC(t *testing.T) {

func TestSendToPubkey(t *testing.T) {
p2pkMintPath := filepath.Join(".", "p2pkmint1")
p2pkMint, err := testutils.CreateTestMintServer(lnd1, "8889", p2pkMintPath, dbMigrationPath, 0)
p2pkMint, err := testutils.CreateTestMintServer(lnd1, "8889", 0, p2pkMintPath, dbMigrationPath, 0)
if err != nil {
t.Fatal(err)
}
Expand All @@ -994,7 +1096,7 @@ func TestSendToPubkey(t *testing.T) {
p2pkMintURL := "http://127.0.0.1:8889"

p2pkMintPath2 := filepath.Join(".", "p2pkmint2")
p2pkMint2, err := testutils.CreateTestMintServer(lnd2, "8890", p2pkMintPath2, dbMigrationPath, 0)
p2pkMint2, err := testutils.CreateTestMintServer(lnd2, "8890", 0, p2pkMintPath2, dbMigrationPath, 0)
if err != nil {
t.Fatal(err)
}
Expand Down

0 comments on commit 8a36ddf

Please sign in to comment.