返回 DeepSeek-Reasonix
jump_resolve_test.go
根目录 / internal / remote / jump_resolve_test.go
1 package remote
2
3 import (
4 "os"
5 "path/filepath"
6 "testing"
7
8 "reasonix/internal/config"
9 )
10
11 func TestResolveJumpHostsUsesConfiguredAliasesAndSSHConfig(t *testing.T) {
12 home := t.TempDir()
13 t.Setenv("HOME", home)
14 t.Setenv("USERPROFILE", home)
15 sshDir := filepath.Join(home, ".ssh")
16 if err := os.MkdirAll(sshDir, 0o700); err != nil {
17 t.Fatal(err)
18 }
19 text := "Host bastion\n HostName 10.0.0.8\n User jump-user\n Port 2202\n IdentityFile ~/.ssh/jump_key\n"
20 if err := os.WriteFile(filepath.Join(sshDir, "config"), []byte(text), 0o600); err != nil {
21 t.Fatal(err)
22 }
23 sshCfg, err := LoadSSHConfig(filepath.Join(sshDir, "config"))
24 if err != nil {
25 t.Fatal(err)
26 }
27 cfg := config.Default()
28 cfg.Remote.Hosts = []config.RemoteHostEntry{{
29 Name: "second", Host: "10.0.0.9", User: "ops",
30 PasswordEnv: "SECOND_PASSWORD", ProxyJump: "ignored-nested-hop",
31 }}
32 hops, err := ResolveJumpHosts(cfg, []string{"bastion", "second"}, sshCfg)
33 if err != nil {
34 t.Fatal(err)
35 }
36 if got := hops[0]; got.HostName != "10.0.0.8" || got.User != "jump-user" || got.Port != 2202 || got.IdentityFile != filepath.Join(home, ".ssh", "jump_key") {
37 t.Fatalf("ssh_config jump was not fully resolved: %+v", got)
38 }
39 if got := hops[1]; got.PasswordEnv != "SECOND_PASSWORD" || len(got.ProxyJump) != 0 {
40 t.Fatalf("Reasonix jump credentials/nested-chain handling wrong: %+v", got)
41 }
42 }
43
43 lines GO