test(ssh): cover GetUser config parsing

pkg/ssh/config_getuser_test.go

@@ -0,0 +1,110 @@
+package ssh
+
+import (
+	"os"
+	"path/filepath"
+	"testing"
+
+	"github.com/devsy-org/devsy/pkg/config"
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+)
+
+func writeUserConfig(t *testing.T, path, workspaceID, user string) {
+	t.Helper()
+	require.NoError(t, os.MkdirAll(filepath.Dir(path), 0o700))
+
+	host := workspaceID + config.SSHHostSuffix
+	section := MarkerStartPrefix + host + "\nHost " + host + "\n"
+	if user != "" {
+		section += "  User " + user + "\n"
+	}
+	section += MarkerEndPrefix + host + "\n"
+
+	require.NoError(t, os.WriteFile(path, []byte(section), 0o600))
+}
+
+func TestGetUser_ReturnsRootWhenConfigMissing(t *testing.T) {
+	home := t.TempDir()
+	t.Setenv("HOME", home)
+
+	user, err := GetUser("myws", "", "")
+	require.NoError(t, err)
+	assert.Equal(t, "root", user)
+}
+
+func TestGetUser_ReturnsConfiguredUser(t *testing.T) {
+	home := t.TempDir()
+	t.Setenv("HOME", home)
+	configPath := filepath.Join(home, ".ssh", "config")
+	writeUserConfig(t, configPath, "myws", "vscode")
+
+	user, err := GetUser("myws", configPath, "")
+	require.NoError(t, err)
+	assert.Equal(t, "vscode", user)
+}
+
+func TestGetUser_ReturnsRootWhenNoUserDirective(t *testing.T) {
+	home := t.TempDir()
+	t.Setenv("HOME", home)
+	configPath := filepath.Join(home, ".ssh", "config")
+	writeUserConfig(t, configPath, "myws", "")
+
+	user, err := GetUser("myws", configPath, "")
+	require.NoError(t, err)
+	assert.Equal(t, "root", user)
+}
+
+func TestGetUser_ReturnsRootWhenHostSectionAbsent(t *testing.T) {
+	home := t.TempDir()
+	t.Setenv("HOME", home)
+	configPath := filepath.Join(home, ".ssh", "config")
+	// Config has a section for a different workspace only.
+	writeUserConfig(t, configPath, "otherws", "vscode")
+
+	user, err := GetUser("myws", configPath, "")
+	require.NoError(t, err)
+	assert.Equal(t, "root", user)
+}
+
+func TestGetUser_IncludePathTakesPrecedence(t *testing.T) {
+	home := t.TempDir()
+	t.Setenv("HOME", home)
+
+	configPath := filepath.Join(home, ".ssh", "config")
+	includePath := filepath.Join(home, ".ssh", "devsy_config")
+
+	writeUserConfig(t, configPath, "myws", "from-main")
+	writeUserConfig(t, includePath, "myws", "from-include")
+
+	user, err := GetUser("myws", configPath, includePath)
+	require.NoError(t, err)
+	assert.Equal(t, "from-include", user)
+}
+
+func TestGetUser_ResolvesEmptyPathToDefault(t *testing.T) {
+	home := t.TempDir()
+	t.Setenv("HOME", home)
+	configPath := filepath.Join(home, ".ssh", "config")
+	writeUserConfig(t, configPath, "myws", "vscode")
+
+	user, err := GetUser("myws", "", "")
+	require.NoError(t, err)
+	assert.Equal(t, "vscode", user)
+}
+
+func TestGetUser_UnquotesQuotedUser(t *testing.T) {
+	home := t.TempDir()
+	t.Setenv("HOME", home)
+	configPath := filepath.Join(home, ".ssh", "config")
+
+	host := "myws" + config.SSHHostSuffix
+	section := MarkerStartPrefix + host + "\nHost " + host + "\n" +
+		"  User \"custom-user\"\n" + MarkerEndPrefix + host + "\n"
+	require.NoError(t, os.MkdirAll(filepath.Dir(configPath), 0o700))
+	require.NoError(t, os.WriteFile(configPath, []byte(section), 0o600))
+
+	user, err := GetUser("myws", configPath, "")
+	require.NoError(t, err)
+	assert.Equal(t, "custom-user", user)
+}