91 lines
1.9 KiB
Go
91 lines
1.9 KiB
Go
package cmd
|
|
|
|
import (
|
|
"bytes"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/mirivlad/sshkeeper/internal/model"
|
|
)
|
|
|
|
func TestPromptServerForAddCollectsInteractiveFields(t *testing.T) {
|
|
input := strings.NewReader(strings.Join([]string{
|
|
"prod",
|
|
"Production",
|
|
"prod.example.org",
|
|
"2222",
|
|
"deploy",
|
|
"key",
|
|
"~/.ssh/id_prod",
|
|
"bastion",
|
|
"prod",
|
|
"critical host",
|
|
"tmux attach -t prod",
|
|
"prod,web",
|
|
"",
|
|
}, "\n"))
|
|
var output bytes.Buffer
|
|
|
|
server, err := promptServerForAdd(input, &output)
|
|
if err != nil {
|
|
t.Fatalf("prompt server: %v", err)
|
|
}
|
|
|
|
if server.Alias != "prod" ||
|
|
server.DisplayName != "Production" ||
|
|
server.Host != "prod.example.org" ||
|
|
server.Port != 2222 ||
|
|
server.User != "deploy" ||
|
|
server.AuthMethod != model.AuthKey ||
|
|
server.IdentityFile != "~/.ssh/id_prod" ||
|
|
server.ProxyJump != "bastion" ||
|
|
server.GroupName != "prod" ||
|
|
server.Notes != "critical host" ||
|
|
server.StartupCommand != "tmux attach -t prod" {
|
|
t.Fatalf("unexpected server: %#v", server)
|
|
}
|
|
if strings.Join(server.Tags, ",") != "prod,web" {
|
|
t.Fatalf("tags = %#v", server.Tags)
|
|
}
|
|
if strings.Contains(output.String(), "not yet implemented") {
|
|
t.Fatalf("interactive add should not report unimplemented:\n%s", output.String())
|
|
}
|
|
}
|
|
|
|
func TestPromptServerForAddAppliesDefaults(t *testing.T) {
|
|
input := strings.NewReader(strings.Join([]string{
|
|
"prod",
|
|
"",
|
|
"prod.example.org",
|
|
"",
|
|
"",
|
|
"",
|
|
"",
|
|
"",
|
|
"",
|
|
"",
|
|
"",
|
|
"",
|
|
"",
|
|
}, "\n"))
|
|
var output bytes.Buffer
|
|
|
|
server, err := promptServerForAdd(input, &output)
|
|
if err != nil {
|
|
t.Fatalf("prompt server: %v", err)
|
|
}
|
|
|
|
if server.DisplayName != "prod" {
|
|
t.Fatalf("display name default = %q; want alias", server.DisplayName)
|
|
}
|
|
if server.Port != 22 {
|
|
t.Fatalf("port default = %d; want 22", server.Port)
|
|
}
|
|
if server.User != "root" {
|
|
t.Fatalf("user default = %q; want root", server.User)
|
|
}
|
|
if server.AuthMethod != model.AuthKey {
|
|
t.Fatalf("auth default = %q; want key", server.AuthMethod)
|
|
}
|
|
}
|