core: hide the key generated method
This commit is contained in:
parent
67025d76fc
commit
0a51226353
|
@ -25,8 +25,7 @@ import (
|
||||||
)
|
)
|
||||||
|
|
||||||
func TestEthereumContainer(t *testing.T) {
|
func TestEthereumContainer(t *testing.T) {
|
||||||
keys := core.GenerateClusterKeys(1)
|
envs := core.SetupEnv(1)
|
||||||
envs := core.SetupEnv(keys)
|
|
||||||
defer core.Teardown(envs)
|
defer core.Teardown(envs)
|
||||||
err := core.SetupNodes(envs)
|
err := core.SetupNodes(envs)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
|
|
@ -46,18 +46,6 @@ const (
|
||||||
GenesisJson = "genesis.json"
|
GenesisJson = "genesis.json"
|
||||||
)
|
)
|
||||||
|
|
||||||
func GenerateClusterKeys(numbers int) []*ecdsa.PrivateKey {
|
|
||||||
keys := make([]*ecdsa.PrivateKey, numbers)
|
|
||||||
for i := 0; i < len(keys); i++ {
|
|
||||||
key, err := crypto.GenerateKey()
|
|
||||||
if err != nil {
|
|
||||||
panic("couldn't generate key: " + err.Error())
|
|
||||||
}
|
|
||||||
keys[i] = key
|
|
||||||
}
|
|
||||||
return keys
|
|
||||||
}
|
|
||||||
|
|
||||||
type Env struct {
|
type Env struct {
|
||||||
GethID int
|
GethID int
|
||||||
P2PPort uint16
|
P2PPort uint16
|
||||||
|
@ -73,8 +61,8 @@ func Teardown(envs []*Env) {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func SetupEnv(prvKeys []*ecdsa.PrivateKey) []*Env {
|
func SetupEnv(numbers int) []*Env {
|
||||||
envs := make([]*Env, len(prvKeys))
|
envs := make([]*Env, numbers)
|
||||||
rpcPort := defaultBaseRpcPort
|
rpcPort := defaultBaseRpcPort
|
||||||
p2pPort := defaultP2PPort
|
p2pPort := defaultP2PPort
|
||||||
|
|
||||||
|
@ -84,9 +72,14 @@ func SetupEnv(prvKeys []*ecdsa.PrivateKey) []*Env {
|
||||||
log.Fatalf("Cannot connect to Docker daemon, err: %v", err)
|
log.Fatalf("Cannot connect to Docker daemon, err: %v", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
dataDir, err := saveNodeKey(prvKeys[i])
|
key, err := crypto.GenerateKey()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
panic("Failed to save node key")
|
log.Fatalf("couldn't generate key: " + err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
dataDir, err := saveNodeKey(key)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatalf("Failed to save node key")
|
||||||
}
|
}
|
||||||
|
|
||||||
envs[i] = &Env{
|
envs[i] = &Env{
|
||||||
|
@ -94,7 +87,7 @@ func SetupEnv(prvKeys []*ecdsa.PrivateKey) []*Env {
|
||||||
P2PPort: p2pPort,
|
P2PPort: p2pPort,
|
||||||
RpcPort: rpcPort,
|
RpcPort: rpcPort,
|
||||||
DataDir: dataDir,
|
DataDir: dataDir,
|
||||||
Key: prvKeys[i],
|
Key: key,
|
||||||
Client: client,
|
Client: client,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -22,12 +22,12 @@ import (
|
||||||
)
|
)
|
||||||
|
|
||||||
func TestWriteFile(t *testing.T) {
|
func TestWriteFile(t *testing.T) {
|
||||||
keys := GenerateClusterKeys(4)
|
envs := SetupEnv(4)
|
||||||
envs := SetupEnv(keys)
|
|
||||||
err := SetupNodes(envs)
|
err := SetupNodes(envs)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatal("failed to setup nodes", err)
|
t.Fatal("failed to setup nodes", err)
|
||||||
}
|
}
|
||||||
|
defer Teardown(envs)
|
||||||
for _, env := range envs {
|
for _, env := range envs {
|
||||||
fmt.Println(fmt.Sprintf("%s%d%s%s", "geth ID:", env.GethID, ", dataDir:", env.DataDir))
|
fmt.Println(fmt.Sprintf("%s%d%s%s", "geth ID:", env.GethID, ", dataDir:", env.DataDir))
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue