1a324cf347
Functionality is split into ExternalTokenHelper, which is used if a path is given in a configuration file, and InternalTokenHelper which is used otherwise. The internal helper no longer shells out to the same Vault binary, instead performing the same actions with internal code. This avoids problems using dev mode when there are spaces in paths or when the binary is built in a container without a shell. Fixes #850 among others
79 lines
1.6 KiB
Go
79 lines
1.6 KiB
Go
package token
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/mitchellh/cli"
|
|
)
|
|
|
|
// Test is a public function that can be used in other tests to
|
|
// test that a helper is functioning properly.
|
|
func Test(t *testing.T, h TokenHelper) {
|
|
if err := h.Store("foo"); err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
|
|
v, err := h.Get()
|
|
if err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
|
|
if v != "foo" {
|
|
t.Fatalf("bad: %#v", v)
|
|
}
|
|
|
|
if err := h.Erase(); err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
|
|
v, err = h.Get()
|
|
if err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
|
|
if v != "" {
|
|
t.Fatalf("bad: %#v", v)
|
|
}
|
|
}
|
|
|
|
// TestProcess is used to re-execute this test in order to use it as the
|
|
// helper process. For this to work, the TestExternalTokenHelperProcess function must
|
|
// exist.
|
|
func TestProcess(t *testing.T, s ...string) {
|
|
h := &ExternalTokenHelper{BinaryPath: TestProcessPath(t, s...)}
|
|
Test(t, h)
|
|
}
|
|
|
|
// TestProcessPath returns the path to the test process.
|
|
func TestProcessPath(t *testing.T, s ...string) string {
|
|
cs := []string{"-test.run=TestExternalTokenHelperProcess", "--", "GO_WANT_HELPER_PROCESS"}
|
|
cs = append(cs, s...)
|
|
return fmt.Sprintf(
|
|
"%s %s",
|
|
os.Args[0],
|
|
strings.Join(cs, " "))
|
|
}
|
|
|
|
// TestExternalTokenHelperProcessCLI can be called to implement TestExternalTokenHelperProcess
|
|
// for TestProcess that just executes a CLI command.
|
|
func TestExternalTokenHelperProcessCLI(t *testing.T, cmd cli.Command) {
|
|
args := os.Args
|
|
for len(args) > 0 {
|
|
if args[0] == "--" {
|
|
args = args[1:]
|
|
break
|
|
}
|
|
|
|
args = args[1:]
|
|
}
|
|
if len(args) == 0 || args[0] != "GO_WANT_HELPER_PROCESS" {
|
|
return
|
|
}
|
|
args = args[1:]
|
|
|
|
os.Exit(cmd.Run(args))
|
|
}
|