vault/command/capabilities_test.go

package command

import (
	"testing"

	"github.com/hashicorp/vault/http"
	"github.com/hashicorp/vault/vault"
	"github.com/mitchellh/cli"
)

func TestCapabilities_Args(t *testing.T) {
	core, _, _ := vault.TestCoreUnsealed(t)
	ln, _ := http.TestServer(t, core)
	defer ln.Close()

	ui := new(cli.MockUi)
	c := &CapabilitiesCommand{
		Meta: Meta{
			Ui: ui,
		},
	}

	args := []string{}
	if code := c.Run(args); code == 0 {
		t.Fatalf("expected failure due to no args")
	}

	args = []string{"invalidtoken", "test"}
	if code := c.Run(args); code == 0 {
		t.Fatalf("expected failure due to no invalid token")
	}

	/*
		args = []string{"test"}
		if code := c.Run(args); code != 0 {
			t.Fatalf("bad: %d\n\n%s", code, ui.ErrorWriter.String())
		}

			log.Printf("result1: %s\n", string(ui.OutputWriter.Bytes()))
			if !strings.Contains(string(ui.OutputWriter.Bytes()), "This is a 'root' token.") {
				t.Fatalf("bad: %s", ui.OutputWriter.String())
			}

			args = []string{string(key), "test"}
			if code := c.Run(args); code != 0 {
				t.Fatalf("bad: %d\n\n%s", code, ui.ErrorWriter.String())
			}

			log.Printf("result2: %s\n", string(ui.OutputWriter.Bytes()))
			if !strings.Contains(string(ui.OutputWriter.Bytes()), "This is a 'root' token.") {
				t.Fatalf("bad: %s", ui.OutputWriter.String())
			}
	*/
}