| // Copyright (c) HashiCorp, Inc. |
| // SPDX-License-Identifier: MPL-2.0 |
| |
| package command |
| |
| import ( |
| "strings" |
| "testing" |
| |
| "github.com/mitchellh/cli" |
| ) |
| |
| func testOperatorRotateCommand(tb testing.TB) (*cli.MockUi, *OperatorRotateCommand) { |
| tb.Helper() |
| |
| ui := cli.NewMockUi() |
| return ui, &OperatorRotateCommand{ |
| BaseCommand: &BaseCommand{ |
| UI: ui, |
| }, |
| } |
| } |
| |
| func TestOperatorRotateCommand_Run(t *testing.T) { |
| t.Parallel() |
| |
| cases := []struct { |
| name string |
| args []string |
| out string |
| code int |
| }{ |
| { |
| "too_many_args", |
| []string{"abcd1234"}, |
| "Too many arguments", |
| 1, |
| }, |
| } |
| |
| t.Run("validations", func(t *testing.T) { |
| t.Parallel() |
| |
| for _, tc := range cases { |
| tc := tc |
| |
| t.Run(tc.name, func(t *testing.T) { |
| t.Parallel() |
| |
| client, closer := testVaultServer(t) |
| defer closer() |
| |
| ui, cmd := testOperatorRotateCommand(t) |
| cmd.client = client |
| |
| code := cmd.Run(tc.args) |
| if code != tc.code { |
| t.Errorf("expected %d to be %d", code, tc.code) |
| } |
| |
| combined := ui.OutputWriter.String() + ui.ErrorWriter.String() |
| if !strings.Contains(combined, tc.out) { |
| t.Errorf("expected %q to contain %q", combined, tc.out) |
| } |
| }) |
| } |
| }) |
| |
| t.Run("default", func(t *testing.T) { |
| t.Parallel() |
| |
| client, closer := testVaultServer(t) |
| defer closer() |
| |
| ui, cmd := testOperatorRotateCommand(t) |
| cmd.client = client |
| |
| code := cmd.Run([]string{}) |
| if exp := 0; code != exp { |
| t.Errorf("expected %d to be %d", code, exp) |
| } |
| |
| expected := "Success! Rotated key" |
| combined := ui.OutputWriter.String() + ui.ErrorWriter.String() |
| if !strings.Contains(combined, expected) { |
| t.Errorf("expected %q to contain %q", combined, expected) |
| } |
| |
| status, err := client.Sys().KeyStatus() |
| if err != nil { |
| t.Fatal(err) |
| } |
| if exp := 1; status.Term < exp { |
| t.Errorf("expected %d to be less than %d", status.Term, exp) |
| } |
| }) |
| |
| t.Run("communication_failure", func(t *testing.T) { |
| t.Parallel() |
| |
| client, closer := testVaultServerBad(t) |
| defer closer() |
| |
| ui, cmd := testOperatorRotateCommand(t) |
| cmd.client = client |
| |
| code := cmd.Run([]string{}) |
| if exp := 2; code != exp { |
| t.Errorf("expected %d to be %d", code, exp) |
| } |
| |
| expected := "Error rotating key: " |
| combined := ui.OutputWriter.String() + ui.ErrorWriter.String() |
| if !strings.Contains(combined, expected) { |
| t.Errorf("expected %q to contain %q", combined, expected) |
| } |
| }) |
| |
| t.Run("no_tabs", func(t *testing.T) { |
| t.Parallel() |
| |
| _, cmd := testOperatorRotateCommand(t) |
| assertNoTabs(t, cmd) |
| }) |
| } |