121 lines
3.6 KiB
Go
121 lines
3.6 KiB
Go
package taskcp_test
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/gopatchy/taskcp"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestTaskPrompts(t *testing.T) {
|
|
service := taskcp.New("my_service")
|
|
project := service.AddProject()
|
|
|
|
task := project.InsertTaskBefore(-1).
|
|
WithTitle("Write unit tests").
|
|
WithInstructions("This is a test task.").
|
|
Then(func(project *taskcp.Project, task *taskcp.Task) error {
|
|
return nil
|
|
})
|
|
|
|
successPrompt := task.SuccessPrompt()
|
|
require.Contains(t, successPrompt, "my_service.set_task_success")
|
|
require.Contains(t, successPrompt, fmt.Sprintf(`project_id=%d`, project.ID))
|
|
require.Contains(t, successPrompt, fmt.Sprintf(`task_id=%d`, task.ID))
|
|
|
|
failurePrompt := task.FailurePrompt()
|
|
require.Contains(t, failurePrompt, "my_service.set_task_failure")
|
|
require.Contains(t, failurePrompt, fmt.Sprintf(`project_id=%d`, project.ID))
|
|
require.Contains(t, failurePrompt, fmt.Sprintf(`task_id=%d`, task.ID))
|
|
}
|
|
|
|
func TestPlaceholderExpansion(t *testing.T) {
|
|
service := taskcp.New("my_service")
|
|
project := service.AddProject()
|
|
|
|
task1 := project.InsertTaskBefore(-1).
|
|
WithTitle("Please complete this task.").
|
|
WithInstructions("{SUCCESS_PROMPT}").
|
|
Then(func(project *taskcp.Project, task *taskcp.Task) error {
|
|
return nil
|
|
})
|
|
require.Contains(t, task1.Instructions, "my_service.set_task_success")
|
|
require.NotContains(t, task1.Instructions, "{SUCCESS_PROMPT}")
|
|
|
|
task2 := project.InsertTaskBefore(-1).
|
|
WithTitle("Try this risky operation.").
|
|
WithInstructions("{FAILURE_PROMPT}").
|
|
Then(func(project *taskcp.Project, task *taskcp.Task) error {
|
|
return nil
|
|
})
|
|
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 []int
|
|
|
|
task1 := project.InsertTaskBefore(-1).
|
|
WithTitle("First task").
|
|
Then(func(project *taskcp.Project, task *taskcp.Task) error {
|
|
completed = append(completed, task.ID)
|
|
return nil
|
|
})
|
|
|
|
task2 := project.InsertTaskBefore(-1).
|
|
WithTitle("Second task").
|
|
Then(func(project *taskcp.Project, task *taskcp.Task) error {
|
|
completed = append(completed, task.ID)
|
|
return nil
|
|
})
|
|
|
|
current := project.GetNextTask()
|
|
require.NotNil(t, current)
|
|
require.Equal(t, task1.ID, current.ID)
|
|
|
|
next, err := project.SetTaskSuccess(current.ID, "Task 1 done", "")
|
|
require.NoError(t, err)
|
|
require.NotNil(t, next)
|
|
require.Equal(t, task2.ID, next.ID)
|
|
require.Equal(t, taskcp.TaskStateRunning, next.State)
|
|
|
|
next2, err := project.SetTaskFailure(next.ID, "Task 2 failed", "Error details")
|
|
require.NoError(t, err)
|
|
require.Nil(t, next2)
|
|
|
|
require.Equal(t, []int{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)
|
|
}
|
|
|
|
func TestCallbackError(t *testing.T) {
|
|
service := taskcp.New("test_service")
|
|
project := service.AddProject()
|
|
|
|
expectedErr := fmt.Errorf("callback error")
|
|
|
|
task := project.InsertTaskBefore(-1).
|
|
WithTitle("Task with error callback").
|
|
WithInstructions("This is a test task.").
|
|
WithData("key", "value").
|
|
Then(func(project *taskcp.Project, task *taskcp.Task) error {
|
|
return expectedErr
|
|
})
|
|
|
|
current := project.GetNextTask()
|
|
require.NotNil(t, current)
|
|
require.Equal(t, task.ID, current.ID)
|
|
|
|
_, err := project.SetTaskSuccess(current.ID, "Result", "")
|
|
require.Error(t, err)
|
|
require.Equal(t, expectedErr, err)
|
|
|
|
_, err = project.SetTaskFailure(current.ID, "Task failed", "")
|
|
require.Error(t, err)
|
|
require.Equal(t, expectedErr, err)
|
|
}
|