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) } }