cli: add tests and minor fixes for op api

Trimmed spaces around header values.

Fixed method getting forced to GET.
This commit is contained in:
Michael Schurter 2022-02-24 17:06:07 -08:00
parent 238a732098
commit a42d832f98
2 changed files with 106 additions and 5 deletions

View File

@ -106,7 +106,7 @@ func (c *OperatorAPICommand) Run(args []string) int {
flags.BoolVar(&dryrun, "dryrun", false, "") flags.BoolVar(&dryrun, "dryrun", false, "")
flags.StringVar(&filter, "filter", "", "") flags.StringVar(&filter, "filter", "", "")
flags.BoolVar(&c.verboseFlag, "verbose", false, "") flags.BoolVar(&c.verboseFlag, "verbose", false, "")
flags.StringVar(&c.method, "X", "", "") flags.StringVar(&c.method, "X", "GET", "")
flags.Var(headerFlags, "H", "") flags.Var(headerFlags, "H", "")
if err := flags.Parse(args); err != nil { if err := flags.Parse(args); err != nil {
@ -145,8 +145,6 @@ func (c *OperatorAPICommand) Run(args []string) int {
if c.method == "" { if c.method == "" {
c.method = "POST" c.method = "POST"
} }
} else {
c.method = "GET"
} }
config := c.clientConfig() config := c.clientConfig()
@ -339,7 +337,7 @@ func (c *OperatorAPICommand) apiToCurl(config *api.Config, headers http.Header,
// environment variable. // environment variable.
if headers.Get("X-Nomad-Token") == "" { if headers.Get("X-Nomad-Token") == "" {
if c.Meta.token != "" { if c.Meta.token != "" {
parts = append(parts, fmt.Sprintf(`-H "X-Nomad-Token: %s"`, c.Meta.token)) parts = append(parts, fmt.Sprintf(`-H 'X-Nomad-Token: %s'`, c.Meta.token))
} else if v := os.Getenv("NOMAD_TOKEN"); v != "" { } else if v := os.Getenv("NOMAD_TOKEN"); v != "" {
parts = append(parts, `-H "X-Nomad-Token: ${NOMAD_TOKEN}"`) parts = append(parts, `-H "X-Nomad-Token: ${NOMAD_TOKEN}"`)
} }
@ -443,6 +441,6 @@ func (h *headerFlags) Set(v string) error {
return fmt.Errorf("Headers must be in the form 'Key: Value' but found: %q", v) return fmt.Errorf("Headers must be in the form 'Key: Value' but found: %q", v)
} }
h.headers.Add(parts[0], parts[1]) h.headers.Add(parts[0], strings.TrimSpace(parts[1]))
return nil return nil
} }

View File

@ -0,0 +1,103 @@
package command
import (
"bytes"
"net/http"
"net/http/httptest"
"net/url"
"testing"
"time"
"github.com/mitchellh/cli"
"github.com/stretchr/testify/require"
)
// TestOperatorAPICommand_Paths asserts that the op api command normalizes
// various path formats to the proper full address.
func TestOperatorAPICommand_Paths(t *testing.T) {
hits := make(chan *url.URL, 1)
ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
hits <- r.URL
}))
defer ts.Close()
// Always expect the same URL to be hit
expected := "/v1/jobs"
buf := bytes.NewBuffer(nil)
ui := &cli.BasicUi{
ErrorWriter: buf,
Writer: buf,
}
cmd := &OperatorAPICommand{Meta: Meta{Ui: ui}}
// Assert that absolute paths are appended to the configured address
exitCode := cmd.Run([]string{"-address=" + ts.URL, "/v1/jobs"})
require.Zero(t, exitCode, buf.String())
select {
case hit := <-hits:
require.Equal(t, expected, hit.String())
case <-time.After(10 * time.Second):
t.Fatalf("timed out waiting for hit")
}
buf.Reset()
// Assert that full URLs are used as-is even if an invalid address is
// set.
exitCode = cmd.Run([]string{"-address=ftp://127.0.0.2:1", ts.URL + "/v1/jobs"})
require.Zero(t, exitCode, buf.String())
select {
case hit := <-hits:
require.Equal(t, expected, hit.String())
case <-time.After(10 * time.Second):
t.Fatalf("timed out waiting for hit")
}
buf.Reset()
// Assert that URLs lacking a scheme are used even if an invalid
// address is set.
exitCode = cmd.Run([]string{"-address=ftp://127.0.0.2:1", ts.Listener.Addr().String() + "/v1/jobs"})
require.Zero(t, exitCode, buf.String())
select {
case hit := <-hits:
require.Equal(t, expected, hit.String())
case <-time.After(10 * time.Second):
t.Fatalf("timed out waiting for hit")
}
}
// TestOperatorAPICommand_Curl asserts that -dryrun outputs a valid curl
// command.
func TestOperatorAPICommand_Curl(t *testing.T) {
buf := bytes.NewBuffer(nil)
ui := &cli.BasicUi{
ErrorWriter: buf,
Writer: buf,
}
cmd := &OperatorAPICommand{Meta: Meta{Ui: ui}}
exitCode := cmd.Run([]string{
"-dryrun",
"-address=http://127.0.0.1:1",
"-region=not even a valid region",
`-filter=this == "that" or this != "foo"`,
"-X", "POST",
"-token=acl-token",
"-H", "Some-Other-Header: ok",
"/url",
})
require.Zero(t, exitCode, buf.String())
expected := `curl \
-X POST \
-H 'Some-Other-Header: ok' \
-H 'X-Nomad-Token: acl-token' \
http://127.0.0.1:1/url?filter=this+%3D%3D+%22that%22+or+this+%21%3D+%22foo%22&region=not+even+a+valid+region
`
require.Equal(t, expected, buf.String())
}