wavelength/waved/indexer_proof_keyops_test.go at main · lightninglabs/wavelength · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
package waved

import (
	"context"
	"testing"

	"github.com/btcsuite/btcd/btcec/v2"
	"github.com/lightninglabs/wavelength/indexer"
	"github.com/lightningnetwork/lnd/keychain"
	"github.com/stretchr/testify/require"
)

type stubProofKeyBackend struct {
	deriveKeyResp     *keychain.KeyDescriptor
	deriveKeyErr      error
	deriveKeyLoc      keychain.KeyLocator
	deriveNextKeyResp *keychain.KeyDescriptor
	deriveNextKeyErr  error
	deriveNextFamily  keychain.KeyFamily
	signerKeyDesc     keychain.KeyDescriptor
}

func (b *stubProofKeyBackend) DeriveKey(_ context.Context,
	loc keychain.KeyLocator) (*keychain.KeyDescriptor, error) {

	b.deriveKeyLoc = loc

	return b.deriveKeyResp, b.deriveKeyErr
}

func (b *stubProofKeyBackend) DeriveNextKey(_ context.Context,
	family keychain.KeyFamily) (*keychain.KeyDescriptor, error) {

	b.deriveNextFamily = family

	return b.deriveNextKeyResp, b.deriveNextKeyErr
}

func (b *stubProofKeyBackend) ProofSigner(
	keyDesc keychain.KeyDescriptor) indexer.SchnorrSigner {

	b.signerKeyDesc = keyDesc

	return indexer.NewKeyRingSchnorrSigner(nil, keyDesc)
}

func TestIndexerProofKey(t *testing.T) {
	t.Parallel()

	privKey, err := btcec.NewPrivateKey()
	require.NoError(t, err)

	loc := keychain.KeyLocator{
		Family: oorReceiveKeyFamily,
		Index:  7,
	}
	wantDesc := &keychain.KeyDescriptor{
		KeyLocator: loc,
		PubKey:     privKey.PubKey(),
	}
	backend := &stubProofKeyBackend{
		deriveKeyResp: wantDesc,
	}
	server := &Server{
		proofKeyBackend: backend,
	}

	gotDesc, signer, err := server.IndexerProofKey(t.Context(), loc)
	require.NoError(t, err)
	require.Equal(t, loc, backend.deriveKeyLoc)
	require.Equal(t, wantDesc, gotDesc)
	require.Equal(t, *wantDesc, backend.signerKeyDesc)

	pubKeySource, ok := signer.(interface {
		ProofPubKey([]byte) (*btcec.PublicKey, error)
	})
	require.True(t, ok)

	pubKey, err := pubKeySource.ProofPubKey(nil)
	require.NoError(t, err)
	require.True(t, pubKey.IsEqual(privKey.PubKey()))
}

func TestIndexerProofNextKeyOps(t *testing.T) {
	t.Parallel()

	privKey, err := btcec.NewPrivateKey()
	require.NoError(t, err)

	wantDesc := &keychain.KeyDescriptor{
		KeyLocator: keychain.KeyLocator{
			Family: oorReceiveKeyFamily,
			Index:  11,
		},
		PubKey: privKey.PubKey(),
	}
	backend := &stubProofKeyBackend{
		deriveNextKeyResp: wantDesc,
	}
	server := &Server{
		proofKeyBackend: backend,
	}

	deriveNext, signerFactory, err := server.indexerProofNextKeyOps()
	require.NoError(t, err)

	gotDesc, err := deriveNext(t.Context())
	require.NoError(t, err)
	require.Equal(t, oorReceiveKeyFamily, backend.deriveNextFamily)
	require.Equal(t, wantDesc, gotDesc)

	signer := signerFactory(*wantDesc)
	require.Equal(t, *wantDesc, backend.signerKeyDesc)

	pubKeySource, ok := signer.(interface {
		ProofPubKey([]byte) (*btcec.PublicKey, error)
	})
	require.True(t, ok)

	pubKey, err := pubKeySource.ProofPubKey(nil)
	require.NoError(t, err)
	require.True(t, pubKey.IsEqual(privKey.PubKey()))
}