113 lines
2.9 KiB
Go
113 lines
2.9 KiB
Go
package cmd
|
|
|
|
import (
|
|
"strconv"
|
|
"testing"
|
|
|
|
"github.com/mirivlad/sshkeeper/internal/db"
|
|
"github.com/mirivlad/sshkeeper/internal/model"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
func TestForwardEditUpdatesEnabledFlag(t *testing.T) {
|
|
testDB, err := db.Open(t.TempDir())
|
|
if err != nil {
|
|
t.Fatalf("open db: %v", err)
|
|
}
|
|
defer testDB.Close()
|
|
|
|
previousDB := appDB
|
|
appDB = testDB
|
|
t.Cleanup(func() { appDB = previousDB })
|
|
|
|
server := &model.Server{Alias: "web", Host: "web.example.org", Port: 22, User: "root", AuthMethod: model.AuthKey}
|
|
if err := appDB.CreateServer(server); err != nil {
|
|
t.Fatalf("create server: %v", err)
|
|
}
|
|
forwardID, err := appDB.AddForward(&model.Forward{
|
|
ServerID: server.ID,
|
|
Name: "SOCKS",
|
|
Type: model.ForwardDynamic,
|
|
LocalAddr: "127.0.0.1",
|
|
LocalPort: 1080,
|
|
Enabled: true,
|
|
})
|
|
if err != nil {
|
|
t.Fatalf("add forward: %v", err)
|
|
}
|
|
|
|
cmd := &cobra.Command{}
|
|
cmd.Flags().Bool("enabled", true, "Enable/disable forward")
|
|
if err := cmd.Flags().Set("enabled", "false"); err != nil {
|
|
t.Fatalf("set enabled flag: %v", err)
|
|
}
|
|
|
|
if err := forwardEditCmd.RunE(cmd, []string{strconv.FormatInt(forwardID, 10)}); err != nil {
|
|
t.Fatalf("edit forward: %v", err)
|
|
}
|
|
|
|
forwards, err := appDB.GetForwards(server.ID)
|
|
if err != nil {
|
|
t.Fatalf("get forwards: %v", err)
|
|
}
|
|
if len(forwards) != 1 {
|
|
t.Fatalf("expected one forward, got %d", len(forwards))
|
|
}
|
|
if forwards[0].Enabled {
|
|
t.Fatal("expected forward to be disabled")
|
|
}
|
|
}
|
|
|
|
func TestForwardAddStoresNameAndDescription(t *testing.T) {
|
|
testDB, err := db.Open(t.TempDir())
|
|
if err != nil {
|
|
t.Fatalf("open db: %v", err)
|
|
}
|
|
defer testDB.Close()
|
|
|
|
previousDB := appDB
|
|
appDB = testDB
|
|
t.Cleanup(func() { appDB = previousDB })
|
|
|
|
server := &model.Server{Alias: "web", Host: "web.example.org", Port: 22, User: "root", AuthMethod: model.AuthKey}
|
|
if err := appDB.CreateServer(server); err != nil {
|
|
t.Fatalf("create server: %v", err)
|
|
}
|
|
|
|
cmd := &cobra.Command{}
|
|
cmd.Flags().String("name", "", "")
|
|
cmd.Flags().String("description", "", "")
|
|
cmd.Flags().String("type", "local", "")
|
|
cmd.Flags().String("local-addr", "127.0.0.1", "")
|
|
cmd.Flags().Int("local-port", 0, "")
|
|
cmd.Flags().String("remote-addr", "", "")
|
|
cmd.Flags().Int("remote-port", 0, "")
|
|
for flag, value := range map[string]string{
|
|
"name": "Local PostgreSQL",
|
|
"description": "DB access",
|
|
"type": "local",
|
|
"local-port": "15432",
|
|
"remote-addr": "127.0.0.1",
|
|
"remote-port": "5432",
|
|
} {
|
|
if err := cmd.Flags().Set(flag, value); err != nil {
|
|
t.Fatalf("set %s: %v", flag, err)
|
|
}
|
|
}
|
|
|
|
if err := forwardAddCmd.RunE(cmd, []string{"web"}); err != nil {
|
|
t.Fatalf("add forward: %v", err)
|
|
}
|
|
|
|
forwards, err := appDB.GetForwards(server.ID)
|
|
if err != nil {
|
|
t.Fatalf("get forwards: %v", err)
|
|
}
|
|
if len(forwards) != 1 {
|
|
t.Fatalf("expected one forward, got %d", len(forwards))
|
|
}
|
|
if forwards[0].Name != "Local PostgreSQL" || forwards[0].Description != "DB access" {
|
|
t.Fatalf("unexpected forward metadata: %#v", forwards[0])
|
|
}
|
|
}
|