feat: add generation source handling for task creation and updates
All checks were successful
Build And Publish Production Image / Build And Publish Production Image (push) Successful in 50s
All checks were successful
Build And Publish Production Image / Build And Publish Production Image (push) Successful in 50s
This commit is contained in:
@@ -57,6 +57,7 @@ const mockTask = {
|
||||
prompt: 'Summarize jokes',
|
||||
scheduleCron: '0 9 * * 1',
|
||||
emailLookback: 'last_week' as const,
|
||||
generationSource: 'openai' as const,
|
||||
active: true,
|
||||
createdAt: '2026-03-26T10:00:00Z',
|
||||
}
|
||||
@@ -77,7 +78,7 @@ describe('EditTaskPage', () => {
|
||||
vi.mocked(tasksApi.getTask).mockResolvedValue(mockTask)
|
||||
vi.mocked(tasksApi.buildTaskPreviewPrompt).mockImplementation(
|
||||
(entity, task) =>
|
||||
`PROMPT FOR ${entity.name}: ${task.name} | ${task.prompt} | ${task.scheduleCron} | ${task.emailLookback}`
|
||||
`PROMPT FOR ${entity.name}: ${task.name} | ${task.prompt} | ${task.scheduleCron} | ${task.emailLookback} | ${task.generationSource}`
|
||||
)
|
||||
vi.mocked(tasksApi.activateTask).mockResolvedValue({ ...mockTask, active: true })
|
||||
vi.mocked(tasksApi.inactivateTask).mockResolvedValue({ ...mockTask, active: false })
|
||||
@@ -97,6 +98,7 @@ describe('EditTaskPage', () => {
|
||||
expect(screen.getByRole('heading', { name: /edit task/i })).toBeInTheDocument()
|
||||
expect(screen.getByLabelText(/task name/i)).toHaveValue('Weekly Check-in')
|
||||
expect(screen.getByLabelText(/task prompt/i)).toHaveValue('Summarize jokes')
|
||||
expect(screen.getByLabelText(/^Generation Source$/i)).toHaveValue('openai')
|
||||
expect(screen.getByLabelText(/^Email Period$/i)).toHaveValue('last_week')
|
||||
expect(screen.getByLabelText(/^Minute$/i)).toHaveValue('0')
|
||||
expect(screen.getByLabelText(/^Hour$/i)).toHaveValue('9')
|
||||
@@ -124,6 +126,7 @@ describe('EditTaskPage', () => {
|
||||
prompt: 'Ask about ceremonial coffee',
|
||||
scheduleCron: '0 8 * * 1-5',
|
||||
emailLookback: 'last_day',
|
||||
generationSource: 'llama',
|
||||
})
|
||||
|
||||
const { queryClient } = renderPage()
|
||||
@@ -137,13 +140,16 @@ describe('EditTaskPage', () => {
|
||||
fireEvent.change(screen.getByLabelText(/^Email Period$/i), {
|
||||
target: { value: 'last_day' },
|
||||
})
|
||||
fireEvent.change(screen.getByLabelText(/^Generation Source$/i), {
|
||||
target: { value: 'llama' },
|
||||
})
|
||||
fireEvent.click(screen.getByRole('button', { name: /Weekdays/i }))
|
||||
fireEvent.change(screen.getByLabelText(/^Hour$/i), { target: { value: '8' } })
|
||||
|
||||
expect(screen.getByText(/Final Prompt/i)).toBeInTheDocument()
|
||||
expect(
|
||||
screen.getByText(
|
||||
'PROMPT FOR Entity A: Daily Check-in | Ask about ceremonial coffee | 0 8 * * 1-5 | last_day'
|
||||
'PROMPT FOR Entity A: Daily Check-in | Ask about ceremonial coffee | 0 8 * * 1-5 | last_day | llama'
|
||||
)
|
||||
).toBeInTheDocument()
|
||||
|
||||
@@ -159,6 +165,7 @@ describe('EditTaskPage', () => {
|
||||
prompt: 'Ask about ceremonial coffee',
|
||||
scheduleCron: '0 8 * * 1-5',
|
||||
emailLookback: 'last_day',
|
||||
generationSource: 'llama',
|
||||
})
|
||||
)
|
||||
expect(vi.mocked(tasksApi.generateTaskPreview).mock.calls[0][0]).toEqual('task-1')
|
||||
@@ -171,6 +178,7 @@ describe('EditTaskPage', () => {
|
||||
prompt: 'Ask about ceremonial coffee',
|
||||
scheduleCron: '0 8 * * 1-5',
|
||||
emailLookback: 'last_day',
|
||||
generationSource: 'llama',
|
||||
}),
|
||||
})
|
||||
)
|
||||
@@ -186,6 +194,7 @@ describe('EditTaskPage', () => {
|
||||
prompt: 'Ask about ceremonial coffee',
|
||||
scheduleCron: '0 8 * * 1-5',
|
||||
emailLookback: 'last_day',
|
||||
generationSource: 'llama',
|
||||
})
|
||||
expect(invalidateQueriesSpy).toHaveBeenCalledWith({ queryKey: ['entity-tasks', 'entity-1'] })
|
||||
expect(invalidateQueriesSpy).toHaveBeenCalledWith({ queryKey: ['entity-task', 'task-1'] })
|
||||
|
||||
Reference in New Issue
Block a user