-
Notifications
You must be signed in to change notification settings - Fork 107
/
identity_cli.go
129 lines (110 loc) · 3.91 KB
/
identity_cli.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
package e2e
import (
"context"
"fmt"
fileSigner "github.com/oasisprotocol/oasis-core/go/common/crypto/signature/signers/file"
"github.com/oasisprotocol/oasis-core/go/common/identity"
"github.com/oasisprotocol/oasis-core/go/oasis-node/cmd/common"
"github.com/oasisprotocol/oasis-core/go/oasis-test-runner/env"
"github.com/oasisprotocol/oasis-core/go/oasis-test-runner/oasis"
"github.com/oasisprotocol/oasis-core/go/oasis-test-runner/oasis/cli"
"github.com/oasisprotocol/oasis-core/go/oasis-test-runner/scenario"
)
// IdentityCLI is the identity CLI scenario.
var IdentityCLI scenario.Scenario = &identityCLIImpl{
Scenario: *NewScenario("identity-cli"),
}
type identityCLIImpl struct {
Scenario
dataDir string
}
func (sc *identityCLIImpl) Clone() scenario.Scenario {
return &identityCLIImpl{
Scenario: sc.Scenario.Clone(),
dataDir: sc.dataDir,
}
}
func (sc *identityCLIImpl) PreInit(childEnv *env.Env) error {
return nil
}
func (sc *identityCLIImpl) Init(childEnv *env.Env, net *oasis.Network) error {
dataDir, err := childEnv.NewSubDir("test-identity")
if err != nil {
return fmt.Errorf("scenario/e2e/identity_cli: init failed to create subdir: %w", err)
}
sc.dataDir = dataDir.String()
return nil
}
func (sc *identityCLIImpl) Fixture() (*oasis.NetworkFixture, error) {
return nil, nil
}
func (sc *identityCLIImpl) Run(ctx context.Context, childEnv *env.Env) error {
// Provision node's identity.
args := []string{
"identity", "init",
"--" + common.CfgDataDir, sc.dataDir,
}
nodeBinary, _ := sc.Flags.GetString(cfgNodeBinary)
if err := cli.RunSubCommand(childEnv, sc.Logger, "identity-init", nodeBinary, args); err != nil {
return fmt.Errorf("scenario/e2e/identity_cli: failed provision node's identity: %w", err)
}
if err := sc.loadIdentity(); err != nil {
return fmt.Errorf("scenario/e2e/identity_cli: %w", err)
}
if err := sc.showTLSPubkey(childEnv, false); err != nil {
return fmt.Errorf("scenario/e2e/identity_cli: %w", err)
}
if err := sc.showTLSPubkey(childEnv, true); err != nil {
return fmt.Errorf("scenario/e2e/identity_cli: %w", err)
}
if err := sc.tendermintShowAddress(childEnv, "node"); err != nil {
return fmt.Errorf("scenario/e2e/identity_cli: %w", err)
}
if err := sc.tendermintShowAddress(childEnv, "consensus"); err != nil {
return fmt.Errorf("scenario/e2e/identity_cli: %w", err)
}
return nil
}
func (sc *identityCLIImpl) loadIdentity() error {
sc.Logger.Info("loading generated entity")
factory, err := fileSigner.NewFactory(sc.dataDir, identity.RequiredSignerRoles...)
if err != nil {
return fmt.Errorf("failed to create identity file signer: %w", err)
}
if _, err = identity.Load(sc.dataDir, factory); err != nil {
return fmt.Errorf("failed to load node's identity: %w", err)
}
return nil
}
func (sc *identityCLIImpl) showTLSPubkey(childEnv *env.Env, sentry bool) error {
var subCmd string
switch sentry {
case true:
subCmd = "show-sentry-client-pubkey"
case false:
subCmd = "show-tls-pubkey"
}
sc.Logger.Info(fmt.Sprintf("running %s", subCmd))
args := []string{
"identity", subCmd,
"--" + common.CfgDataDir, sc.dataDir,
}
nodeBinary, _ := sc.Flags.GetString(cfgNodeBinary)
if out, err := cli.RunSubCommandWithOutput(childEnv, sc.Logger, subCmd, nodeBinary, args); err != nil {
return fmt.Errorf("failed to run %s: error: %w output: %s", subCmd, err, out.String())
}
return nil
}
func (sc *identityCLIImpl) tendermintShowAddress(childEnv *env.Env, addrName string) error {
subCmd := fmt.Sprintf("show-%s-address", addrName)
sc.Logger.Info(fmt.Sprintf("running tendermint %s", subCmd))
args := []string{
"identity", "tendermint", subCmd,
"--" + common.CfgDataDir, sc.dataDir,
}
nodeBinary, _ := sc.Flags.GetString(cfgNodeBinary)
if out, err := cli.RunSubCommandWithOutput(childEnv, sc.Logger, subCmd, nodeBinary, args); err != nil {
return fmt.Errorf("failed to get %s's tendermint address: error: %w output: %s", addrName, err, out.String())
}
return nil
}