Add test for SSH env vars

mafredri · mafredri · commit 24f8e6a3b6a8 · 2022-08-23T14:16:31.000+03:00
diff --git a/agent/agent_test.go b/agent/agent_test.go
@@ -232,6 +232,30 @@ func TestAgent(t *testing.T) {
 		require.Equal(t, expect, strings.TrimSpace(string(output)))
 	})
 
+	t.Run("SSH connection env vars", func(t *testing.T) {
+		t.Parallel()
+
+		// Note: the SSH_TTY environment variable should only be set for TTYs.
+		for _, key := range []string{"SSH_CONNECTION", "SSH_CLIENT", "SSH_TTY"} {
+			key := key
+			t.Run(key, func(t *testing.T) {
+				t.Parallel()
+
+				command := "sh -c 'echo $" + key + "'"
+				if runtime.GOOS == "windows" {
+					if key == "SSH_TTY" {
+						t.Skip("The SSH_PTY environment variable is not set on Windows")
+					}
+					command = "cmd.exe /c echo %" + key + "%"
+				}
+				session := setupSSHSession(t, agent.Metadata{})
+				output, err := session.Output(command)
+				require.NoError(t, err)
+				require.NotEmpty(t, strings.TrimSpace(string(output)))
+			})
+		}
+	})
+
 	t.Run("StartupScript", func(t *testing.T) {
 		t.Parallel()
 		tempPath := filepath.Join(t.TempDir(), "content.txt")
