Add test for SSH env vars

mafredri · mafredri · commit 1f5afe35d56a · 2022-08-23T14:00:47.000+03:00
diff --git a/agent/agent_test.go b/agent/agent_test.go
@@ -210,6 +210,30 @@ func TestAgent(t *testing.T) {
 		require.Equal(t, value, 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("EnvironmentVariableExpansion", func(t *testing.T) {
 		t.Parallel()
 		key := "EXAMPLE"
