db_lib/LocalApp_test.go - develop - wax_boy/semaphore_custom

db_lib/LocalApp_test.go
124 строки3 KB
copilot-swe-agent[bot]
test: improve test isolation in TestGetHomeDir
16 фев 2026, 18:14
16 фев 2026, 18:141e30aca
package db_lib

import (
	"os"
	"strings"
	"testing"

	"github.com/semaphoreui/semaphore/db"
	"github.com/semaphoreui/semaphore/util"
)

// contains checks if a slice contains a specific string
func contains(slice []string, item string) bool {
	for _, s := range slice {
		if strings.HasPrefix(s, item) {
			return true
		}
	}
	return false
}

func TestGetEnvironmentVars(t *testing.T) {
	os.Setenv("SEMAPHORE_TEST", "test123")  //nolint:errcheck
	os.Setenv("SEMAPHORE_TEST2", "test222") //nolint:errcheck
	os.Setenv("PASSWORD", "test222")        //nolint:errcheck

	util.Config = &util.ConfigType{
		ForwardedEnvVars: []string{"SEMAPHORE_TEST"},
		EnvVars: map[string]string{
			"ANSIBLE_FORCE_COLOR": "False",
		},
	}

	res := getEnvironmentVars()

	expected := []string{
		"SEMAPHORE_TEST=test123",
		"ANSIBLE_FORCE_COLOR=False",
		"PATH=",
	}

	if len(res) != len(expected) {
		t.Errorf("Expected %v, got %v", expected, res)
	}

	for _, e := range expected {
		if !contains(res, e) {
			t.Errorf("Expected %v, got %v", expected, res)
		}
	}
}

func TestGetHomeDir(t *testing.T) {
	repo := db.Repository{
		ProjectID: 42,
	}
	templateID := 114

	// Set a known HOME value for testing
	originalHome := os.Getenv("HOME")
	testHome := "/home/testuser"
	os.Setenv("HOME", testHome) //nolint:errcheck
	defer os.Setenv("HOME", originalHome) //nolint:errcheck

	// Save original config and restore after all tests
	originalConfig := util.Config
	defer func() { util.Config = originalConfig }()

	tests := []struct {
		name         string
		homeDirMode  string
		tmpPath      string
		expectedHome string
		description  string
	}{
		{
			name:         "ProjectHome mode",
			homeDirMode:  util.HomeDirModeProjectHome,
			tmpPath:      "/tmp/semaphore",
			expectedHome: "/tmp/semaphore/project_42",
			description:  "Should return project temp directory",
		},
		{
			name:         "TemplateDir mode",
			homeDirMode:  util.HomeDirModeTemplateDir,
			tmpPath:      "/tmp/semaphore",
			expectedHome: testHome,
			description:  "Should return real user HOME",
		},
		{
			name:         "UserHome mode",
			homeDirMode:  util.HomeDirModeUserHome,
			tmpPath:      "/tmp/semaphore",
			expectedHome: testHome,
			description:  "Should return real user HOME",
		},
		{
			name:         "Empty/default mode",
			homeDirMode:  "",
			tmpPath:      "/tmp/semaphore",
			expectedHome: "",
			description:  "Should return empty string for unknown mode",
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			// Setup config for this test case
			util.Config = &util.ConfigType{
				HomeDirMode: tt.homeDirMode,
				TmpPath:     tt.tmpPath,
			}

			// Call getHomeDir
			result := getHomeDir(repo, templateID)

			// Verify the result
			if result != tt.expectedHome {
				t.Errorf("%s: expected HOME=%s, got HOME=%s",
					tt.description, tt.expectedHome, result)
			}
		})
	}
}