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:
Dave Collins 2016-10-19 16:38:21 -05:00
parent 49cbaf23dd
commit da04285e0d
No known key found for this signature in database
GPG key ID: B8904D9D9C93D1F2
2 changed files with 16 additions and 5 deletions

View file

@ -12,11 +12,8 @@ import (
"os/exec"
"path/filepath"
"runtime"
"strings"
"time"
"github.com/btcsuite/btcd/wire"
rpc "github.com/btcsuite/btcrpcclient"
"github.com/btcsuite/btcutil"
)
@ -90,8 +87,6 @@ func (n *nodeConfig) setDefaults() error {
// process.
func (n *nodeConfig) arguments() []string {
args := []string{}
// --simnet
args = append(args, fmt.Sprintf("--%s", strings.ToLower(wire.SimNet.String())))
if n.rpcUser != "" {
// --rpcuser
args = append(args, fmt.Sprintf("--rpcuser=%s", n.rpcUser))

View file

@ -99,6 +99,22 @@ func New(activeNet *chaincfg.Params, handlers *btcrpcclient.NotificationHandlers
harnessStateMtx.Lock()
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))
nodeTestData, err := ioutil.TempDir("", "rpctest-"+harnessID)
if err != nil {