import { open } from 'k6/experimental/fs';
import csv from 'k6/experimental/csv';
import { SharedArray } from 'k6/data';
import http from 'k6/http';
import { check, sleep } from 'k6';
import { vu } from 'k6/execution';

const csvData = new SharedArray('CSV data', function () {
  const file = open('./users.csv');
  return csv.parse(file, { header: true, asObjects: true });
});

export const options = {
  vus: 10,
  duration: '30s',
  thresholds: {
    http_req_failed: ['rate<0.01'],
    http_req_duration: ['p(95)<200'],
  },
};

export default function () {
  const user = csvData[vu.idInTest % csvData.length];
  console.log(`VU ${vu.idInTest} usando usuário: ${user.username}`);

  const payload = JSON.stringify({ username: user.username, password: user.password });
  const params = { headers: { 'Content-Type': 'application/json' } };
  const res = http.post('https://api.example.com/login', payload, params);

  check(res, {
    'status é 200': (r) => r.status === 200,
    'login bem-sucedido': (r) => r.json().message === 'Login successful',
  });

  sleep(1);
}

export const options = {
  scenarios: {
    tps_load_test: {
      executor: 'constant-arrival-rate',
      rate: 300, // TPS
      timeUnit: '1s',
      duration: '5m',
      preAllocatedVUs: 50,
      maxVUs: 200,
      tags: { test_type: 'tps_validation' },
    },
  },
};

export const options = {
  scenarios: {
    global_tps_test: {
      executor: 'constant-arrival-rate',
      rate: 300,
      timeUnit: '1s',
      duration: '5m',
      preAllocatedVUs: 50,
      maxVUs: 500,
    },
  },
  thresholds: {
    http_req_failed: ['rate<0.01'],
    http_req_duration: ['p(95)<2000'],
    'http_req_duration{status:429}': ['count<100'],
  },
};

export const options = {
  scenarios: {
    tps_target: {
      executor: 'constant-arrival-rate',
      rate: 300,
      timeUnit: '1s',
      duration: '1m',
      preAllocatedVUs: 100,
      maxVUs: 500,
    },
  },
};