ros/claude-task-master
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

tests: adjusts the tests to properly pass.

Browse Source
This commit is contained in:
Eyal Toledano
2025-05-07 13:54:01 -04:00
parent 245c3cb398
commit 2b3ae8bf89
3 changed files with 45 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
.changeset/nice-lies-cover.md Normal file
View File

@@ -0,0 +1,5 @@
---
'task-master-ai': patch
---
Adds costs information to AI commands using input/output tokens and model costs.

4
scripts/modules/config-manager.js
View File

@@ -691,8 +691,8 @@ export {
// Core config access
getConfig,
writeConfig,
ConfigurationError, // Export custom error type
isConfigFilePresent, // Add the new function export
ConfigurationError,
isConfigFilePresent,
// Validation
validateProvider,

44
tests/unit/ai-services-unified.test.js
View File

@@ -8,6 +8,29 @@ const mockGetResearchModelId = jest.fn();
const mockGetFallbackProvider = jest.fn();
const mockGetFallbackModelId = jest.fn();
const mockGetParametersForRole = jest.fn();
const mockGetUserId = jest.fn();
// --- Mock MODEL_MAP Data ---
// Provide a simplified structure sufficient for cost calculation tests
const mockModelMap = {
anthropic: [
{
id: 'test-main-model',
cost_per_1m_tokens: { input: 3, output: 15, currency: 'USD' }
},
{
id: 'test-fallback-model',
cost_per_1m_tokens: { input: 3, output: 15, currency: 'USD' }
}
],
perplexity: [
{
id: 'test-research-model',
cost_per_1m_tokens: { input: 1, output: 1, currency: 'USD' }
}
]
// Add other providers/models if needed for specific tests
};
jest.unstable_mockModule('../../scripts/modules/config-manager.js', () => ({
getMainProvider: mockGetMainProvider,
@@ -16,7 +39,9 @@ jest.unstable_mockModule('../../scripts/modules/config-manager.js', () => ({
getResearchModelId: mockGetResearchModelId,
getFallbackProvider: mockGetFallbackProvider,
getFallbackModelId: mockGetFallbackModelId,
getParametersForRole: mockGetParametersForRole
getParametersForRole: mockGetParametersForRole,
getUserId: mockGetUserId,
MODEL_MAP: mockModelMap
}));
// Mock AI Provider Modules
@@ -44,10 +69,13 @@ jest.unstable_mockModule('../../src/ai-providers/perplexity.js', () => ({
const mockLog = jest.fn();
const mockResolveEnvVariable = jest.fn();
const mockFindProjectRoot = jest.fn();
const mockIsSilentMode = jest.fn();
jest.unstable_mockModule('../../scripts/modules/utils.js', () => ({
log: mockLog,
resolveEnvVariable: mockResolveEnvVariable,
findProjectRoot: mockFindProjectRoot
findProjectRoot: mockFindProjectRoot,
isSilentMode: mockIsSilentMode
}));
// Import the module to test (AFTER mocks)
@@ -97,7 +125,8 @@ describe('Unified AI Services', () => {
};
const result = await generateTextService(params);
expect(result).toBe('Main provider response');
expect(result.mainResult).toBe('Main provider response');
expect(result).toHaveProperty('telemetryData');
expect(mockGetMainProvider).toHaveBeenCalledWith(fakeProjectRoot);
expect(mockGetMainModelId).toHaveBeenCalledWith(fakeProjectRoot);
expect(mockGetParametersForRole).toHaveBeenCalledWith(
@@ -137,7 +166,8 @@ describe('Unified AI Services', () => {
};
const result = await generateTextService(params);
expect(result).toBe('Fallback provider response');
expect(result.mainResult).toBe('Fallback provider response');
expect(result).toHaveProperty('telemetryData');
expect(mockGetMainProvider).toHaveBeenCalledWith(explicitRoot);
expect(mockGetFallbackProvider).toHaveBeenCalledWith(explicitRoot);
expect(mockGetParametersForRole).toHaveBeenCalledWith(
@@ -180,7 +210,8 @@ describe('Unified AI Services', () => {
const params = { role: 'main', prompt: 'Research fallback test' };
const result = await generateTextService(params);
expect(result).toBe('Research provider response');
expect(result.mainResult).toBe('Research provider response');
expect(result).toHaveProperty('telemetryData');
expect(mockGetMainProvider).toHaveBeenCalledWith(fakeProjectRoot);
expect(mockGetFallbackProvider).toHaveBeenCalledWith(fakeProjectRoot);
expect(mockGetResearchProvider).toHaveBeenCalledWith(fakeProjectRoot);
@@ -252,7 +283,8 @@ describe('Unified AI Services', () => {
const params = { role: 'main', prompt: 'Retry success test' };
const result = await generateTextService(params);
expect(result).toBe('Success after retry');
expect(result.mainResult).toBe('Success after retry');
expect(result).toHaveProperty('telemetryData');
expect(mockGenerateAnthropicText).toHaveBeenCalledTimes(2); // Initial + 1 retry
expect(mockLog).toHaveBeenCalledWith(
'info',