rpctest: Choose flags based on provided params.
This modifies the rpctest framework to start btcd with the appropriate network flags depending on the provided parameters. Previously, it always started btcd with --simnet even if other parameters, such as those for the regression test network, were provided.
This commit is contained in:
parent
49cbaf23dd
commit
da04285e0d
2 changed files with 16 additions and 5 deletions
|
@ -12,11 +12,8 @@ import (
|
||||||
"os/exec"
|
"os/exec"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
"runtime"
|
"runtime"
|
||||||
"strings"
|
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/btcsuite/btcd/wire"
|
|
||||||
|
|
||||||
rpc "github.com/btcsuite/btcrpcclient"
|
rpc "github.com/btcsuite/btcrpcclient"
|
||||||
"github.com/btcsuite/btcutil"
|
"github.com/btcsuite/btcutil"
|
||||||
)
|
)
|
||||||
|
@ -90,8 +87,6 @@ func (n *nodeConfig) setDefaults() error {
|
||||||
// process.
|
// process.
|
||||||
func (n *nodeConfig) arguments() []string {
|
func (n *nodeConfig) arguments() []string {
|
||||||
args := []string{}
|
args := []string{}
|
||||||
// --simnet
|
|
||||||
args = append(args, fmt.Sprintf("--%s", strings.ToLower(wire.SimNet.String())))
|
|
||||||
if n.rpcUser != "" {
|
if n.rpcUser != "" {
|
||||||
// --rpcuser
|
// --rpcuser
|
||||||
args = append(args, fmt.Sprintf("--rpcuser=%s", n.rpcUser))
|
args = append(args, fmt.Sprintf("--rpcuser=%s", n.rpcUser))
|
||||||
|
|
|
@ -99,6 +99,22 @@ func New(activeNet *chaincfg.Params, handlers *btcrpcclient.NotificationHandlers
|
||||||
harnessStateMtx.Lock()
|
harnessStateMtx.Lock()
|
||||||
defer harnessStateMtx.Unlock()
|
defer harnessStateMtx.Unlock()
|
||||||
|
|
||||||
|
// Add a flag for the appropriate network type based on the provided
|
||||||
|
// chain params.
|
||||||
|
switch activeNet.Net {
|
||||||
|
case wire.MainNet:
|
||||||
|
// No extra flags since mainnet is the default
|
||||||
|
case wire.TestNet3:
|
||||||
|
extraArgs = append(extraArgs, "--testnet")
|
||||||
|
case wire.TestNet:
|
||||||
|
extraArgs = append(extraArgs, "--regtest")
|
||||||
|
case wire.SimNet:
|
||||||
|
extraArgs = append(extraArgs, "--simnet")
|
||||||
|
default:
|
||||||
|
return nil, fmt.Errorf("rpctest.New must be called with one " +
|
||||||
|
"of the supported chain networks")
|
||||||
|
}
|
||||||
|
|
||||||
harnessID := strconv.Itoa(int(numTestInstances))
|
harnessID := strconv.Itoa(int(numTestInstances))
|
||||||
nodeTestData, err := ioutil.TempDir("", "rpctest-"+harnessID)
|
nodeTestData, err := ioutil.TempDir("", "rpctest-"+harnessID)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
|
Loading…
Add table
Reference in a new issue