go-pulse/pulse/sacrifice_credits_test.go

package pulse

import (
	"math/big"
	"testing"

	"github.com/ethereum/go-ethereum/common"
	"github.com/ethereum/go-ethereum/common/math"
	"github.com/ethereum/go-ethereum/core/rawdb"
	"github.com/ethereum/go-ethereum/core/state"
	"github.com/ethereum/go-ethereum/params"
	"github.com/ethereum/go-ethereum/trie"
)

func TestApplySacrificeCredits(t *testing.T) {
	// Init
	var pulseChainTestnetTreasuryBalance math.HexOrDecimal256
	pulseChainTestnetTreasuryBalance.UnmarshalText([]byte("0xC9F2C9CD04674EDEA40000000"))

	db := rawdb.NewMemoryDatabase()
	state, _ := state.New(common.Hash{}, state.NewDatabaseWithConfig(db, &trie.Config{Preimages: true}), nil)
	treasury := &params.Treasury{
		Addr:    "0xceB59257450820132aB274ED61C49E5FD96E8868",
		Balance: &pulseChainTestnetTreasuryBalance,
	}

	// Exec
	applySacrificeCredits(state, treasury)

	// Verify
	actual := state.GetBalance(common.HexToAddress(treasury.Addr))
	expected := (*big.Int)(treasury.Balance)
	if actual.Cmp(expected) != 0 {
		t.Errorf("Invalid treasury balance, actual: %d, expected: %d", actual, expected)
	} else {
		t.Log("Treasury allocating successful")
	}

	// from the credits.csv file in compressed-allocations
	actual = state.GetBalance(common.HexToAddress("0x000000005dCEE11e13fb536Fa40d65450F53c5a8"))
	expected, _ = new(big.Int).SetString("64000000000000000000", 10)

	if actual.Cmp(expected) != 0 {
		t.Errorf("Invalid sacrifice credit balance, actual: %d, expected: %d", actual, expected)
	} else {
		t.Log("Sacrifice allocation successful")
	}
}