mirror of https://github.com/hashicorp/consul
66 lines
1.1 KiB
Go
66 lines
1.1 KiB
Go
package command
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/hashicorp/consul/command/base"
|
|
"github.com/mitchellh/cli"
|
|
)
|
|
|
|
func TestKVImportCommand_Run(t *testing.T) {
|
|
srv, client := testAgentWithAPIClient(t)
|
|
defer srv.Shutdown()
|
|
waitForLeader(t, srv.httpAddr)
|
|
|
|
const json = `[
|
|
{
|
|
"key": "foo",
|
|
"flags": 0,
|
|
"value": "YmFyCg=="
|
|
},
|
|
{
|
|
"key": "foo/a",
|
|
"flags": 0,
|
|
"value": "YmF6Cg=="
|
|
}
|
|
]`
|
|
|
|
ui := new(cli.MockUi)
|
|
c := &KVImportCommand{
|
|
Command: base.Command{
|
|
Ui: ui,
|
|
Flags: base.FlagSetHTTP,
|
|
},
|
|
testStdin: strings.NewReader(json),
|
|
}
|
|
|
|
args := []string{
|
|
"-http-addr=" + srv.httpAddr,
|
|
"-",
|
|
}
|
|
|
|
code := c.Run(args)
|
|
if code != 0 {
|
|
t.Fatalf("bad: %d. %#v", code, ui.ErrorWriter.String())
|
|
}
|
|
|
|
pair, _, err := client.KV().Get("foo", nil)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if strings.TrimSpace(string(pair.Value)) != "bar" {
|
|
t.Fatalf("bad: expected: bar, got %s", pair.Value)
|
|
}
|
|
|
|
pair, _, err = client.KV().Get("foo/a", nil)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if strings.TrimSpace(string(pair.Value)) != "baz" {
|
|
t.Fatalf("bad: expected: baz, got %s", pair.Value)
|
|
}
|
|
}
|