import http from 'k6/http';
import { check, sleep, group } from 'k6';

const vus = parseInt(__ENV.TEST_VUS || '10');
const duration = __ENV.TEST_DURATION || '30s';
const targetUrl = __ENV.TARGET_URL || 'http://frontend.demo-apps.svc.cluster.local:80';

export const options = {
  scenarios: {
    load_test: {
      executor: 'ramping-vus',
      startVUs: 0,
      stages: [
        { duration: '10s', target: vus },
        { duration: duration, target: vus },
        { duration: '5s', target: 0 },
      ],
    },
  },
  thresholds: {
    http_req_duration: ['p(95)<500'],
    http_req_failed: ['rate<0.01'],
  },
};

http.setResponseCallback(http.expectedStatuses({ min: 200, max: 399 }));

export default function () {
  group('Welcome Page', () => {
    const res = http.get(`${targetUrl}/`);
    check(res, {
      'status is 200': (r) => r.status === 200,
      'response time < 500ms': (r) => r.timings.duration < 500,
    });
  });

  group('Owner API', () => {
    const ownerCreationRes = http.post(`${targetUrl}/owners/new`, 'firstName=John&lastName=Doe&address=123+Main+St&city=Springfield&telephone=1234567890');
    check(ownerCreationRes, {
      'status is 302': (r) => r.status === 302,
    });

    const ownerSearchRes = http.get(`${targetUrl}/owners?lastName=Doe`);
    check(ownerSearchRes, {
      'status is 200': (r) => r.status === 200,
      'response time < 500ms': (r) => r.timings.duration < 500,
    });
  });

  group('Vet API', () => {
    const vetListRes = http.get(`${targetUrl}/vets`);
    check(vetListRes, {
      'status is 200': (r) => r.status === 200,
      'response time < 500ms': (r) => r.timings.duration < 500,
    });
  });

  group('System API', () => {
    const crashRes = http.get(`${targetUrl}/oups`, { responseCallback: http.expectedStatuses(500) });
    check(crashRes, {
      'status is 500': (r) => r.status === 500,
    });
  });

  sleep(0.5);
}