Feat/add.azure.and.other.providers (#607)
* fix: claude-4 not having the right max_tokens * feat: add bedrock support * chore: fix package-lock.json * fix: rename baseUrl to baseURL * feat: add azure support * fix: final touches of azure integration * feat: add google vertex provider * chore: fix tests and refactor task-manager.test.js * chore: move task 92 to 94
This commit is contained in:
Ralph Khreish
464
tests/unit/scripts/modules/task-manager/set-task-status.test.js
Normal file
464
tests/unit/scripts/modules/task-manager/set-task-status.test.js
Normal file
@@ -0,0 +1,464 @@
|
||||
/**
|
||||
* Tests for the set-task-status.js module
|
||||
*/
|
||||
import { jest } from '@jest/globals';
|
||||
|
||||
// Mock the dependencies before importing the module under test
|
||||
jest.unstable_mockModule('../../../../../scripts/modules/utils.js', () => ({
|
||||
readJSON: jest.fn(),
|
||||
writeJSON: jest.fn(),
|
||||
log: jest.fn(),
|
||||
CONFIG: {
|
||||
model: 'mock-claude-model',
|
||||
maxTokens: 4000,
|
||||
temperature: 0.7,
|
||||
debug: false
|
||||
},
|
||||
sanitizePrompt: jest.fn((prompt) => prompt),
|
||||
truncate: jest.fn((text) => text),
|
||||
isSilentMode: jest.fn(() => false),
|
||||
findTaskById: jest.fn((tasks, id) => tasks.find((t) => t.id === parseInt(id)))
|
||||
}));
|
||||
|
||||
jest.unstable_mockModule(
|
||||
'../../../../../scripts/modules/task-manager/generate-task-files.js',
|
||||
() => ({
|
||||
default: jest.fn().mockResolvedValue()
|
||||
})
|
||||
);
|
||||
|
||||
jest.unstable_mockModule('../../../../../scripts/modules/ui.js', () => ({
|
||||
formatDependenciesWithStatus: jest.fn(),
|
||||
displayBanner: jest.fn(),
|
||||
displayTaskList: jest.fn(),
|
||||
startLoadingIndicator: jest.fn(() => ({ stop: jest.fn() })),
|
||||
stopLoadingIndicator: jest.fn(),
|
||||
getStatusWithColor: jest.fn((status) => status)
|
||||
}));
|
||||
|
||||
jest.unstable_mockModule('../../../../../src/constants/task-status.js', () => ({
|
||||
isValidTaskStatus: jest.fn((status) =>
|
||||
[
|
||||
'pending',
|
||||
'done',
|
||||
'in-progress',
|
||||
'review',
|
||||
'deferred',
|
||||
'cancelled'
|
||||
].includes(status)
|
||||
),
|
||||
TASK_STATUS_OPTIONS: [
|
||||
'pending',
|
||||
'done',
|
||||
'in-progress',
|
||||
'review',
|
||||
'deferred',
|
||||
'cancelled'
|
||||
]
|
||||
}));
|
||||
|
||||
jest.unstable_mockModule(
|
||||
'../../../../../scripts/modules/task-manager/update-single-task-status.js',
|
||||
() => ({
|
||||
default: jest.fn()
|
||||
})
|
||||
);
|
||||
|
||||
jest.unstable_mockModule(
|
||||
'../../../../../scripts/modules/dependency-manager.js',
|
||||
() => ({
|
||||
validateTaskDependencies: jest.fn()
|
||||
})
|
||||
);
|
||||
|
||||
jest.unstable_mockModule(
|
||||
'../../../../../scripts/modules/config-manager.js',
|
||||
() => ({
|
||||
getDebugFlag: jest.fn(() => false)
|
||||
})
|
||||
);
|
||||
|
||||
// Import the mocked modules
|
||||
const { readJSON, writeJSON, log, findTaskById } = await import(
|
||||
'../../../../../scripts/modules/utils.js'
|
||||
);
|
||||
|
||||
const generateTaskFiles = (
|
||||
await import(
|
||||
'../../../../../scripts/modules/task-manager/generate-task-files.js'
|
||||
)
|
||||
).default;
|
||||
|
||||
const updateSingleTaskStatus = (
|
||||
await import(
|
||||
'../../../../../scripts/modules/task-manager/update-single-task-status.js'
|
||||
)
|
||||
).default;
|
||||
|
||||
// Import the module under test
|
||||
const { default: setTaskStatus } = await import(
|
||||
'../../../../../scripts/modules/task-manager/set-task-status.js'
|
||||
);
|
||||
|
||||
// Sample data for tests (from main test file)
|
||||
const sampleTasks = {
|
||||
meta: { projectName: 'Test Project' },
|
||||
tasks: [
|
||||
{
|
||||
id: 1,
|
||||
title: 'Task 1',
|
||||
description: 'First task description',
|
||||
status: 'pending',
|
||||
dependencies: [],
|
||||
priority: 'high',
|
||||
details: 'Detailed information for task 1',
|
||||
testStrategy: 'Test strategy for task 1'
|
||||
},
|
||||
{
|
||||
id: 2,
|
||||
title: 'Task 2',
|
||||
description: 'Second task description',
|
||||
status: 'pending',
|
||||
dependencies: [1],
|
||||
priority: 'medium',
|
||||
details: 'Detailed information for task 2',
|
||||
testStrategy: 'Test strategy for task 2'
|
||||
},
|
||||
{
|
||||
id: 3,
|
||||
title: 'Task with Subtasks',
|
||||
description: 'Task with subtasks description',
|
||||
status: 'pending',
|
||||
dependencies: [1, 2],
|
||||
priority: 'high',
|
||||
details: 'Detailed information for task 3',
|
||||
testStrategy: 'Test strategy for task 3',
|
||||
subtasks: [
|
||||
{
|
||||
id: 1,
|
||||
title: 'Subtask 1',
|
||||
description: 'First subtask',
|
||||
status: 'pending',
|
||||
dependencies: [],
|
||||
details: 'Details for subtask 1'
|
||||
},
|
||||
{
|
||||
id: 2,
|
||||
title: 'Subtask 2',
|
||||
description: 'Second subtask',
|
||||
status: 'pending',
|
||||
dependencies: [1],
|
||||
details: 'Details for subtask 2'
|
||||
}
|
||||
]
|
||||
}
|
||||
]
|
||||
};
|
||||
|
||||
describe('setTaskStatus', () => {
|
||||
beforeEach(() => {
|
||||
jest.clearAllMocks();
|
||||
|
||||
// Mock console methods to suppress output
|
||||
jest.spyOn(console, 'log').mockImplementation(() => {});
|
||||
jest.spyOn(console, 'error').mockImplementation(() => {});
|
||||
|
||||
// Mock process.exit to prevent actual exit
|
||||
jest.spyOn(process, 'exit').mockImplementation((code) => {
|
||||
throw new Error(`process.exit: ${code}`);
|
||||
});
|
||||
|
||||
// Set up updateSingleTaskStatus mock to actually update the data
|
||||
updateSingleTaskStatus.mockImplementation(
|
||||
async (tasksPath, taskId, newStatus, data) => {
|
||||
// Handle subtask notation (e.g., "3.1")
|
||||
if (taskId.includes('.')) {
|
||||
const [parentId, subtaskId] = taskId
|
||||
.split('.')
|
||||
.map((id) => parseInt(id, 10));
|
||||
const parentTask = data.tasks.find((t) => t.id === parentId);
|
||||
if (!parentTask) {
|
||||
throw new Error(`Parent task ${parentId} not found`);
|
||||
}
|
||||
if (!parentTask.subtasks) {
|
||||
throw new Error(`Parent task ${parentId} has no subtasks`);
|
||||
}
|
||||
const subtask = parentTask.subtasks.find((st) => st.id === subtaskId);
|
||||
if (!subtask) {
|
||||
throw new Error(
|
||||
`Subtask ${subtaskId} not found in parent task ${parentId}`
|
||||
);
|
||||
}
|
||||
subtask.status = newStatus;
|
||||
} else {
|
||||
// Handle regular task
|
||||
const task = data.tasks.find((t) => t.id === parseInt(taskId, 10));
|
||||
if (!task) {
|
||||
throw new Error(`Task ${taskId} not found`);
|
||||
}
|
||||
task.status = newStatus;
|
||||
|
||||
// If marking parent as done, mark all subtasks as done too
|
||||
if (newStatus === 'done' && task.subtasks) {
|
||||
task.subtasks.forEach((subtask) => {
|
||||
subtask.status = 'done';
|
||||
});
|
||||
}
|
||||
}
|
||||
}
|
||||
);
|
||||
});
|
||||
|
||||
afterEach(() => {
|
||||
// Restore console methods
|
||||
jest.restoreAllMocks();
|
||||
});
|
||||
|
||||
test('should update task status in tasks.json', async () => {
|
||||
// Arrange
|
||||
const testTasksData = JSON.parse(JSON.stringify(sampleTasks));
|
||||
const tasksPath = '/mock/path/tasks.json';
|
||||
|
||||
readJSON.mockReturnValue(testTasksData);
|
||||
|
||||
// Act
|
||||
await setTaskStatus(tasksPath, '2', 'done', {
|
||||
mcpLog: { info: jest.fn() }
|
||||
});
|
||||
|
||||
// Assert
|
||||
expect(readJSON).toHaveBeenCalledWith(tasksPath);
|
||||
expect(writeJSON).toHaveBeenCalledWith(
|
||||
tasksPath,
|
||||
expect.objectContaining({
|
||||
tasks: expect.arrayContaining([
|
||||
expect.objectContaining({ id: 2, status: 'done' })
|
||||
])
|
||||
})
|
||||
);
|
||||
expect(generateTaskFiles).toHaveBeenCalledWith(
|
||||
tasksPath,
|
||||
expect.any(String),
|
||||
expect.any(Object)
|
||||
);
|
||||
});
|
||||
|
||||
test('should update subtask status when using dot notation', async () => {
|
||||
// Arrange
|
||||
const testTasksData = JSON.parse(JSON.stringify(sampleTasks));
|
||||
const tasksPath = '/mock/path/tasks.json';
|
||||
|
||||
readJSON.mockReturnValue(testTasksData);
|
||||
|
||||
// Act
|
||||
await setTaskStatus(tasksPath, '3.1', 'done', {
|
||||
mcpLog: { info: jest.fn() }
|
||||
});
|
||||
|
||||
// Assert
|
||||
expect(readJSON).toHaveBeenCalledWith(tasksPath);
|
||||
expect(writeJSON).toHaveBeenCalledWith(
|
||||
tasksPath,
|
||||
expect.objectContaining({
|
||||
tasks: expect.arrayContaining([
|
||||
expect.objectContaining({
|
||||
id: 3,
|
||||
subtasks: expect.arrayContaining([
|
||||
expect.objectContaining({ id: 1, status: 'done' })
|
||||
])
|
||||
})
|
||||
])
|
||||
})
|
||||
);
|
||||
});
|
||||
|
||||
test('should update multiple tasks when given comma-separated IDs', async () => {
|
||||
// Arrange
|
||||
const testTasksData = JSON.parse(JSON.stringify(sampleTasks));
|
||||
const tasksPath = '/mock/path/tasks.json';
|
||||
|
||||
readJSON.mockReturnValue(testTasksData);
|
||||
|
||||
// Act
|
||||
await setTaskStatus(tasksPath, '1,2', 'done', {
|
||||
mcpLog: { info: jest.fn() }
|
||||
});
|
||||
|
||||
// Assert
|
||||
expect(readJSON).toHaveBeenCalledWith(tasksPath);
|
||||
expect(writeJSON).toHaveBeenCalledWith(
|
||||
tasksPath,
|
||||
expect.objectContaining({
|
||||
tasks: expect.arrayContaining([
|
||||
expect.objectContaining({ id: 1, status: 'done' }),
|
||||
expect.objectContaining({ id: 2, status: 'done' })
|
||||
])
|
||||
})
|
||||
);
|
||||
});
|
||||
|
||||
test('should automatically mark subtasks as done when parent is marked done', async () => {
|
||||
// Arrange
|
||||
const testTasksData = JSON.parse(JSON.stringify(sampleTasks));
|
||||
const tasksPath = '/mock/path/tasks.json';
|
||||
|
||||
readJSON.mockReturnValue(testTasksData);
|
||||
|
||||
// Act
|
||||
await setTaskStatus(tasksPath, '3', 'done', {
|
||||
mcpLog: { info: jest.fn() }
|
||||
});
|
||||
|
||||
// Assert
|
||||
expect(writeJSON).toHaveBeenCalledWith(
|
||||
tasksPath,
|
||||
expect.objectContaining({
|
||||
tasks: expect.arrayContaining([
|
||||
expect.objectContaining({
|
||||
id: 3,
|
||||
status: 'done',
|
||||
subtasks: expect.arrayContaining([
|
||||
expect.objectContaining({ id: 1, status: 'done' }),
|
||||
expect.objectContaining({ id: 2, status: 'done' })
|
||||
])
|
||||
})
|
||||
])
|
||||
})
|
||||
);
|
||||
});
|
||||
|
||||
test('should throw error for non-existent task ID', async () => {
|
||||
// Arrange
|
||||
const testTasksData = JSON.parse(JSON.stringify(sampleTasks));
|
||||
const tasksPath = '/mock/path/tasks.json';
|
||||
|
||||
readJSON.mockReturnValue(testTasksData);
|
||||
|
||||
// Act & Assert
|
||||
await expect(
|
||||
setTaskStatus(tasksPath, '99', 'done', { mcpLog: { info: jest.fn() } })
|
||||
).rejects.toThrow('Task 99 not found');
|
||||
});
|
||||
|
||||
test('should throw error for invalid status', async () => {
|
||||
// Arrange
|
||||
const testTasksData = JSON.parse(JSON.stringify(sampleTasks));
|
||||
const tasksPath = '/mock/path/tasks.json';
|
||||
|
||||
readJSON.mockReturnValue(testTasksData);
|
||||
|
||||
// Act & Assert
|
||||
await expect(
|
||||
setTaskStatus(tasksPath, '2', 'InvalidStatus', {
|
||||
mcpLog: { info: jest.fn() }
|
||||
})
|
||||
).rejects.toThrow(/Invalid status value: InvalidStatus/);
|
||||
});
|
||||
|
||||
test('should handle parent tasks without subtasks when updating subtask', async () => {
|
||||
// Arrange
|
||||
const testTasksData = JSON.parse(JSON.stringify(sampleTasks));
|
||||
// Remove subtasks from task 3
|
||||
testTasksData.tasks[2] = { ...testTasksData.tasks[2] };
|
||||
delete testTasksData.tasks[2].subtasks;
|
||||
|
||||
const tasksPath = '/mock/path/tasks.json';
|
||||
readJSON.mockReturnValue(testTasksData);
|
||||
|
||||
// Act & Assert
|
||||
await expect(
|
||||
setTaskStatus(tasksPath, '3.1', 'done', { mcpLog: { info: jest.fn() } })
|
||||
).rejects.toThrow('has no subtasks');
|
||||
});
|
||||
|
||||
test('should handle non-existent subtask ID', async () => {
|
||||
// Arrange
|
||||
const testTasksData = JSON.parse(JSON.stringify(sampleTasks));
|
||||
const tasksPath = '/mock/path/tasks.json';
|
||||
|
||||
readJSON.mockReturnValue(testTasksData);
|
||||
|
||||
// Act & Assert
|
||||
await expect(
|
||||
setTaskStatus(tasksPath, '3.99', 'done', { mcpLog: { info: jest.fn() } })
|
||||
).rejects.toThrow('Subtask 99 not found');
|
||||
});
|
||||
|
||||
test('should handle file read errors', async () => {
|
||||
// Arrange
|
||||
const tasksPath = 'tasks/tasks.json';
|
||||
const taskId = '2';
|
||||
const newStatus = 'done';
|
||||
|
||||
readJSON.mockImplementation(() => {
|
||||
throw new Error('File not found');
|
||||
});
|
||||
|
||||
// Act & Assert
|
||||
await expect(
|
||||
setTaskStatus(tasksPath, taskId, newStatus, {
|
||||
mcpLog: { info: jest.fn() }
|
||||
})
|
||||
).rejects.toThrow('File not found');
|
||||
|
||||
// Verify that writeJSON was not called due to read error
|
||||
expect(writeJSON).not.toHaveBeenCalled();
|
||||
});
|
||||
|
||||
test('should handle empty task ID input', async () => {
|
||||
// Arrange
|
||||
const tasksPath = 'tasks/tasks.json';
|
||||
const emptyTaskId = '';
|
||||
const newStatus = 'done';
|
||||
|
||||
// Act & Assert
|
||||
await expect(
|
||||
setTaskStatus(tasksPath, emptyTaskId, newStatus, {
|
||||
mcpLog: { info: jest.fn() }
|
||||
})
|
||||
).rejects.toThrow();
|
||||
|
||||
// Verify that updateSingleTaskStatus was not called
|
||||
expect(updateSingleTaskStatus).not.toHaveBeenCalled();
|
||||
});
|
||||
|
||||
test('should handle whitespace in comma-separated IDs', async () => {
|
||||
// Arrange
|
||||
const testTasksData = JSON.parse(JSON.stringify(sampleTasks));
|
||||
const tasksPath = 'tasks/tasks.json';
|
||||
const taskIds = ' 1 , 2 , 3 '; // IDs with whitespace
|
||||
const newStatus = 'in-progress';
|
||||
|
||||
readJSON.mockReturnValue(testTasksData);
|
||||
|
||||
// Act
|
||||
const result = await setTaskStatus(tasksPath, taskIds, newStatus, {
|
||||
mcpLog: { info: jest.fn() }
|
||||
});
|
||||
|
||||
// Assert
|
||||
expect(updateSingleTaskStatus).toHaveBeenCalledTimes(3);
|
||||
expect(updateSingleTaskStatus).toHaveBeenCalledWith(
|
||||
tasksPath,
|
||||
'1',
|
||||
newStatus,
|
||||
testTasksData,
|
||||
false
|
||||
);
|
||||
expect(updateSingleTaskStatus).toHaveBeenCalledWith(
|
||||
tasksPath,
|
||||
'2',
|
||||
newStatus,
|
||||
testTasksData,
|
||||
false
|
||||
);
|
||||
expect(updateSingleTaskStatus).toHaveBeenCalledWith(
|
||||
tasksPath,
|
||||
'3',
|
||||
newStatus,
|
||||
testTasksData,
|
||||
false
|
||||
);
|
||||
expect(result).toBeDefined();
|
||||
});
|
||||
});
|
||||
Reference in New Issue
Block a user