11d7cae82b
Previously, rpctest would start a btcd node using the btcd executable in the environment PATH. This caused difficult-to-find issues where the code would be tested against an older version of btcd, or another fork entirely. Now it compiles btcd the first time it is needed and uses that fresh version when launching nodes.
74 lines
2.1 KiB
Go
74 lines
2.1 KiB
Go
// Copyright (c) 2017 The btcsuite developers
|
|
// Use of this source code is governed by an ISC
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package rpctest
|
|
|
|
import (
|
|
"fmt"
|
|
"go/build"
|
|
"os/exec"
|
|
"path/filepath"
|
|
"runtime"
|
|
"sync"
|
|
)
|
|
|
|
var (
|
|
// compileMtx guards access to the executable path so that the project is
|
|
// only compiled once.
|
|
compileMtx sync.Mutex
|
|
|
|
// executablePath is the path to the compiled executable. This is the empty
|
|
// string until btcd is compiled. This should not be accessed directly;
|
|
// instead use the function btcdExecutablePath().
|
|
executablePath string
|
|
)
|
|
|
|
// btcdExecutablePath returns a path to the btcd executable to be used by
|
|
// rpctests. To ensure the code tests against the most up-to-date version of
|
|
// btcd, this method compiles btcd the first time it is called. After that, the
|
|
// generated binary is used for subsequent test harnesses. The executable file
|
|
// is not cleaned up, but since it lives at a static path in a temp directory,
|
|
// it is not a big deal.
|
|
func btcdExecutablePath() (string, error) {
|
|
compileMtx.Lock()
|
|
defer compileMtx.Unlock()
|
|
|
|
// If btcd has already been compiled, just use that.
|
|
if len(executablePath) != 0 {
|
|
return executablePath, nil
|
|
}
|
|
|
|
testDir, err := baseDir()
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
// Determine import path of this package. Not necessarily btcsuite/btcd if
|
|
// this is a forked repo.
|
|
_, rpctestDir, _, ok := runtime.Caller(1)
|
|
if !ok {
|
|
return "", fmt.Errorf("Cannot get path to btcd source code")
|
|
}
|
|
btcdPkgPath := filepath.Join(rpctestDir, "..", "..", "..")
|
|
btcdPkg, err := build.ImportDir(btcdPkgPath, build.FindOnly)
|
|
if err != nil {
|
|
return "", fmt.Errorf("Failed to build btcd: %v", err)
|
|
}
|
|
|
|
// Build btcd and output an executable in a static temp path.
|
|
outputPath := filepath.Join(testDir, "btcd")
|
|
if runtime.GOOS == "windows" {
|
|
outputPath += ".exe"
|
|
}
|
|
cmd := exec.Command("go", "build", "-o", outputPath, btcdPkg.ImportPath)
|
|
err = cmd.Run()
|
|
if err != nil {
|
|
return "", fmt.Errorf("Failed to build btcd: %v", err)
|
|
}
|
|
|
|
// Save executable path so future calls do not recompile.
|
|
executablePath = outputPath
|
|
return executablePath, nil
|
|
}
|