Newer
Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
package main
import (
"errors"
"os"
"testing"
)
func withTestEnv(env map[string]string, f func()) {
getenv = func(key string) string {
return env[key]
}
defer func() {
getenv = os.Getenv
}()
f()
}
func TestEnv_GetSSHAuthInfo(t *testing.T) {
withTestEnv(map[string]string{
"SSH_AUTH_INFO_0": "publickey ssh-ed25519 AAAAC3NzaC1lZDI1NTE5AAAAIKHGU3+iLYiepGfenTlieDiv4hY4r0PlZa+mnb7NoKkc",
}, func() {
_, err := envGetSSHAuthInfo()
if err != nil {
t.Fatalf("envGetSSHAuthInfo: %v", err)
}
})
}
func TestEnv_GetSSHAuthInfo_NonPublicKey(t *testing.T) {
withTestEnv(map[string]string{
"SSH_AUTH_INFO_0": "password abcdef",
}, func() {
_, err := envGetSSHAuthInfo()
if !errors.Is(err, errNonPublicKeyLogin) {
t.Fatalf("envGetSSHAuthInfo: unexpected error %v", err)
}
})
}
func TestEnv_GetUser(t *testing.T) {
withTestEnv(map[string]string{
"PAM_USER": "user1",
}, func() {
u, err := envGetUser()
if err != nil || u != "user1" {
t.Fatalf("envGetUser: u=%v err=%v", u, err)
}
})
}
func TestEnv_GetSessionID(t *testing.T) {
for _, td := range []struct {
env map[string]string
id string
err error
}{
{
env: map[string]string{
"SSH_CONNECTION": "127.0.0.1 10023",
"XDG_SESSION_ID": "abcdef",
},
id: "7b272ca2f26f5e5f132bd84e94a774b2",
},
{
env: map[string]string{
"SSH_CONNECTION": "127.0.0.1 10023",
},
id: "ed83f217c70180f5ae9886a7cf2518d4",
},
{
env: map[string]string{
"XDG_SESSION_ID": "abcdef",
},
err: errNoSSHConnection,
},
} {
withTestEnv(td.env, func() {
id, err := envGetSessionID()
if !errors.Is(err, td.err) {
t.Errorf("%v - unexpected error %v", td.env, err)
}
if id != td.id {
t.Errorf("%v - unexpected id %s", td.env, id)
}
})
}
}
func TestEnv_GetRemoteIP(t *testing.T) {
for _, env := range []map[string]string{
{
"SSH_CONNECTION": "127.0.0.1 10023",
},
{
"PAM_RHOST": "127.0.0.1",
},
} {
withTestEnv(env, func() {
ip := envGetRemoteIP()
if ip != "127.0.0.1" {
t.Fatalf("unexpected ip for %v - %s", env, ip)
}
})
}
}