ac3adff1e8
This linter ensures that all errors are checked. This is disabled in the client directories, since its not needed on those writes
77 lines
1.5 KiB
Go
77 lines
1.5 KiB
Go
package server
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"net"
|
|
"os"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/spf13/cobra"
|
|
"github.com/spf13/viper"
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/tendermint/tmlibs/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)
|
|
}
|
|
}
|
|
}
|
|
|
|
// Run or Timout RunE of command passed in
|
|
func RunOrTimeout(cmd *cobra.Command, timeout time.Duration, t *testing.T) chan error {
|
|
done := make(chan error)
|
|
go func(out chan<- error) {
|
|
// this should NOT exit
|
|
err := cmd.RunE(nil, nil)
|
|
if err != nil {
|
|
out <- err
|
|
}
|
|
out <- fmt.Errorf("start died for unknown reasons")
|
|
}(done)
|
|
timer := time.NewTimer(timeout)
|
|
|
|
select {
|
|
case err := <-done:
|
|
require.NoError(t, err)
|
|
case <-timer.C:
|
|
return done
|
|
}
|
|
return done
|
|
}
|