110 lines
3.1 KiB
Go
110 lines
3.1 KiB
Go
|
package workspace
|
||
|
|
|
||
|
|
import (
|
||
|
|
"os"
|
||
|
|
"os/exec"
|
||
|
|
"path/filepath"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
)
|
||
|
|
|
||
|
|
func setupTestRepo(t *testing.T, dir, name string) string {
|
||
|
|
t.Helper()
|
||
|
|
repoPath := filepath.Join(dir, name)
|
||
|
|
require.NoError(t, os.MkdirAll(repoPath, 0755))
|
||
|
|
|
||
|
|
cmds := [][]string{
|
||
|
|
{"git", "init"},
|
||
|
|
{"git", "config", "user.email", "test@test.com"},
|
||
|
|
{"git", "config", "user.name", "Test"},
|
||
|
|
{"git", "commit", "--allow-empty", "-m", "initial"},
|
||
|
|
}
|
||
|
|
for _, c := range cmds {
|
||
|
|
cmd := exec.Command(c[0], c[1:]...)
|
||
|
|
cmd.Dir = repoPath
|
||
|
|
out, err := cmd.CombinedOutput()
|
||
|
|
require.NoError(t, err, "cmd %v failed: %s", c, string(out))
|
||
|
|
}
|
||
|
|
return repoPath
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestTaskWorkspacePath(t *testing.T) {
|
||
|
|
path := taskWorkspacePath("/home/user/Code/host-uk", 101, 343)
|
||
|
|
assert.Equal(t, "/home/user/Code/host-uk/.core/workspace/p101/i343", path)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestCreateWorktree_Good(t *testing.T) {
|
||
|
|
tmp := t.TempDir()
|
||
|
|
repoPath := setupTestRepo(t, tmp, "test-repo")
|
||
|
|
worktreePath := filepath.Join(tmp, "workspace", "test-repo")
|
||
|
|
|
||
|
|
err := createWorktree(t.Context(), repoPath, worktreePath, "issue/123")
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
// Verify worktree exists
|
||
|
|
assert.DirExists(t, worktreePath)
|
||
|
|
assert.FileExists(t, filepath.Join(worktreePath, ".git"))
|
||
|
|
|
||
|
|
// Verify branch
|
||
|
|
branch := gitOutput(worktreePath, "rev-parse", "--abbrev-ref", "HEAD")
|
||
|
|
assert.Equal(t, "issue/123", trimNL(branch))
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestCreateWorktree_BranchExists(t *testing.T) {
|
||
|
|
tmp := t.TempDir()
|
||
|
|
repoPath := setupTestRepo(t, tmp, "test-repo")
|
||
|
|
|
||
|
|
// Create branch first
|
||
|
|
cmd := exec.Command("git", "branch", "issue/456")
|
||
|
|
cmd.Dir = repoPath
|
||
|
|
require.NoError(t, cmd.Run())
|
||
|
|
|
||
|
|
worktreePath := filepath.Join(tmp, "workspace", "test-repo")
|
||
|
|
err := createWorktree(t.Context(), repoPath, worktreePath, "issue/456")
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
assert.DirExists(t, worktreePath)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestCheckWorkspaceSafety_Clean(t *testing.T) {
|
||
|
|
tmp := t.TempDir()
|
||
|
|
wsPath := filepath.Join(tmp, "workspace")
|
||
|
|
require.NoError(t, os.MkdirAll(wsPath, 0755))
|
||
|
|
|
||
|
|
repoPath := setupTestRepo(t, tmp, "origin-repo")
|
||
|
|
worktreePath := filepath.Join(wsPath, "origin-repo")
|
||
|
|
require.NoError(t, createWorktree(t.Context(), repoPath, worktreePath, "test-branch"))
|
||
|
|
|
||
|
|
dirty, reasons := checkWorkspaceSafety(wsPath)
|
||
|
|
assert.False(t, dirty)
|
||
|
|
assert.Empty(t, reasons)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestCheckWorkspaceSafety_Dirty(t *testing.T) {
|
||
|
|
tmp := t.TempDir()
|
||
|
|
wsPath := filepath.Join(tmp, "workspace")
|
||
|
|
require.NoError(t, os.MkdirAll(wsPath, 0755))
|
||
|
|
|
||
|
|
repoPath := setupTestRepo(t, tmp, "origin-repo")
|
||
|
|
worktreePath := filepath.Join(wsPath, "origin-repo")
|
||
|
|
require.NoError(t, createWorktree(t.Context(), repoPath, worktreePath, "test-branch"))
|
||
|
|
|
||
|
|
// Create uncommitted file
|
||
|
|
require.NoError(t, os.WriteFile(filepath.Join(worktreePath, "dirty.txt"), []byte("dirty"), 0644))
|
||
|
|
|
||
|
|
dirty, reasons := checkWorkspaceSafety(wsPath)
|
||
|
|
assert.True(t, dirty)
|
||
|
|
assert.Contains(t, reasons[0], "uncommitted changes")
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestEpicBranchName(t *testing.T) {
|
||
|
|
assert.Equal(t, "epic/101", epicBranchName(101))
|
||
|
|
assert.Equal(t, "epic/42", epicBranchName(42))
|
||
|
|
}
|
||
|
|
|
||
|
|
func trimNL(s string) string {
|
||
|
|
return s[:len(s)-1]
|
||
|
|
}
|