mirror of
https://github.com/YspCoder/clawgo.git
synced 2026-04-13 21:57:29 +08:00
143 lines
3.8 KiB
Go
143 lines
3.8 KiB
Go
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)
|
||
}
|
||
}
|