import http from 'k6/http';
import { check, fail } from 'k6';

export const options = {
  vus: 1,
  duration: '10s',
};

export default function() {
  const url = 'http://43.198.252.255:8000/api/v1//txt2img/base/llm';
  const payload = JSON.stringify({
		      "test": 1
		      })
  const param = {
    headers:{
		'apikey': 'L6eXzteHYa2Uq0ar4kdMaBOiOIjmRI08',
		'Content-Type': 'application/json'
	},
  };
  var res = http.post(url, payload, param);
  const statusCheck = check(res, {
    'status is 200': (r) => r.status === 200,
  });
  const taskCheck = check(res, {
    'task is success': (r) => r.json('task.is_success') === true,
  });
  if (!statusCheck) {
    // console.log('task status failed', res.status);
      fail('task ' + res.headers['Task-Id'] + ' failed with status error: ' + res.status);
  } else if (!taskCheck) {
      fail('task ' + res.headers['Task-Id'] + ' failed with execute error: ' + res.json('task.error'));
  }
  // console.log("res is ", res);
}