mirror of
https://github.com/holos-run/holos.git
synced 2026-03-19 16:54:58 +00:00
0d7033d063
This patch adds a holos create secret command that behaves like kubectl create secret, but for the specific use case of provisioning holos clusters. ``` ❯ holos create secret k2-talos --cluster-name=k2 --from-file=secrets.yaml 4:48PM INF secret.go:104 created: k2-talos-49546d9fd7 version=0.45.0 secret=k2-talos-49546d9fd7 name=k2-talos namespace=secrets ``` Once the corresponding `holos get secret` subcommands are implemented the kv subcommand may be removed.
129 lines
3.1 KiB
Go
129 lines
3.1 KiB
Go
package cli
|
|
|
|
import (
|
|
"bytes"
|
|
"github.com/holos-run/holos/pkg/holos"
|
|
"github.com/holos-run/holos/pkg/logger"
|
|
"github.com/holos-run/holos/pkg/version"
|
|
"github.com/spf13/cobra"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func newCommand() (*cobra.Command, *bytes.Buffer) {
|
|
var b1, b2 bytes.Buffer
|
|
// discard stdout for now, it's a bunch of usage messages.
|
|
cmd := New(holos.New(holos.Stdout(&b1), holos.Stderr(&b2)))
|
|
return cmd, &b2
|
|
}
|
|
|
|
func TestNewRoot(t *testing.T) {
|
|
cmd, _ := newCommand()
|
|
if err := cmd.Execute(); err != nil {
|
|
t.Fatalf("could not execute: %v", err)
|
|
}
|
|
}
|
|
|
|
type argsTestCase struct {
|
|
format string
|
|
level string
|
|
drop string
|
|
}
|
|
|
|
func (tc argsTestCase) args() []string {
|
|
return []string{"--log-format", tc.format, "--log-level", tc.level}
|
|
}
|
|
|
|
func TestValidArgs(t *testing.T) {
|
|
t.Parallel()
|
|
formats := []string{"text", "json"}
|
|
levels := []string{"debug", "info", "warn", "error"}
|
|
drops := []string{"version"}
|
|
for _, format := range formats {
|
|
for _, level := range levels {
|
|
for _, drop := range drops {
|
|
tc := argsTestCase{format, level, drop}
|
|
t.Run(strings.Join(tc.args(), " "), func(t *testing.T) {
|
|
cmd, _ := newCommand()
|
|
cmd.SetArgs(tc.args())
|
|
if err := cmd.Execute(); err != nil {
|
|
t.Errorf("unexpected error: %v", err)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestLogOutput(t *testing.T) {
|
|
// Lifecycle message is always displayed
|
|
cmd, b := newCommand()
|
|
cmd.SetArgs([]string{"--log-level=debug"})
|
|
if err := cmd.Execute(); err != nil {
|
|
t.Fatalf("could not execute: %v", err)
|
|
}
|
|
have := strings.TrimSpace(b.String())
|
|
want := "finalized config from flags"
|
|
if !strings.Contains(have, want) {
|
|
t.Fatalf("have does not contain want\n\thave: %#v\n\twant: %#v", have, want)
|
|
}
|
|
}
|
|
|
|
func TestLogDrop(t *testing.T) {
|
|
// Log attributes can be filtered out by the user
|
|
cmd, b := newCommand()
|
|
cmd.SetArgs([]string{"--log-level=debug", "--log-drop=version,another"})
|
|
if err := cmd.Execute(); err != nil {
|
|
t.Fatalf("could not execute: %v", err)
|
|
}
|
|
stderr := b.String()
|
|
if strings.Contains(stderr, "version") {
|
|
t.Fatalf("want version dropped got: %v", stderr)
|
|
}
|
|
}
|
|
|
|
func TestInvalidArgs(t *testing.T) {
|
|
invalidArgs := [][]string{
|
|
{"--log-format=yaml"},
|
|
{"--log-level=something"},
|
|
}
|
|
for _, args := range invalidArgs {
|
|
var b bytes.Buffer
|
|
cmd := New(holos.New(holos.Stdout(&b)))
|
|
cmd.SetArgs(args)
|
|
err := cmd.Execute()
|
|
if err == nil {
|
|
t.Fatalf("expected error from args: %v", args)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestLoggerFromContext(t *testing.T) {
|
|
cmd, b := newCommand()
|
|
if err := cmd.Execute(); err != nil {
|
|
t.Fatalf("could not execute: %v", err)
|
|
}
|
|
log := logger.FromContext(cmd.Context())
|
|
want := "612c48a3-30c9-44e3-b8b6-394191d99935"
|
|
log.Info(want)
|
|
have := b.String()
|
|
if !strings.Contains(have, want) {
|
|
t.Fatalf("want: %v have: %v", want, have)
|
|
}
|
|
}
|
|
|
|
func TestVersion(t *testing.T) {
|
|
var b bytes.Buffer
|
|
cmd := New(holos.New(holos.Stdout(&b)))
|
|
cmd.SetOut(&b)
|
|
cmd.SetArgs([]string{"--version"})
|
|
if err := cmd.Execute(); err != nil {
|
|
t.Fatalf("could not execute: %v", err)
|
|
}
|
|
want := version.Version + "\n"
|
|
have := b.String()
|
|
if want != have {
|
|
t.Fatalf("want: %v have: %v", want, have)
|
|
}
|
|
}
|