Function TestMain

p2pserver/message/utils/msg_handler_test.go:107–152

(m *testing.M)

}

func TestMain(m *testing.M) {
	log.InitLog(log.InfoLog, log.Stdout)
	// Start local network server and create message router
	network = NewMockP2p()

	events.Init()
	// Initial a ledger
	var err error
	ledger.DefLedger, err = ledger.NewLedger(config.DEFAULT_DATA_DIR, 0)
	if err != nil {
		log.Fatalf("NewLedger error %s", err)
	}

	var bookkeepers []keypair.PublicKey
	testBookkeeperAccounts := make([]*account.Account, 0)
	for i := 0; i < 7; i++ {
		acc := account.NewAccount("")
		testBookkeeperAccounts = append(testBookkeeperAccounts, acc)
		bookkeepers = append(bookkeepers, acc.PublicKey)
	}

	config.DefConfig.Genesis = testGenesisConfig
	genesisConfig := config.DefConfig.Genesis

	// update peers in genesis
	for i, p := range genesisConfig.VBFT.Peers {
		if i < len(testBookkeeperAccounts) {
			p.PeerPubkey = vconfig.PubkeyID(testBookkeeperAccounts[i].PublicKey)
			p.Address = testBookkeeperAccounts[i].Address.ToBase58()
		}
	}

	block, err := genesis.BuildGenesisBlock(bookkeepers, genesisConfig)
	if err != nil {
		log.Fatalf("failed to build genesis block: %s", err)
	}
	err = ledger.DefLedger.Init(bookkeepers, block)
	if err != nil {
		log.Fatalf("DefLedger.Init error %s", err)
	}

	m.Run()

	ledger.DefLedger.Close()
	os.RemoveAll(config.DEFAULT_DATA_DIR)
}

// TestVersionHandle tests Function VersionHandle handling a version message
func TestVersionHandle(t *testing.T) {