mirror of
https://github.com/alibaba/open-code-review.git
synced 2026-07-09 17:28:58 +00:00
Add integration-style tests with fake LLM clients for agent dispatch and llmloop runner, plus new unit test files for gitcmd, session/history, tool/code_comment, tool/filereader_read, and viewer/store packages.
151 lines
3.2 KiB
Go
151 lines
3.2 KiB
Go
package gitcmd
|
|
|
|
import (
|
|
"context"
|
|
"io"
|
|
"os"
|
|
"os/exec"
|
|
"path/filepath"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func initRepo(t *testing.T) string {
|
|
t.Helper()
|
|
dir := t.TempDir()
|
|
run := func(args ...string) {
|
|
cmd := exec.Command("git", args...)
|
|
cmd.Dir = dir
|
|
cmd.Env = append(os.Environ(),
|
|
"GIT_AUTHOR_NAME=test",
|
|
"GIT_AUTHOR_EMAIL=test@test.com",
|
|
"GIT_COMMITTER_NAME=test",
|
|
"GIT_COMMITTER_EMAIL=test@test.com",
|
|
)
|
|
out, err := cmd.CombinedOutput()
|
|
if err != nil {
|
|
t.Fatalf("git %v: %v\n%s", args, err, out)
|
|
}
|
|
}
|
|
run("init")
|
|
run("config", "user.email", "test@test.com")
|
|
run("config", "user.name", "test")
|
|
if err := os.WriteFile(filepath.Join(dir, "hello.txt"), []byte("hello\n"), 0644); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
run("add", "hello.txt")
|
|
run("commit", "-m", "init")
|
|
return dir
|
|
}
|
|
|
|
func TestRunner_New(t *testing.T) {
|
|
r := New(0)
|
|
if r == nil {
|
|
t.Fatal("New(0) returned nil")
|
|
}
|
|
if cap(r.sem) != defaultMaxConcurrent {
|
|
t.Errorf("default capacity = %d, want %d", cap(r.sem), defaultMaxConcurrent)
|
|
}
|
|
|
|
r2 := New(4)
|
|
if cap(r2.sem) != 4 {
|
|
t.Errorf("capacity = %d, want 4", cap(r2.sem))
|
|
}
|
|
}
|
|
|
|
func TestRunner_Run(t *testing.T) {
|
|
dir := initRepo(t)
|
|
r := New(2)
|
|
|
|
out, err := r.Run(context.Background(), dir, "log", "--oneline")
|
|
if err != nil {
|
|
t.Fatalf("Run error: %v", err)
|
|
}
|
|
if !strings.Contains(out, "init") {
|
|
t.Errorf("expected 'init' in output: %q", out)
|
|
}
|
|
}
|
|
|
|
func TestRunner_Run_InvalidCommand(t *testing.T) {
|
|
dir := initRepo(t)
|
|
r := New(2)
|
|
|
|
_, err := r.Run(context.Background(), dir, "nonexistent-subcommand")
|
|
if err == nil {
|
|
t.Error("expected error for invalid git subcommand")
|
|
}
|
|
}
|
|
|
|
func TestRunner_Output(t *testing.T) {
|
|
dir := initRepo(t)
|
|
r := New(2)
|
|
|
|
out, err := r.Output(context.Background(), dir, "rev-parse", "HEAD")
|
|
if err != nil {
|
|
t.Fatalf("Output error: %v", err)
|
|
}
|
|
hash := strings.TrimSpace(string(out))
|
|
if len(hash) != 40 {
|
|
t.Errorf("expected 40-char hash, got %q", hash)
|
|
}
|
|
}
|
|
|
|
func TestRunner_RunSplit(t *testing.T) {
|
|
dir := initRepo(t)
|
|
r := New(2)
|
|
|
|
stdout, stderr, err := r.RunSplit(context.Background(), dir, "status", "--short")
|
|
if err != nil {
|
|
t.Fatalf("RunSplit error: %v", err)
|
|
}
|
|
_ = stderr
|
|
if strings.Contains(stdout, "??") {
|
|
t.Errorf("unexpected untracked files in clean repo: %q", stdout)
|
|
}
|
|
}
|
|
|
|
func TestRunner_Stream(t *testing.T) {
|
|
dir := initRepo(t)
|
|
r := New(2)
|
|
|
|
var content string
|
|
err := r.Stream(context.Background(), dir, func(stdout io.Reader) error {
|
|
data, err := io.ReadAll(stdout)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
content = string(data)
|
|
return nil
|
|
}, "show", "HEAD:hello.txt")
|
|
if err != nil {
|
|
t.Fatalf("Stream error: %v", err)
|
|
}
|
|
if content != "hello\n" {
|
|
t.Errorf("Stream content = %q, want %q", content, "hello\n")
|
|
}
|
|
}
|
|
|
|
func TestRunner_ContextCancelled(t *testing.T) {
|
|
r := New(1)
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
cancel()
|
|
|
|
_, err := r.Run(ctx, ".", "status")
|
|
if err == nil {
|
|
t.Error("expected error for cancelled context")
|
|
}
|
|
}
|
|
|
|
func TestRunner_AcquireTimeout(t *testing.T) {
|
|
r := New(1)
|
|
r.sem <- struct{}{}
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), 50*time.Millisecond)
|
|
defer cancel()
|
|
|
|
_, err := r.Run(ctx, ".", "status")
|
|
if err == nil {
|
|
t.Error("expected timeout error when semaphore full")
|
|
}
|
|
}
|