199 lines
6.7 KiB
Go
199 lines
6.7 KiB
Go
// Copyright 2016 The go-ethereum Authors
|
|
// This file is part of go-ethereum.
|
|
//
|
|
// go-ethereum is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// go-ethereum is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU General Public License
|
|
// along with go-ethereum. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
"strconv"
|
|
"testing"
|
|
)
|
|
|
|
var customGenesisTests = []struct {
|
|
genesis string
|
|
query string
|
|
result string
|
|
}{
|
|
// Genesis file with an empty chain configuration (ensure missing fields work)
|
|
{
|
|
genesis: `{
|
|
"alloc" : {},
|
|
"coinbase" : "0x0000000000000000000000000000000000000000",
|
|
"difficulty" : "0x20000",
|
|
"extraData" : "",
|
|
"gasLimit" : "0x2fefd8",
|
|
"nonce" : "0x0000000000001338",
|
|
"mixhash" : "0x0000000000000000000000000000000000000000000000000000000000000000",
|
|
"parentHash" : "0x0000000000000000000000000000000000000000000000000000000000000000",
|
|
"timestamp" : "0x00",
|
|
"config" : {
|
|
"terminalTotalDifficultyPassed": true
|
|
}
|
|
}`,
|
|
query: "eth.getBlock(0).nonce",
|
|
result: "0x0000000000001338",
|
|
},
|
|
// Genesis file with specific chain configurations
|
|
{
|
|
genesis: `{
|
|
"alloc" : {},
|
|
"coinbase" : "0x0000000000000000000000000000000000000000",
|
|
"difficulty" : "0x20000",
|
|
"extraData" : "",
|
|
"gasLimit" : "0x2fefd8",
|
|
"nonce" : "0x0000000000001339",
|
|
"mixhash" : "0x0000000000000000000000000000000000000000000000000000000000000000",
|
|
"parentHash" : "0x0000000000000000000000000000000000000000000000000000000000000000",
|
|
"timestamp" : "0x00",
|
|
"config" : {
|
|
"homesteadBlock" : 42,
|
|
"daoForkBlock" : 141,
|
|
"daoForkSupport" : true,
|
|
"terminalTotalDifficultyPassed" : true
|
|
}
|
|
}`,
|
|
query: "eth.getBlock(0).nonce",
|
|
result: "0x0000000000001339",
|
|
},
|
|
}
|
|
|
|
// Tests that initializing Geth with a custom genesis block and chain definitions
|
|
// work properly.
|
|
func TestCustomGenesis(t *testing.T) {
|
|
t.Parallel()
|
|
for i, tt := range customGenesisTests {
|
|
// Create a temporary data directory to use and inspect later
|
|
datadir := t.TempDir()
|
|
|
|
// Initialize the data directory with the custom genesis block
|
|
json := filepath.Join(datadir, "genesis.json")
|
|
if err := os.WriteFile(json, []byte(tt.genesis), 0600); err != nil {
|
|
t.Fatalf("test %d: failed to write genesis file: %v", i, err)
|
|
}
|
|
runGeth(t, "--datadir", datadir, "init", json).WaitExit()
|
|
|
|
// Query the custom genesis block
|
|
geth := runGeth(t, "--networkid", "1337", "--syncmode=full", "--cache", "16",
|
|
"--datadir", datadir, "--maxpeers", "0", "--port", "0", "--authrpc.port", "0",
|
|
"--nodiscover", "--nat", "none", "--ipcdisable",
|
|
"--exec", tt.query, "console")
|
|
geth.ExpectRegexp(tt.result)
|
|
geth.ExpectExit()
|
|
}
|
|
}
|
|
|
|
// TestCustomBackend that the backend selection and detection (leveldb vs pebble) works properly.
|
|
func TestCustomBackend(t *testing.T) {
|
|
t.Parallel()
|
|
// Test pebble, but only on 64-bit platforms
|
|
if strconv.IntSize != 64 {
|
|
t.Skip("Custom backends are only available on 64-bit platform")
|
|
}
|
|
genesis := `{
|
|
"alloc" : {},
|
|
"coinbase" : "0x0000000000000000000000000000000000000000",
|
|
"difficulty" : "0x20000",
|
|
"extraData" : "",
|
|
"gasLimit" : "0x2fefd8",
|
|
"nonce" : "0x0000000000001338",
|
|
"mixhash" : "0x0000000000000000000000000000000000000000000000000000000000000000",
|
|
"parentHash" : "0x0000000000000000000000000000000000000000000000000000000000000000",
|
|
"timestamp" : "0x00",
|
|
"config" : {
|
|
"terminalTotalDifficultyPassed": true
|
|
}
|
|
}`
|
|
type backendTest struct {
|
|
initArgs []string
|
|
initExpect string
|
|
execArgs []string
|
|
execExpect string
|
|
}
|
|
testfunc := func(t *testing.T, tt backendTest) error {
|
|
// Create a temporary data directory to use and inspect later
|
|
datadir := t.TempDir()
|
|
|
|
// Initialize the data directory with the custom genesis block
|
|
json := filepath.Join(datadir, "genesis.json")
|
|
if err := os.WriteFile(json, []byte(genesis), 0600); err != nil {
|
|
return fmt.Errorf("failed to write genesis file: %v", err)
|
|
}
|
|
{ // Init
|
|
args := append(tt.initArgs, "--datadir", datadir, "init", json)
|
|
geth := runGeth(t, args...)
|
|
geth.ExpectRegexp(tt.initExpect)
|
|
geth.ExpectExit()
|
|
}
|
|
{ // Exec + query
|
|
args := append(tt.execArgs, "--networkid", "1337", "--syncmode=full", "--cache", "16",
|
|
"--datadir", datadir, "--maxpeers", "0", "--port", "0", "--authrpc.port", "0",
|
|
"--nodiscover", "--nat", "none", "--ipcdisable",
|
|
"--exec", "eth.getBlock(0).nonce", "console")
|
|
geth := runGeth(t, args...)
|
|
geth.ExpectRegexp(tt.execExpect)
|
|
geth.ExpectExit()
|
|
}
|
|
return nil
|
|
}
|
|
for i, tt := range []backendTest{
|
|
{ // When not specified, it should default to pebble
|
|
execArgs: []string{"--db.engine", "pebble"},
|
|
execExpect: "0x0000000000001338",
|
|
},
|
|
{ // Explicit leveldb
|
|
initArgs: []string{"--db.engine", "leveldb"},
|
|
execArgs: []string{"--db.engine", "leveldb"},
|
|
execExpect: "0x0000000000001338",
|
|
},
|
|
{ // Explicit leveldb first, then autodiscover
|
|
initArgs: []string{"--db.engine", "leveldb"},
|
|
execExpect: "0x0000000000001338",
|
|
},
|
|
{ // Explicit pebble
|
|
initArgs: []string{"--db.engine", "pebble"},
|
|
execArgs: []string{"--db.engine", "pebble"},
|
|
execExpect: "0x0000000000001338",
|
|
},
|
|
{ // Explicit pebble, then auto-discover
|
|
initArgs: []string{"--db.engine", "pebble"},
|
|
execExpect: "0x0000000000001338",
|
|
},
|
|
{ // Can't start pebble on top of leveldb
|
|
initArgs: []string{"--db.engine", "leveldb"},
|
|
execArgs: []string{"--db.engine", "pebble"},
|
|
execExpect: `Fatal: Could not open database: db.engine choice was pebble but found pre-existing leveldb database in specified data directory`,
|
|
},
|
|
{ // Can't start leveldb on top of pebble
|
|
initArgs: []string{"--db.engine", "pebble"},
|
|
execArgs: []string{"--db.engine", "leveldb"},
|
|
execExpect: `Fatal: Could not open database: db.engine choice was leveldb but found pre-existing pebble database in specified data directory`,
|
|
},
|
|
{ // Reject invalid backend choice
|
|
initArgs: []string{"--db.engine", "mssql"},
|
|
initExpect: `Fatal: Invalid choice for db.engine 'mssql', allowed 'leveldb' or 'pebble'`,
|
|
// Since the init fails, this will return the (default) mainnet genesis
|
|
// block nonce
|
|
execExpect: `0x0000000000000042`,
|
|
},
|
|
} {
|
|
if err := testfunc(t, tt); err != nil {
|
|
t.Fatalf("test %d-leveldb: %v", i, err)
|
|
}
|
|
}
|
|
}
|