Add MCP prompt methods and placeholder support to tasks

taskcp_test.go

@@ -7,19 +7,64 @@ import (
 	"github.com/stretchr/testify/require"
 )
 
-func TestTaskCP(t *testing.T) {
-	tcp := taskcp.New()
 
-	p := tcp.AddProject()
-	require.NotNil(t, p)
-
-	tk := p.InsertTaskBefore(p.NextTaskID, "Hello, world!", func(task *taskcp.Task) {
-		t.Logf("Task %s changed: %+v", task.ID, task)
-	})
-	require.NotNil(t, tk)
-
-	p.SetTaskSuccess(tk.ID, "Hello, world!", "Notes")
-	require.Equal(t, taskcp.TaskStateSuccess, tk.State)
-	require.Equal(t, "Hello, world!", tk.Result)
-	require.Equal(t, "Notes", tk.Notes)
+func TestTaskPrompts(t *testing.T) {
+	service := taskcp.New("my_service")
+	project := service.AddProject()
+	
+	task := project.InsertTaskBefore("", "Write unit tests", func(task *taskcp.Task) {})
+	
+	successPrompt := task.SuccessPrompt()
+	require.Contains(t, successPrompt, "my_service.set_task_success")
+	require.Contains(t, successPrompt, `project_id="`+project.ID+`"`)
+	require.Contains(t, successPrompt, `task_id="`+task.ID+`"`)
+	
+	failurePrompt := task.FailurePrompt()
+	require.Contains(t, failurePrompt, "my_service.set_task_failure")
+	require.Contains(t, failurePrompt, `project_id="`+project.ID+`"`)
+	require.Contains(t, failurePrompt, `task_id="`+task.ID+`"`)
 }
+
+func TestPlaceholderExpansion(t *testing.T) {
+	service := taskcp.New("my_service")
+	project := service.AddProject()
+	
+	task1 := project.InsertTaskBefore("", "Please complete this task. {SUCCESS_PROMPT}", func(task *taskcp.Task) {})
+	require.Contains(t, task1.Instructions, "my_service.set_task_success")
+	require.NotContains(t, task1.Instructions, "{SUCCESS_PROMPT}")
+	
+	task2 := project.InsertTaskBefore("", "Try this risky operation. {FAILURE_PROMPT}", func(task *taskcp.Task) {})
+	require.Contains(t, task2.Instructions, "my_service.set_task_failure")
+	require.NotContains(t, task2.Instructions, "{FAILURE_PROMPT}")
+}
+
+func TestTaskFlow(t *testing.T) {
+	service := taskcp.New("test_service")
+	project := service.AddProject()
+	
+	var completed []string
+	
+	task1 := project.InsertTaskBefore("", "First task", func(task *taskcp.Task) {
+		completed = append(completed, task.ID)
+	})
+	
+	task2 := project.InsertTaskBefore("", "Second task", func(task *taskcp.Task) {
+		completed = append(completed, task.ID)
+	})
+	
+	current := project.GetNextTask()
+	require.NotNil(t, current)
+	require.Equal(t, task1.ID, current.ID)
+	
+	next := project.SetTaskSuccess(current.ID, "Task 1 done", "")
+	require.NotNil(t, next)
+	require.Equal(t, task2.ID, next.ID)
+	require.Equal(t, taskcp.TaskStateRunning, next.State)
+	
+	next2 := project.SetTaskFailure(next.ID, "Task 2 failed", "Error details")
+	require.Nil(t, next2)
+	
+	require.Equal(t, []string{task1.ID, task2.ID}, completed)
+	require.Equal(t, taskcp.TaskStateSuccess, project.Tasks[task1.ID].State)
+	require.Equal(t, taskcp.TaskStateFailure, project.Tasks[task2.ID].State)
+}