53 lines
1015 B
Go
53 lines
1015 B
Go
package server
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"net"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/spf13/viper"
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/tendermint/tendermint/libs/cli"
|
|
)
|
|
|
|
// Get a free address for a test tendermint server
|
|
// protocol is either tcp, http, etc
|
|
func FreeTCPAddr() (addr, port string, err error) {
|
|
l, err := net.Listen("tcp", "0.0.0.0:0")
|
|
if err != nil {
|
|
return "", "", err
|
|
}
|
|
|
|
closer := func() {
|
|
err := l.Close()
|
|
if err != nil {
|
|
// TODO: Handle with #870
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
defer closer()
|
|
|
|
portI := l.Addr().(*net.TCPAddr).Port
|
|
port = fmt.Sprintf("%d", portI)
|
|
addr = fmt.Sprintf("tcp://0.0.0.0:%s", port)
|
|
return
|
|
}
|
|
|
|
// setupViper creates a homedir to run inside,
|
|
// and returns a cleanup function to defer
|
|
func setupViper(t *testing.T) func() {
|
|
rootDir, err := ioutil.TempDir("", "mock-sdk-cmd")
|
|
require.Nil(t, err)
|
|
viper.Set(cli.HomeFlag, rootDir)
|
|
return func() {
|
|
err := os.RemoveAll(rootDir)
|
|
if err != nil {
|
|
// TODO: Handle with #870
|
|
panic(err)
|
|
}
|
|
}
|
|
}
|