eBlock.go

package testHelper

//A package for functions used multiple times in tests that aren't useful in production code.

import (
	"fmt"

	"github.com/FactomProject/factomd/anchor"
	"github.com/FactomProject/factomd/common/directoryBlock"
	"github.com/FactomProject/factomd/common/entryBlock"
	"github.com/FactomProject/factomd/common/interfaces"
	"github.com/FactomProject/factomd/common/primitives"
)

func CreateTestEntryBlock(p interfaces.IEntryBlock) (*entryBlock.EBlock, []*entryBlock.Entry) {
	prev, ok := p.(*entryBlock.EBlock)
	if ok == false {
		prev = nil
	}

	e := entryBlock.NewEBlock()
	entries := []*entryBlock.Entry{}

	if prev != nil {
		keyMR, err := prev.KeyMR()
		if err != nil {
			panic(err)
		}

		e.Header.SetPrevKeyMR(keyMR)
		hash, err := prev.Hash()
		if err != nil {
			panic(err)
		}
		e.Header.SetPrevFullHash(hash)
		e.Header.SetDBHeight(prev.GetHeader().GetDBHeight() + 1)

		e.Header.SetChainID(prev.GetHeader().GetChainID())
		entry := CreateTestEntry(e.Header.GetDBHeight())
		e.AddEBEntry(entry)
		e.AddEndOfMinuteMarker(uint8(e.GetDatabaseHeight()%10 + 1))
		entries = append(entries, entry)
	} else {
		e.Header.SetPrevKeyMR(primitives.NewZeroHash())
		e.Header.SetDBHeight(0)
		e.Header.SetChainID(GetChainID())

		entry := CreateFirstTestEntry()
		e.AddEBEntry(entry)
		e.AddEndOfMinuteMarker(uint8(e.GetDatabaseHeight()%10 + 1))
		entries = append(entries, entry)
	}

	return e, entries
}

func CreateTestEntryBlockWithContentN(p interfaces.IEntryBlock, content uint32) (*entryBlock.EBlock, []*entryBlock.Entry) {
	prev, ok := p.(*entryBlock.EBlock)
	if ok == false {
		prev = nil
	}

	e := entryBlock.NewEBlock()
	entries := []*entryBlock.Entry{}

	if prev != nil {
		keyMR, err := prev.KeyMR()
		if err != nil {
			panic(err)
		}

		e.Header.SetPrevKeyMR(keyMR)
		hash, err := prev.Hash()
		if err != nil {
			panic(err)
		}
		e.Header.SetPrevFullHash(hash)
		e.Header.SetDBHeight(prev.GetHeader().GetDBHeight() + 1)

		e.Header.SetChainID(prev.GetHeader().GetChainID())
		entry := CreateTestEntry(content)
		e.AddEBEntry(entry)
		e.AddEndOfMinuteMarker(uint8(e.GetDatabaseHeight()%10 + 1))
		entries = append(entries, entry)
	} else {
		e.Header.SetPrevKeyMR(primitives.NewZeroHash())
		e.Header.SetDBHeight(0)
		e.Header.SetChainID(GetChainID())

		entry := CreateFirstTestEntry()
		e.AddEBEntry(entry)
		e.AddEndOfMinuteMarker(uint8(e.GetDatabaseHeight()%10 + 1))
		entries = append(entries, entry)
	}

	return e, entries
}

func CreateTestAnchorEntryBlock(p interfaces.IEntryBlock, prevDBlock *directoryBlock.DirectoryBlock) (*entryBlock.EBlock, []*entryBlock.Entry) {
	prev, ok := p.(*entryBlock.EBlock)
	if ok == false {
		prev = nil
	}

	if prevDBlock == nil && prev != nil {
		return nil, nil
	}
	e := entryBlock.NewEBlock()
	entries := []*entryBlock.Entry{}

	if prev != nil {
		keyMR, err := prev.KeyMR()
		if err != nil {
			panic(err)
		}

		e.Header.SetPrevKeyMR(keyMR)
		hash, err := prev.Hash()
		if err != nil {
			panic(err)
		}
		e.Header.SetPrevFullHash(hash)
		e.Header.SetDBHeight(prev.GetHeader().GetDBHeight() + 1)

		e.Header.SetChainID(prev.GetHeader().GetChainID())
		entry := CreateTestAnchorEnry(prevDBlock)
		e.AddEBEntry(entry)
		e.AddEndOfMinuteMarker(uint8(e.GetDatabaseHeight()%10 + 1))
		entries = append(entries, entry)
	} else {
		e.Header.SetPrevKeyMR(primitives.NewZeroHash())
		e.Header.SetDBHeight(0)
		e.Header.SetChainID(GetAnchorChainID())

		entry := CreateFirstAnchorEntry()
		e.AddEBEntry(entry)
		e.AddEndOfMinuteMarker(uint8(e.GetDatabaseHeight()%10 + 1))
		entries = append(entries, entry)
	}

	return e, entries
}

func GetChainID() interfaces.IHash {
	return CreateFirstTestEntry().GetChainIDHash()
}

func GetAnchorChainID() interfaces.IHash {
	return CreateFirstAnchorEntry().GetChainIDHash()
}

func CreateFirstTestEntry() *entryBlock.Entry {
	answer := new(entryBlock.Entry)

	answer.Version = 1
	answer.ExtIDs = []primitives.ByteSlice{primitives.ByteSlice{Bytes: []byte("Test1")}, primitives.ByteSlice{Bytes: []byte("Test2")}}
	answer.Content = primitives.ByteSlice{Bytes: []byte("Test content, please ignore")}
	answer.ChainID = entryBlock.NewChainID(answer)

	return answer
}

func CreateFirstAnchorEntry() *entryBlock.Entry {
	answer := new(entryBlock.Entry)

	answer.Version = 0
	answer.ExtIDs = []primitives.ByteSlice{primitives.ByteSlice{Bytes: []byte("FactomAnchorChain")}}
	answer.Content = primitives.ByteSlice{Bytes: []byte("This is the Factom anchor chain, which records the anchors Factom puts on Bitcoin and other networks.\n")}
	answer.ChainID = entryBlock.NewChainID(answer)

	return answer
}

func CreateTestEntry(n uint32) *entryBlock.Entry {
	answer := entryBlock.NewEntry()

	answer.ChainID = GetChainID()
	answer.Version = 1
	answer.ExtIDs = []primitives.ByteSlice{primitives.ByteSlice{Bytes: []byte(fmt.Sprintf("ExtID %v", n))}}
	answer.Content = primitives.ByteSlice{Bytes: []byte(fmt.Sprintf("Content %v", n))}

	return answer
}

func CreateTestAnchorEnry(dBlock *directoryBlock.DirectoryBlock) *entryBlock.Entry {
	answer := entryBlock.NewEntry()

	answer.ChainID = GetAnchorChainID()
	answer.Version = 0
	answer.ExtIDs = nil

	height := dBlock.GetHeader().GetDBHeight()

	ar := new(anchor.AnchorRecord)
	ar.AnchorRecordVer = 1
	ar.DBHeight = height
	ar.KeyMR = dBlock.DatabasePrimaryIndex().String()
	ar.RecordHeight = ar.DBHeight

	ar.Bitcoin = new(anchor.BitcoinStruct)
	ar.Bitcoin.Address = "1HLoD9E4SDFFPDiYfNYnkBLQ85Y51J3Zb1"
	ar.Bitcoin.TXID = fmt.Sprintf("%x", IntToByteSlice(int(height)))
	ar.Bitcoin.BlockHeight = int32(height)
	ar.Bitcoin.BlockHash = fmt.Sprintf("%x", IntToByteSlice(255-int(height)))
	ar.Bitcoin.Offset = int32(height % 10)

	if height%2 == 0 {
		hex, err := ar.MarshalAndSign(NewPrimitivesPrivateKey(0))
		if err != nil {
			panic(err)
		}

		answer.Content = primitives.ByteSlice{Bytes: hex}
	} else {
		hex, eIDs, err := ar.MarshalAndSignV2(NewPrimitivesPrivateKey(0))
		if err != nil {
			panic(err)
		}

		answer.Content = primitives.ByteSlice{Bytes: hex}
		bs := primitives.ByteSlice{}
		err = bs.UnmarshalBinary(eIDs)
		if err != nil {
			panic(err)
		}
		answer.ExtIDs = []primitives.ByteSlice{bs}
	}

	return answer
}