Add automatic task

pkg/agent/loop_directive_test.go

@@ -0,0 +1,142 @@
+package agent
+
+import (
+	"testing"
+	"time"
+)
+
+func TestParseTaskExecutionDirectives_RunCommand(t *testing.T) {
+	d := parseTaskExecutionDirectives("/run 修复构建脚本 --stage-report")
+	if d.task != "修复构建脚本" {
+		t.Fatalf("unexpected task: %q", d.task)
+	}
+	if !d.stageReport {
+		t.Fatalf("expected stage report enabled")
+	}
+}
+
+func TestParseTaskExecutionDirectives_NaturalLanguage(t *testing.T) {
+	d := parseTaskExecutionDirectives("你可以自动运行任务：整理日志，但是每到一个阶段给我报告一下任务完成情况")
+	if d.task != "整理日志" {
+		t.Fatalf("unexpected task: %q", d.task)
+	}
+	if !d.stageReport {
+		t.Fatalf("expected stage report enabled")
+	}
+}
+
+func TestParseTaskExecutionDirectives_Default(t *testing.T) {
+	d := parseTaskExecutionDirectives("帮我看看今天的日志异常")
+	if d.task != "帮我看看今天的日志异常" {
+		t.Fatalf("unexpected task: %q", d.task)
+	}
+	if d.stageReport {
+		t.Fatalf("expected stage report disabled")
+	}
+}
+
+func TestParseAutoLearnInterval(t *testing.T) {
+	d, err := parseAutoLearnInterval("5m")
+	if err != nil {
+		t.Fatalf("unexpected error: %v", err)
+	}
+	if d != 5*time.Minute {
+		t.Fatalf("unexpected duration: %s", d)
+	}
+
+	d, err = parseAutoLearnInterval("2")
+	if err != nil {
+		t.Fatalf("unexpected error: %v", err)
+	}
+	if d != 2*time.Minute {
+		t.Fatalf("unexpected duration: %s", d)
+	}
+}
+
+func TestParseAutoLearnInterval_Invalid(t *testing.T) {
+	if _, err := parseAutoLearnInterval("oops"); err == nil {
+		t.Fatalf("expected error")
+	}
+}
+
+func TestParseAutoLearnIntent_StartNaturalLanguage(t *testing.T) {
+	intent, ok := parseAutoLearnIntent("请开始自动学习，每5分钟执行一轮")
+	if !ok {
+		t.Fatalf("expected intent")
+	}
+	if intent.action != "start" {
+		t.Fatalf("unexpected action: %s", intent.action)
+	}
+	if intent.interval == nil || *intent.interval != 5*time.Minute {
+		t.Fatalf("unexpected interval: %v", intent.interval)
+	}
+}
+
+func TestParseAutoLearnIntent_StopNaturalLanguage(t *testing.T) {
+	intent, ok := parseAutoLearnIntent("先暂停自动学习")
+	if !ok {
+		t.Fatalf("expected intent")
+	}
+	if intent.action != "stop" {
+		t.Fatalf("unexpected action: %s", intent.action)
+	}
+}
+
+func TestParseAutoLearnIntent_StatusNaturalLanguage(t *testing.T) {
+	intent, ok := parseAutoLearnIntent("帮我看下自动学习状态")
+	if !ok {
+		t.Fatalf("expected intent")
+	}
+	if intent.action != "status" {
+		t.Fatalf("unexpected action: %s", intent.action)
+	}
+}
+
+func TestParseAutonomyIntent_StartNaturalLanguage(t *testing.T) {
+	intent, ok := parseAutonomyIntent("以后你自动拆解并自动执行任务，每15分钟主动找我汇报一次")
+	if !ok {
+		t.Fatalf("expected intent")
+	}
+	if intent.action != "start" {
+		t.Fatalf("unexpected action: %s", intent.action)
+	}
+	if intent.idleInterval == nil || *intent.idleInterval != 15*time.Minute {
+		t.Fatalf("unexpected interval: %v", intent.idleInterval)
+	}
+}
+
+func TestParseAutonomyIntent_StopNaturalLanguage(t *testing.T) {
+	intent, ok := parseAutonomyIntent("先不要主动找我，关闭自主模式")
+	if !ok {
+		t.Fatalf("expected intent")
+	}
+	if intent.action != "stop" {
+		t.Fatalf("unexpected action: %s", intent.action)
+	}
+}
+
+func TestParseAutonomyIntent_StatusNaturalLanguage(t *testing.T) {
+	intent, ok := parseAutonomyIntent("帮我看下自主模式状态")
+	if !ok {
+		t.Fatalf("expected intent")
+	}
+	if intent.action != "status" {
+		t.Fatalf("unexpected action: %s", intent.action)
+	}
+}
+
+func TestParseAutonomyIdleInterval(t *testing.T) {
+	d, err := parseAutonomyIdleInterval("45m")
+	if err != nil {
+		t.Fatalf("unexpected error: %v", err)
+	}
+	if d != 45*time.Minute {
+		t.Fatalf("unexpected duration: %s", d)
+	}
+}
+
+func TestParseAutonomyIntent_NoFalsePositiveOnSingleTask(t *testing.T) {
+	if intent, ok := parseAutonomyIntent("请自动执行这个任务"); ok {
+		t.Fatalf("expected no intent, got: %+v", intent)
+	}
+}