149 lines
4.2 KiB
Go
149 lines
4.2 KiB
Go
package chainntnfs
|
|
|
|
import (
|
|
"bytes"
|
|
"io/ioutil"
|
|
"testing"
|
|
|
|
"github.com/btcsuite/btcd/chaincfg/chainhash"
|
|
"github.com/btcsuite/btcd/wire"
|
|
"github.com/lightningnetwork/lnd/channeldb"
|
|
)
|
|
|
|
func initHintCache(t *testing.T) *HeightHintCache {
|
|
t.Helper()
|
|
|
|
tempDir, err := ioutil.TempDir("", "kek")
|
|
if err != nil {
|
|
t.Fatalf("unable to create temp dir: %v", err)
|
|
}
|
|
db, err := channeldb.Open(tempDir)
|
|
if err != nil {
|
|
t.Fatalf("unable to create db: %v", err)
|
|
}
|
|
hintCache, err := NewHeightHintCache(db)
|
|
if err != nil {
|
|
t.Fatalf("unable to create hint cache: %v", err)
|
|
}
|
|
|
|
return hintCache
|
|
}
|
|
|
|
// TestHeightHintCacheConfirms ensures that the height hint cache properly
|
|
// caches confirm hints for transactions.
|
|
func TestHeightHintCacheConfirms(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
hintCache := initHintCache(t)
|
|
|
|
// Querying for a transaction hash not found within the cache should
|
|
// return an error indication so.
|
|
var unknownHash chainhash.Hash
|
|
_, err := hintCache.QueryConfirmHint(unknownHash)
|
|
if err != ErrConfirmHintNotFound {
|
|
t.Fatalf("expected ErrConfirmHintNotFound, got: %v", err)
|
|
}
|
|
|
|
// Now, we'll create some transaction hashes and commit them to the
|
|
// cache with the same confirm hint.
|
|
const height = 100
|
|
const numHashes = 5
|
|
txHashes := make([]chainhash.Hash, numHashes)
|
|
for i := 0; i < numHashes; i++ {
|
|
var txHash chainhash.Hash
|
|
copy(txHash[:], bytes.Repeat([]byte{byte(i)}, 32))
|
|
txHashes[i] = txHash
|
|
}
|
|
|
|
if err := hintCache.CommitConfirmHint(height, txHashes...); err != nil {
|
|
t.Fatalf("unable to add entries to cache: %v", err)
|
|
}
|
|
|
|
// With the hashes committed, we'll now query the cache to ensure that
|
|
// we're able to properly retrieve the confirm hints.
|
|
for _, txHash := range txHashes {
|
|
confirmHint, err := hintCache.QueryConfirmHint(txHash)
|
|
if err != nil {
|
|
t.Fatalf("unable to query for hint: %v", err)
|
|
}
|
|
if confirmHint != height {
|
|
t.Fatalf("expected confirm hint %d, got %d", height,
|
|
confirmHint)
|
|
}
|
|
}
|
|
|
|
// We'll also attempt to purge all of them in a single database
|
|
// transaction.
|
|
if err := hintCache.PurgeConfirmHint(txHashes...); err != nil {
|
|
t.Fatalf("unable to remove confirm hints: %v", err)
|
|
}
|
|
|
|
// Finally, we'll attempt to query for each hash. We should expect not
|
|
// to find a hint for any of them.
|
|
for _, txHash := range txHashes {
|
|
_, err := hintCache.QueryConfirmHint(txHash)
|
|
if err != ErrConfirmHintNotFound {
|
|
t.Fatalf("expected ErrConfirmHintNotFound, got :%v", err)
|
|
}
|
|
}
|
|
}
|
|
|
|
// TestHeightHintCacheSpends ensures that the height hint cache properly caches
|
|
// spend hints for outpoints.
|
|
func TestHeightHintCacheSpends(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
hintCache := initHintCache(t)
|
|
|
|
// Querying for an outpoint not found within the cache should return an
|
|
// error indication so.
|
|
var unknownOutPoint wire.OutPoint
|
|
_, err := hintCache.QuerySpendHint(unknownOutPoint)
|
|
if err != ErrSpendHintNotFound {
|
|
t.Fatalf("expected ErrSpendHintNotFound, got: %v", err)
|
|
}
|
|
|
|
// Now, we'll create some outpoints and commit them to the cache with
|
|
// the same spend hint.
|
|
const height = 100
|
|
const numOutpoints = 5
|
|
var txHash chainhash.Hash
|
|
copy(txHash[:], bytes.Repeat([]byte{0xFF}, 32))
|
|
outpoints := make([]wire.OutPoint, numOutpoints)
|
|
for i := uint32(0); i < numOutpoints; i++ {
|
|
outpoints[i] = wire.OutPoint{Hash: txHash, Index: i}
|
|
}
|
|
|
|
if err := hintCache.CommitSpendHint(height, outpoints...); err != nil {
|
|
t.Fatalf("unable to add entry to cache: %v", err)
|
|
}
|
|
|
|
// With the outpoints committed, we'll now query the cache to ensure
|
|
// that we're able to properly retrieve the confirm hints.
|
|
for _, op := range outpoints {
|
|
spendHint, err := hintCache.QuerySpendHint(op)
|
|
if err != nil {
|
|
t.Fatalf("unable to query for hint: %v", err)
|
|
}
|
|
if spendHint != height {
|
|
t.Fatalf("expected spend hint %d, got %d", height,
|
|
spendHint)
|
|
}
|
|
}
|
|
|
|
// We'll also attempt to purge all of them in a single database
|
|
// transaction.
|
|
if err := hintCache.PurgeSpendHint(outpoints...); err != nil {
|
|
t.Fatalf("unable to remove spend hint: %v", err)
|
|
}
|
|
|
|
// Finally, we'll attempt to query for each outpoint. We should expect
|
|
// not to find a hint for any of them.
|
|
for _, op := range outpoints {
|
|
_, err = hintCache.QuerySpendHint(op)
|
|
if err != ErrSpendHintNotFound {
|
|
t.Fatalf("expected ErrSpendHintNotFound, got: %v", err)
|
|
}
|
|
}
|
|
}
|