chore: improve add-task command e2e test
This commit is contained in:
109
tests/e2e/utils/logger.cjs
Normal file
109
tests/e2e/utils/logger.cjs
Normal file
@@ -0,0 +1,109 @@
|
||||
// Simple console colors fallback if chalk is not available
|
||||
const colors = {
|
||||
green: (text) => `\x1b[32m${text}\x1b[0m`,
|
||||
red: (text) => `\x1b[31m${text}\x1b[0m`,
|
||||
yellow: (text) => `\x1b[33m${text}\x1b[0m`,
|
||||
blue: (text) => `\x1b[34m${text}\x1b[0m`,
|
||||
cyan: (text) => `\x1b[36m${text}\x1b[0m`,
|
||||
gray: (text) => `\x1b[90m${text}\x1b[0m`
|
||||
};
|
||||
|
||||
class TestLogger {
|
||||
constructor(testName = 'test') {
|
||||
this.testName = testName;
|
||||
this.startTime = Date.now();
|
||||
this.stepCount = 0;
|
||||
this.logBuffer = [];
|
||||
this.totalCost = 0;
|
||||
}
|
||||
|
||||
_formatMessage(level, message, options = {}) {
|
||||
const timestamp = new Date().toISOString();
|
||||
const elapsed = ((Date.now() - this.startTime) / 1000).toFixed(2);
|
||||
const formattedMessage = `[${timestamp}] [${elapsed}s] [${level}] ${message}`;
|
||||
|
||||
// Add to buffer for later saving if needed
|
||||
this.logBuffer.push(formattedMessage);
|
||||
|
||||
return formattedMessage;
|
||||
}
|
||||
|
||||
_log(level, message, color) {
|
||||
const formatted = this._formatMessage(level, message);
|
||||
|
||||
if (process.env.E2E_VERBOSE !== 'false') {
|
||||
console.log(color ? color(formatted) : formatted);
|
||||
}
|
||||
}
|
||||
|
||||
info(message) {
|
||||
this._log('INFO', message, colors.blue);
|
||||
}
|
||||
|
||||
success(message) {
|
||||
this._log('SUCCESS', message, colors.green);
|
||||
}
|
||||
|
||||
error(message) {
|
||||
this._log('ERROR', message, colors.red);
|
||||
}
|
||||
|
||||
warning(message) {
|
||||
this._log('WARNING', message, colors.yellow);
|
||||
}
|
||||
|
||||
step(message) {
|
||||
this.stepCount++;
|
||||
this._log('STEP', `Step ${this.stepCount}: ${message}`, colors.cyan);
|
||||
}
|
||||
|
||||
debug(message) {
|
||||
if (process.env.DEBUG) {
|
||||
this._log('DEBUG', message, colors.gray);
|
||||
}
|
||||
}
|
||||
|
||||
flush() {
|
||||
// In CommonJS version, we'll just clear the buffer
|
||||
// Real implementation would write to file if needed
|
||||
this.logBuffer = [];
|
||||
}
|
||||
|
||||
summary() {
|
||||
const duration = ((Date.now() - this.startTime) / 1000).toFixed(2);
|
||||
const summary = `Test completed in ${duration}s`;
|
||||
this.info(summary);
|
||||
return {
|
||||
duration: parseFloat(duration),
|
||||
steps: this.stepCount,
|
||||
totalCost: this.totalCost
|
||||
};
|
||||
}
|
||||
|
||||
extractAndAddCost(output) {
|
||||
// Extract cost information from LLM output
|
||||
const costPatterns = [
|
||||
/Total Cost: \$?([\d.]+)/i,
|
||||
/Cost: \$?([\d.]+)/i,
|
||||
/Estimated cost: \$?([\d.]+)/i
|
||||
];
|
||||
|
||||
for (const pattern of costPatterns) {
|
||||
const match = output.match(pattern);
|
||||
if (match) {
|
||||
const cost = parseFloat(match[1]);
|
||||
this.totalCost += cost;
|
||||
this.debug(
|
||||
`Added cost: $${cost} (Total: $${this.totalCost.toFixed(4)})`
|
||||
);
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
getTotalCost() {
|
||||
return this.totalCost;
|
||||
}
|
||||
}
|
||||
|
||||
module.exports = { TestLogger };
|
||||
Reference in New Issue
Block a user