lntest/node: adds SetExtraArgs to change node behavior after restart

This commit is contained in:
Conner Fromknecht committed 2018-03-09 21:18:16 -08:00
1 parent 90dec0ad6a
commit 4bff32232c
1 file changed
+6
+6
View File
@@ -395,6 +395,12 @@ func (hn *HarnessNode) connectRPC() (*grpc.ClientConn, error) {
return grpc.Dial(hn.cfg.RPCAddr(), opts...)
}
// SetExtraArgs assigns the ExtraArgs field for the node's configuration. The
// changes will take effect on restart.
func (hn *HarnessNode) SetExtraArgs(extraArgs []string) {
hn.cfg.ExtraArgs = extraArgs
}
// cleanup cleans up all the temporary files created by the node's process.
func (hn *HarnessNode) cleanup() error {
return os.RemoveAll(hn.cfg.BaseDir)