mirror of
https://github.com/kaspanet/kaspad.git
synced 2025-06-07 22:56:41 +00:00
6b5b4bfb2a
* [NOD-1164] Defined DatabaseContext as the basic object of dbaccess * [NOD-1164] Update everything to use databaseContext * [NOD-1164] Fix tests * [NOD-1164] Add comments * [NOD-1164] Removed databaseContext from blockNode * [NOD-1164] Enforce DatabaseContext != nil * [NOD-1164] Remove redundant and wrong comment line
85 lines
2.3 KiB
Go
85 lines
2.3 KiB
Go
package dbaccess
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"reflect"
|
|
"testing"
|
|
|
|
"github.com/kaspanet/kaspad/dagconfig"
|
|
"github.com/kaspanet/kaspad/util"
|
|
)
|
|
|
|
func TestBlockStoreSanity(t *testing.T) {
|
|
// Create a temp db to run tests against
|
|
path, err := ioutil.TempDir("", "TestBlockStoreSanity")
|
|
if err != nil {
|
|
t.Fatalf("TestBlockStoreSanity: TempDir unexpectedly "+
|
|
"failed: %s", err)
|
|
}
|
|
databaseContext, err := New(path)
|
|
if err != nil {
|
|
t.Fatalf("TestBlockStoreSanity: Open unexpectedly "+
|
|
"failed: %s", err)
|
|
}
|
|
defer func() {
|
|
err := databaseContext.Close()
|
|
if err != nil {
|
|
t.Fatalf("TestBlockStoreSanity: Close unexpectedly "+
|
|
"failed: %s", err)
|
|
}
|
|
}()
|
|
|
|
// Store the genesis block
|
|
genesis := util.NewBlock(dagconfig.MainnetParams.GenesisBlock)
|
|
genesisHash := genesis.Hash()
|
|
genesisBytes, err := genesis.Bytes()
|
|
if err != nil {
|
|
t.Fatalf("TestBlockStoreSanity: util.Block.Bytes unexpectedly "+
|
|
"failed: %s", err)
|
|
}
|
|
dbTx, err := databaseContext.NewTx()
|
|
if err != nil {
|
|
t.Fatalf("Failed to open database "+
|
|
"transaction: %s", err)
|
|
}
|
|
defer dbTx.RollbackUnlessClosed()
|
|
err = StoreBlock(dbTx, genesisHash, genesisBytes)
|
|
if err != nil {
|
|
t.Fatalf("TestBlockStoreSanity: StoreBlock unexpectedly "+
|
|
"failed: %s", err)
|
|
}
|
|
err = dbTx.Commit()
|
|
if err != nil {
|
|
t.Fatalf("Failed to commit database "+
|
|
"transaction: %s", err)
|
|
}
|
|
|
|
// Make sure the genesis block now exists in the db
|
|
exists, err := HasBlock(databaseContext, genesisHash)
|
|
if err != nil {
|
|
t.Fatalf("TestBlockStoreSanity: HasBlock unexpectedly "+
|
|
"failed: %s", err)
|
|
}
|
|
if !exists {
|
|
t.Fatalf("TestBlockStoreSanity: just-inserted block is " +
|
|
"missing from the database")
|
|
}
|
|
|
|
// Fetch the genesis block back from the db and make sure
|
|
// that it's equal to the original
|
|
fetchedGenesisBytes, err := FetchBlock(databaseContext, genesisHash)
|
|
if err != nil {
|
|
t.Fatalf("TestBlockStoreSanity: FetchBlock unexpectedly "+
|
|
"failed: %s", err)
|
|
}
|
|
fetchedGenesis, err := util.NewBlockFromBytes(fetchedGenesisBytes)
|
|
if err != nil {
|
|
t.Fatalf("TestBlockStoreSanity: NewBlockFromBytes unexpectedly "+
|
|
"failed: %s", err)
|
|
}
|
|
if !reflect.DeepEqual(genesis.MsgBlock(), fetchedGenesis.MsgBlock()) {
|
|
t.Fatalf("TestBlockStoreSanity: just-inserted block is " +
|
|
"not equal to its database counterpart.")
|
|
}
|
|
}
|