artillery-processor.js

module.exports = {
    generateRandomEmail: function (context, events, done) {
        // Ensure context.vars exists
        if (!context.vars) {
            context.vars = {};
        }

        const timestamp = Date.now();
        const random = Math.floor(Math.random() * 10000);
        context.vars.email = `test${timestamp}${random}@example.com`;
        context.vars.password = 'testpassword123';
        context.vars.name = `Test User ${random}`;
        return done();
    },

    logResponseTime: function (requestParams, response, context, ee, next) {
        if (response && response.body && response.body.data) {
            console.log(
                `Request: ${requestParams.name || 'unknown'}, Response Time: ${response.timings ? response.timings.response : 'unknown'}ms`
            );
        }
        return next();
    },

    trackRabbitMQMetrics: function (context, events, done) {
        console.log('Tracking RabbitMQ metrics...');
        return done();
    },

    assertResponseTime: function (requestParams, response, context, ee, next) {
        if (
            requestParams.name === 'sendMessage' &&
            response &&
            response.timings
        ) {
            if (response.timings.response > 500) {
                console.warn(
                    `WARNING: sendMessage took ${response.timings.response}ms (expected < 500ms)`
                );
            }
        }
        return next();
    },

    generateTestMessage: function (context, events, done) {
        // Ensure context.vars exists
        if (!context.vars) {
            context.vars = {};
        }

        const messages = [
            'Hello from performance test!',
            'Testing message sending speed',
            'Performance test message with some content',
            'Quick message for load testing',
            'Another test message for performance evaluation',
        ];

        context.vars.testMessage =
            messages[Math.floor(Math.random() * messages.length)];
        return done();
    },

    monitorMemory: function (context, events, done) {
        const used = process.memoryUsage();
        console.log(
            `Artillery Memory Usage - RSS: ${Math.round((used.rss / 1024 / 1024) * 100) / 100} MB`
        );
        return done();
    },

    setupPhase: function (context, events, done) {
        console.log('Setting up performance test...');

        if (!global.performanceMetrics) {
            global.performanceMetrics = {
                sendMessageTimes: [],
                authTimes: [],
                errors: 0,
                startTime: Date.now(),
            };
        }

        return done();
    },

    recordMetrics: function (requestParams, response, context, ee, next) {
        if (!global.performanceMetrics) {
            global.performanceMetrics = {
                sendMessageTimes: [],
                authTimes: [],
                errors: 0,
                startTime: Date.now(),
            };
        }

        if (response && response.timings && response.timings.response) {
            const responseTime = response.timings.response;

            if (requestParams.name === 'sendMessage') {
                global.performanceMetrics.sendMessageTimes.push(responseTime);
            } else if (
                requestParams.url &&
                requestParams.url.includes('signin')
            ) {
                global.performanceMetrics.authTimes.push(responseTime);
            }
        }

        if (
            (response && response.statusCode >= 400) ||
            (response && response.body && response.body.errors)
        ) {
            global.performanceMetrics.errors++;
        }

        return next();
    },

    printSummary: function (context, events, done) {
        if (
            global.performanceMetrics &&
            global.performanceMetrics.sendMessageTimes.length > 0
        ) {
            const sendTimes = global.performanceMetrics.sendMessageTimes;
            const avgSendTime =
                sendTimes.reduce((a, b) => a + b, 0) / sendTimes.length;
            const maxSendTime = Math.max(...sendTimes);
            const minSendTime = Math.min(...sendTimes);

            console.log('\n=== PERFORMANCE SUMMARY ===');
            console.log(
                `Send Message - Avg: ${avgSendTime.toFixed(2)}ms, Min: ${minSendTime}ms, Max: ${maxSendTime}ms`
            );
            console.log(`Total Errors: ${global.performanceMetrics.errors}`);
            console.log(
                `Test Duration: ${((Date.now() - global.performanceMetrics.startTime) / 1000).toFixed(2)}s`
            );
            console.log('===========================\n');
        }

        return done();
    },

    logResponse: function (requestParams, response, context, ee, next) {
        console.log('\n=== DEBUG RESPONSE ===');
        console.log(`URL: ${requestParams.url}`);
        console.log(`Status: ${response.statusCode}`);
        console.log(`Body:`, JSON.stringify(response.body, null, 2));
        console.log('======================\n');
        return next();
    },
};