115 lines
3.5 KiB
Go
115 lines
3.5 KiB
Go
// Copyright (c) 2013-2014 The btcsuite developers
|
|
// Use of this source code is governed by an ISC
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package memdb_test
|
|
|
|
import (
|
|
"reflect"
|
|
"testing"
|
|
|
|
"github.com/btcsuite/btcd/chaincfg"
|
|
"github.com/btcsuite/btcd/database"
|
|
"github.com/btcsuite/btcd/database/memdb"
|
|
"github.com/btcsuite/btcd/wire"
|
|
"github.com/btcsuite/btcutil"
|
|
)
|
|
|
|
// TestClosed ensure calling the interface functions on a closed database
|
|
// returns appropriate errors for the interface functions that return errors
|
|
// and does not panic or otherwise misbehave for functions which do not return
|
|
// errors.
|
|
func TestClosed(t *testing.T) {
|
|
db, err := database.CreateDB("memdb")
|
|
if err != nil {
|
|
t.Errorf("Failed to open test database %v", err)
|
|
return
|
|
}
|
|
_, err = db.InsertBlock(btcutil.NewBlock(chaincfg.MainNetParams.GenesisBlock))
|
|
if err != nil {
|
|
t.Errorf("InsertBlock: %v", err)
|
|
}
|
|
if err := db.Close(); err != nil {
|
|
t.Errorf("Close: unexpected error %v", err)
|
|
}
|
|
|
|
genesisHash := chaincfg.MainNetParams.GenesisHash
|
|
if err := db.DropAfterBlockBySha(genesisHash); err != memdb.ErrDbClosed {
|
|
t.Errorf("DropAfterBlockBySha: unexpected error %v", err)
|
|
}
|
|
|
|
if _, err := db.ExistsSha(genesisHash); err != memdb.ErrDbClosed {
|
|
t.Errorf("ExistsSha: Unexpected error: %v", err)
|
|
}
|
|
|
|
if _, err := db.FetchBlockBySha(genesisHash); err != memdb.ErrDbClosed {
|
|
t.Errorf("FetchBlockBySha: unexpected error %v", err)
|
|
}
|
|
|
|
if _, err := db.FetchBlockShaByHeight(0); err != memdb.ErrDbClosed {
|
|
t.Errorf("FetchBlockShaByHeight: unexpected error %v", err)
|
|
}
|
|
|
|
if _, err := db.FetchHeightRange(0, 1); err != memdb.ErrDbClosed {
|
|
t.Errorf("FetchHeightRange: unexpected error %v", err)
|
|
}
|
|
|
|
genesisCoinbaseTx := chaincfg.MainNetParams.GenesisBlock.Transactions[0]
|
|
coinbaseHash := genesisCoinbaseTx.TxSha()
|
|
if _, err := db.ExistsTxSha(&coinbaseHash); err != memdb.ErrDbClosed {
|
|
t.Errorf("ExistsTxSha: unexpected error %v", err)
|
|
}
|
|
|
|
if _, err := db.FetchTxBySha(genesisHash); err != memdb.ErrDbClosed {
|
|
t.Errorf("FetchTxBySha: unexpected error %v", err)
|
|
}
|
|
|
|
requestHashes := []*wire.ShaHash{genesisHash}
|
|
reply := db.FetchTxByShaList(requestHashes)
|
|
if len(reply) != len(requestHashes) {
|
|
t.Errorf("FetchUnSpentTxByShaList unexpected number of replies "+
|
|
"got: %d, want: %d", len(reply), len(requestHashes))
|
|
}
|
|
for i, txLR := range reply {
|
|
wantReply := &database.TxListReply{
|
|
Sha: requestHashes[i],
|
|
Err: memdb.ErrDbClosed,
|
|
}
|
|
if !reflect.DeepEqual(wantReply, txLR) {
|
|
t.Errorf("FetchTxByShaList unexpected reply\ngot: %v\n"+
|
|
"want: %v", txLR, wantReply)
|
|
}
|
|
}
|
|
|
|
reply = db.FetchUnSpentTxByShaList(requestHashes)
|
|
if len(reply) != len(requestHashes) {
|
|
t.Errorf("FetchUnSpentTxByShaList unexpected number of replies "+
|
|
"got: %d, want: %d", len(reply), len(requestHashes))
|
|
}
|
|
for i, txLR := range reply {
|
|
wantReply := &database.TxListReply{
|
|
Sha: requestHashes[i],
|
|
Err: memdb.ErrDbClosed,
|
|
}
|
|
if !reflect.DeepEqual(wantReply, txLR) {
|
|
t.Errorf("FetchUnSpentTxByShaList unexpected reply\n"+
|
|
"got: %v\nwant: %v", txLR, wantReply)
|
|
}
|
|
}
|
|
|
|
if _, _, err := db.NewestSha(); err != memdb.ErrDbClosed {
|
|
t.Errorf("NewestSha: unexpected error %v", err)
|
|
}
|
|
|
|
if err := db.Sync(); err != memdb.ErrDbClosed {
|
|
t.Errorf("Sync: unexpected error %v", err)
|
|
}
|
|
|
|
if err := db.RollbackClose(); err != memdb.ErrDbClosed {
|
|
t.Errorf("RollbackClose: unexpected error %v", err)
|
|
}
|
|
|
|
if err := db.Close(); err != memdb.ErrDbClosed {
|
|
t.Errorf("Close: unexpected error %v", err)
|
|
}
|
|
}
|