119 lines
2.9 KiB
JavaScript
119 lines
2.9 KiB
JavaScript
import { jest } from '@jest/globals';
|
|
import fs from 'fs';
|
|
import path from 'path';
|
|
import os from 'os';
|
|
|
|
// Mock external modules
|
|
jest.mock('child_process', () => ({
|
|
execSync: jest.fn()
|
|
}));
|
|
|
|
// Mock console methods
|
|
jest.mock('console', () => ({
|
|
log: jest.fn(),
|
|
info: jest.fn(),
|
|
warn: jest.fn(),
|
|
error: jest.fn(),
|
|
clear: jest.fn()
|
|
}));
|
|
|
|
describe('Trae Integration', () => {
|
|
let tempDir;
|
|
|
|
beforeEach(() => {
|
|
jest.clearAllMocks();
|
|
|
|
// Create a temporary directory for testing
|
|
tempDir = fs.mkdtempSync(path.join(os.tmpdir(), 'task-master-test-'));
|
|
|
|
// Spy on fs methods
|
|
jest.spyOn(fs, 'writeFileSync').mockImplementation(() => {});
|
|
jest.spyOn(fs, 'readFileSync').mockImplementation((filePath) => {
|
|
if (filePath.toString().includes('.trae')) {
|
|
return 'Existing trae rules content';
|
|
}
|
|
return '{}';
|
|
});
|
|
jest.spyOn(fs, 'existsSync').mockImplementation(() => false);
|
|
jest.spyOn(fs, 'mkdirSync').mockImplementation(() => {});
|
|
});
|
|
|
|
afterEach(() => {
|
|
// Clean up the temporary directory
|
|
try {
|
|
fs.rmSync(tempDir, { recursive: true, force: true });
|
|
} catch (err) {
|
|
console.error(`Error cleaning up: ${err.message}`);
|
|
}
|
|
});
|
|
|
|
// Test function that simulates the createProjectStructure behavior for Trae files
|
|
function mockCreateTraeStructure() {
|
|
// Create main .trae directory
|
|
fs.mkdirSync(path.join(tempDir, '.trae'), { recursive: true });
|
|
|
|
// Create rules directory
|
|
fs.mkdirSync(path.join(tempDir, '.trae', 'rules'), { recursive: true });
|
|
|
|
// Create rule files
|
|
const ruleFiles = [
|
|
'dev_workflow.md',
|
|
'taskmaster.md',
|
|
'architecture.md',
|
|
'commands.md',
|
|
'dependencies.md'
|
|
];
|
|
|
|
for (const ruleFile of ruleFiles) {
|
|
fs.writeFileSync(
|
|
path.join(tempDir, '.trae', 'rules', ruleFile),
|
|
`Content for ${ruleFile}`
|
|
);
|
|
}
|
|
}
|
|
|
|
test('creates all required .trae directories', () => {
|
|
// Act
|
|
mockCreateTraeStructure();
|
|
|
|
// Assert
|
|
expect(fs.mkdirSync).toHaveBeenCalledWith(path.join(tempDir, '.trae'), {
|
|
recursive: true
|
|
});
|
|
expect(fs.mkdirSync).toHaveBeenCalledWith(
|
|
path.join(tempDir, '.trae', 'rules'),
|
|
{ recursive: true }
|
|
);
|
|
});
|
|
|
|
test('creates rule files for Trae', () => {
|
|
// Act
|
|
mockCreateTraeStructure();
|
|
|
|
// Assert - check rule files are created
|
|
expect(fs.writeFileSync).toHaveBeenCalledWith(
|
|
path.join(tempDir, '.trae', 'rules', 'dev_workflow.md'),
|
|
expect.any(String)
|
|
);
|
|
expect(fs.writeFileSync).toHaveBeenCalledWith(
|
|
path.join(tempDir, '.trae', 'rules', 'taskmaster.md'),
|
|
expect.any(String)
|
|
);
|
|
expect(fs.writeFileSync).toHaveBeenCalledWith(
|
|
path.join(tempDir, '.trae', 'rules', 'architecture.md'),
|
|
expect.any(String)
|
|
);
|
|
});
|
|
|
|
test('does not create MCP configuration files', () => {
|
|
// Act
|
|
mockCreateTraeStructure();
|
|
|
|
// Assert - Trae doesn't use MCP configuration
|
|
expect(fs.writeFileSync).not.toHaveBeenCalledWith(
|
|
path.join(tempDir, '.trae', 'mcp.json'),
|
|
expect.any(String)
|
|
);
|
|
});
|
|
});
|