mirror of
https://github.com/featurebasedb/featurebase.git
synced 2026-10-07 19:37:51 +00:00
372b7ca1c4
1. fixed a couple bugs with config file reading. 2. made cmd.Serve a global variable so tests could inspect it. 3. added table style testing for pilosa server 4. added the ability to stop server programmatically which also causes the cobra command running it to return.
71 lines
1.7 KiB
Go
71 lines
1.7 KiB
Go
package cmd_test
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"io/ioutil"
|
|
"os"
|
|
"strings"
|
|
"testing"
|
|
|
|
"time"
|
|
|
|
"github.com/pilosa/pilosa/cmd"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
func failErr(t *testing.T, err error, context ...string) {
|
|
ctx := strings.Join(context, "; ")
|
|
if err != nil {
|
|
t.Fatal(ctx, ": ", err)
|
|
}
|
|
}
|
|
|
|
// tExec executes the given `cmd`, which will be writing its output to `w`, and
|
|
// can be read from `out`. It will fail the test if the command does not return
|
|
// within 1 second. Useful for testing help messages and such.
|
|
func tExec(t *testing.T, cmd *cobra.Command, out io.Reader, w io.WriteCloser) (output []byte) {
|
|
done := make(chan struct{})
|
|
go func() {
|
|
var err error
|
|
output, err = ioutil.ReadAll(out)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
close(done)
|
|
}()
|
|
fmt.Println("executing")
|
|
err := cmd.Execute()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := w.Close(); err != nil {
|
|
t.Fatalf("closing cmd's stdout: %v", err)
|
|
}
|
|
select {
|
|
case <-done:
|
|
case <-time.After(time.Second * 1):
|
|
t.Fatal("Test failed due to command execution timeout")
|
|
}
|
|
return output
|
|
}
|
|
|
|
// ExecNewRootCommand executes the pilosa root command with the given arguments
|
|
// and returns it's output. It will fail if the command does not complete within
|
|
// 1 second.
|
|
func ExecNewRootCommand(t *testing.T, args ...string) string {
|
|
out, w := io.Pipe()
|
|
rc := cmd.NewRootCommand(os.Stdin, w, w)
|
|
rc.SetArgs(args)
|
|
output := tExec(t, rc, out, w)
|
|
return string(output)
|
|
}
|
|
|
|
func TestRootCommand(t *testing.T) {
|
|
outStr := ExecNewRootCommand(t, "--help")
|
|
if !strings.Contains(outStr, "Usage:") ||
|
|
!strings.Contains(outStr, "Available Commands:") ||
|
|
!strings.Contains(outStr, "--help") {
|
|
t.Fatalf("Expected standard usage message from RootCommand, but got: %s", outStr)
|
|
}
|
|
}
|