aboutsummaryrefslogtreecommitdiff
path: root/test/blockinfodatabase_test.go
blob: 5205c9946f065cbfdb11a7e6052112a187645639 (plain)
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
package test

import (
	"Chain/pkg/blockchain/blockinfodatabase"
	"reflect"
	"testing"
)

func TestStoreBlockRecord(t *testing.T) {
	defer cleanUp()
	blockinfo := blockinfodatabase.New(blockinfodatabase.DefaultConfig())
	br := MockedBlockRecord()
	blockinfo.StoreBlockRecord("hash", br)
}

func TestGetSameRecord(t *testing.T) {
	defer cleanUp()
	blockinfo := blockinfodatabase.New(blockinfodatabase.DefaultConfig())
	br := MockedBlockRecord()
	blockinfo.StoreBlockRecord("hash", br)
	br2 := blockinfo.GetBlockRecord("hash")
	if !reflect.DeepEqual(br, br2) {
		t.Errorf("Block records not equal")
	}
}

func TestGetDifferentRecords(t *testing.T) {
	defer cleanUp()
	blockinfo := blockinfodatabase.New(blockinfodatabase.DefaultConfig())
	br := MockedBlockRecord()
	br2 := MockedBlockRecord()
	br2.UndoEndOffset = 20
	blockinfo.StoreBlockRecord("hash", br)
	blockinfo.StoreBlockRecord("hash2", br2)
	rbr := blockinfo.GetBlockRecord("hash")
	rbr2 := blockinfo.GetBlockRecord("hash2")
	if reflect.DeepEqual(rbr, rbr2) {
		t.Errorf("Block records should not be equal")
	}
}