Function main

cmd/test-conn/main-test-conn.go:48–104

()

}

func main() {
	timeoutFlag := flag.Duration("t", 60*time.Second, "connection timeout")
	interactiveFlag := flag.Bool("i", false, "interactive mode (prompt for user input)")
	versionFlag := flag.Bool("v", false, "show version")

	flag.Usage = usage
	flag.Parse()

	if *versionFlag {
		fmt.Printf("test-conn version %s (built %s)\n", WaveVersion, BuildTime)
		os.Exit(0)
	}

	args := flag.Args()
	if len(args) < 2 {
		usage()
	}

	command := args[0]
	connName := args[1]

	autoAccept := !*interactiveFlag

	err := initTestHarness(autoAccept)
	if err != nil {
		log.Fatalf("Failed to initialize: %v", err)
	}

	switch command {
	case "ssh", "connect":
		err = testBasicConnect(connName, *timeoutFlag)

	case "exec":
		if len(args) < 3 {
			log.Fatalf("exec command requires a command argument")
		}
		cmd := args[2]
		err = testShellWithCommand(connName, cmd, *timeoutFlag)

	case "wshexec":
		if len(args) < 3 {
			log.Fatalf("wshexec command requires a command argument")
		}
		cmd := args[2]
		err = testWshExec(connName, cmd, *timeoutFlag)

	case "shell":
		err = testInteractiveShell(connName, *timeoutFlag)

	default:
		log.Fatalf("Unknown command: %s", command)
	}

	if err != nil {
		log.Fatalf("Error: %v", err)
	}
}