github.com/igggame/nebulas-go@v2.1.0+incompatible/core/genesis_test.go (about)

     1  // Copyright (C) 2017 go-nebulas authors
     2  //
     3  // This file is part of the go-nebulas library.
     4  //
     5  // the go-nebulas library is free software: you can redistribute it and/or modify
     6  // it under the terms of the GNU General Public License as published by
     7  // the Free Software Foundation, either version 3 of the License, or
     8  // (at your option) any later version.
     9  //
    10  // the go-nebulas library is distributed in the hope that it will be useful,
    11  // but WITHOUT ANY WARRANTY; without even the implied warranty of
    12  // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    13  // GNU General Public License for more details.
    14  //
    15  // You should have received a copy of the GNU General Public License
    16  // along with the go-nebulas library.  If not, see <http://www.gnu.org/licenses/>.
    17  //
    18  
    19  package core
    20  
    21  import (
    22  	"testing"
    23  
    24  	"github.com/stretchr/testify/assert"
    25  )
    26  
    27  func TestNewGenesisBlock(t *testing.T) {
    28  	neb := testNeb(t)
    29  	chain := neb.chain
    30  	genesis := neb.chain.genesisBlock
    31  	conf := MockGenesisConf()
    32  
    33  	for _, v := range conf.TokenDistribution {
    34  		addr, _ := AddressParse(v.Address)
    35  		acc, err := genesis.worldState.GetOrCreateUserAccount(addr.Bytes())
    36  		assert.Nil(t, err)
    37  		assert.Equal(t, acc.Balance().String(), v.Value)
    38  	}
    39  
    40  	dumpConf, err := DumpGenesis(chain)
    41  	assert.Nil(t, err)
    42  	assert.Equal(t, dumpConf.Meta.ChainId, conf.Meta.ChainId)
    43  	assert.Equal(t, dumpConf.TokenDistribution, conf.TokenDistribution)
    44  }
    45  
    46  func TestInvalidAddressInTokenDistribution(t *testing.T) {
    47  	mockConf := MockGenesisConf()
    48  	mockConf.TokenDistribution[0].Address = "n1UZtMgi94oE913L2Sa2C9XwvAzNTQ82v64121"
    49  	chain := testNeb(t).chain
    50  	_, err := NewGenesisBlock(mockConf, chain)
    51  	assert.Equal(t, err, ErrInvalidAddressFormat)
    52  }