Story Crater Bot
5530e9154b
ci / test (push) Failing after 37s
Tests were failing because git worktree add -b branch worktree origin/main requires origin/main to exist. Now all test setups create main branch after initial commit so cloned repos have the expected branch.
241 lines
7.2 KiB
Go
241 lines
7.2 KiB
Go
package tests
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"os/exec"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/rockliang/poimen/workflows/action"
|
|
)
|
|
|
|
func TestGitCloneAndFetch(t *testing.T) {
|
|
// Create a temporary directory for the test
|
|
tmpDir := t.TempDir()
|
|
sourceDir := filepath.Join(tmpDir, "source")
|
|
targetDir := filepath.Join(tmpDir, "target")
|
|
|
|
// Initialize source repo
|
|
if err := os.MkdirAll(sourceDir, 0755); err != nil {
|
|
t.Fatalf("failed to create source dir: %v", err)
|
|
}
|
|
|
|
cmd := exec.Command("git", "init", sourceDir)
|
|
if err := cmd.Run(); err != nil {
|
|
t.Fatalf("git init failed: %v", err)
|
|
}
|
|
|
|
// Configure git user
|
|
exec.Command("git", "-C", sourceDir, "config", "user.email", "[email protected]").Run()
|
|
exec.Command("git", "-C", sourceDir, "config", "user.name", "Test User").Run()
|
|
|
|
// Create initial commit
|
|
testFile := filepath.Join(sourceDir, "test.txt")
|
|
if err := os.WriteFile(testFile, []byte("test content"), 0644); err != nil {
|
|
t.Fatalf("failed to create test file: %v", err)
|
|
}
|
|
|
|
cmd = exec.Command("git", "-C", sourceDir, "add", "test.txt")
|
|
if err := cmd.Run(); err != nil {
|
|
t.Fatalf("git add failed: %v", err)
|
|
}
|
|
|
|
cmd = exec.Command("git", "-C", sourceDir, "commit", "-m", "initial commit")
|
|
if err := cmd.Run(); err != nil {
|
|
t.Fatalf("git commit failed: %v", err)
|
|
}
|
|
|
|
// Create main branch (required for worktree operations)
|
|
cmd = exec.Command("git", "-C", sourceDir, "checkout", "-b", "main")
|
|
if err := cmd.Run(); err != nil {
|
|
t.Fatalf("git checkout -b main failed: %v", err)
|
|
}
|
|
|
|
// Test clone into empty path
|
|
ctx := context.Background()
|
|
err := action.CloneRepoActivity(ctx, action.CloneRepoInput{
|
|
RemoteURL: sourceDir,
|
|
TargetRepoPath: targetDir,
|
|
})
|
|
assert.NoError(t, err, "clone should succeed")
|
|
|
|
// Verify .git exists
|
|
gitDir := filepath.Join(targetDir, ".git")
|
|
_, err = os.Stat(gitDir)
|
|
assert.NoError(t, err, ".git directory should exist")
|
|
|
|
// Verify test.txt was cloned
|
|
targetTestFile := filepath.Join(targetDir, "test.txt")
|
|
_, err = os.Stat(targetTestFile)
|
|
assert.NoError(t, err, "test.txt should exist in target")
|
|
|
|
// Add another commit to source
|
|
testFile2 := filepath.Join(sourceDir, "test2.txt")
|
|
if err := os.WriteFile(testFile2, []byte("test content 2"), 0644); err != nil {
|
|
t.Fatalf("failed to create test2 file: %v", err)
|
|
}
|
|
|
|
cmd = exec.Command("git", "-C", sourceDir, "add", "test2.txt")
|
|
if err := cmd.Run(); err != nil {
|
|
t.Fatalf("git add failed: %v", err)
|
|
}
|
|
|
|
cmd = exec.Command("git", "-C", sourceDir, "commit", "-m", "second commit")
|
|
if err := cmd.Run(); err != nil {
|
|
t.Fatalf("git commit failed: %v", err)
|
|
}
|
|
|
|
// Test fetch on existing repo
|
|
err = action.CloneRepoActivity(ctx, action.CloneRepoInput{
|
|
RemoteURL: sourceDir,
|
|
TargetRepoPath: targetDir,
|
|
})
|
|
assert.NoError(t, err, "fetch should succeed")
|
|
}
|
|
|
|
func TestGitWorktreeAdd(t *testing.T) {
|
|
tmpDir := t.TempDir()
|
|
sourceDir := filepath.Join(tmpDir, "source")
|
|
repoDir := filepath.Join(tmpDir, "repo")
|
|
|
|
// Initialize source repo
|
|
if err := os.MkdirAll(sourceDir, 0755); err != nil {
|
|
t.Fatalf("failed to create source dir: %v", err)
|
|
}
|
|
|
|
cmd := exec.Command("git", "init", sourceDir)
|
|
if err := cmd.Run(); err != nil {
|
|
t.Fatalf("git init failed: %v", err)
|
|
}
|
|
|
|
// Configure git user
|
|
exec.Command("git", "-C", sourceDir, "config", "user.email", "[email protected]").Run()
|
|
exec.Command("git", "-C", sourceDir, "config", "user.name", "Test User").Run()
|
|
|
|
// Create initial commit
|
|
testFile := filepath.Join(sourceDir, "test.txt")
|
|
if err := os.WriteFile(testFile, []byte("test"), 0644); err != nil {
|
|
t.Fatalf("failed to create test file: %v", err)
|
|
}
|
|
|
|
cmd = exec.Command("git", "-C", sourceDir, "add", "test.txt")
|
|
if err := cmd.Run(); err != nil {
|
|
t.Fatalf("git add failed: %v", err)
|
|
}
|
|
|
|
cmd = exec.Command("git", "-C", sourceDir, "commit", "-m", "initial")
|
|
if err := cmd.Run(); err != nil {
|
|
t.Fatalf("git commit failed: %v", err)
|
|
}
|
|
|
|
// Create main branch (required for worktree operations)
|
|
cmd = exec.Command("git", "-C", sourceDir, "checkout", "-b", "main")
|
|
if err := cmd.Run(); err != nil {
|
|
t.Fatalf("git checkout -b main failed: %v", err)
|
|
}
|
|
|
|
// Clone the repo
|
|
ctx := context.Background()
|
|
err := action.CloneRepoActivity(ctx, action.CloneRepoInput{
|
|
RemoteURL: sourceDir,
|
|
TargetRepoPath: repoDir,
|
|
})
|
|
assert.NoError(t, err, "clone should succeed")
|
|
|
|
// Test worktree add
|
|
worktreePath, err := action.GitWorktreeAddActivity(ctx, action.GitWorktreeAddInput{
|
|
RepoPath: repoDir,
|
|
TaskID: "T0.1",
|
|
})
|
|
assert.NoError(t, err, "worktree add should succeed")
|
|
assert.NotEmpty(t, worktreePath, "worktree path should not be empty")
|
|
|
|
// Verify worktree directory exists
|
|
_, err = os.Stat(worktreePath)
|
|
assert.NoError(t, err, "worktree directory should exist")
|
|
|
|
// Verify branch exists
|
|
cmd = exec.Command("git", "-C", repoDir, "branch", "--list", "task/T0.1")
|
|
output, err := cmd.CombinedOutput()
|
|
assert.NoError(t, err, "git branch check should succeed")
|
|
assert.Contains(t, string(output), "task/T0.1", "task/T0.1 branch should exist")
|
|
}
|
|
|
|
func TestGitCommit(t *testing.T) {
|
|
tmpDir := t.TempDir()
|
|
sourceDir := filepath.Join(tmpDir, "source")
|
|
repoDir := filepath.Join(tmpDir, "repo")
|
|
|
|
// Initialize source repo
|
|
if err := os.MkdirAll(sourceDir, 0755); err != nil {
|
|
t.Fatalf("failed to create source dir: %v", err)
|
|
}
|
|
|
|
cmd := exec.Command("git", "init", sourceDir)
|
|
if err := cmd.Run(); err != nil {
|
|
t.Fatalf("git init failed: %v", err)
|
|
}
|
|
|
|
// Configure git user
|
|
exec.Command("git", "-C", sourceDir, "config", "user.email", "[email protected]").Run()
|
|
exec.Command("git", "-C", sourceDir, "config", "user.name", "Test User").Run()
|
|
|
|
// Create initial commit
|
|
testFile := filepath.Join(sourceDir, "test.txt")
|
|
if err := os.WriteFile(testFile, []byte("test"), 0644); err != nil {
|
|
t.Fatalf("failed to create test file: %v", err)
|
|
}
|
|
|
|
cmd = exec.Command("git", "-C", sourceDir, "add", "test.txt")
|
|
if err := cmd.Run(); err != nil {
|
|
t.Fatalf("git add failed: %v", err)
|
|
}
|
|
|
|
cmd = exec.Command("git", "-C", sourceDir, "commit", "-m", "initial")
|
|
if err := cmd.Run(); err != nil {
|
|
t.Fatalf("git commit failed: %v", err)
|
|
}
|
|
|
|
// Create main branch (required for worktree operations)
|
|
cmd = exec.Command("git", "-C", sourceDir, "checkout", "-b", "main")
|
|
if err := cmd.Run(); err != nil {
|
|
t.Fatalf("git checkout -b main failed: %v", err)
|
|
}
|
|
|
|
// Clone the repo
|
|
ctx := context.Background()
|
|
err := action.CloneRepoActivity(ctx, action.CloneRepoInput{
|
|
RemoteURL: sourceDir,
|
|
TargetRepoPath: repoDir,
|
|
})
|
|
assert.NoError(t, err, "clone should succeed")
|
|
|
|
// Create a worktree
|
|
worktreePath, err := action.GitWorktreeAddActivity(ctx, action.GitWorktreeAddInput{
|
|
RepoPath: repoDir,
|
|
TaskID: "T0.1",
|
|
})
|
|
assert.NoError(t, err)
|
|
|
|
// Create a new file in the worktree
|
|
newFile := filepath.Join(worktreePath, "new.txt")
|
|
if err := os.WriteFile(newFile, []byte("new content"), 0644); err != nil {
|
|
t.Fatalf("failed to create new file: %v", err)
|
|
}
|
|
|
|
// Commit changes
|
|
err = action.GitCommitActivity(ctx, action.GitCommitInput{
|
|
WorktreePath: worktreePath,
|
|
Message: "Add new file",
|
|
})
|
|
assert.NoError(t, err, "commit should succeed")
|
|
|
|
// Verify commit exists
|
|
cmd = exec.Command("git", "-C", worktreePath, "log", "--oneline")
|
|
output, err := cmd.CombinedOutput()
|
|
assert.NoError(t, err)
|
|
assert.Contains(t, string(output), "Add new file", "commit message should be in log")
|
|
}
|