48 lines
1.3 KiB
JavaScript
48 lines
1.3 KiB
JavaScript
|
// Copyright (C) CVAT.ai Corporation
|
||
|
|
//
|
||
|
|
// SPDX-License-Identifier: MIT
|
||
|
|
import http from 'k6/http';
|
||
|
|
import { validateResponse } from '../../utils/validation.js';
|
||
|
|
import { BASE_URL } from '../../variables/constants.js';
|
||
|
|
|
||
|
|
function createJob(token, jobSpec) {
|
||
|
|
const response = http.post(`${BASE_URL}/jobs`, JSON.stringify(jobSpec), {
|
||
|
|
headers: {
|
||
|
|
Authorization: `Token ${token}`,
|
||
|
|
'Content-Type': 'application/json',
|
||
|
|
},
|
||
|
|
});
|
||
|
|
if (validateResponse(response, 201, 'Create Job')) {
|
||
|
|
return response.json().id;
|
||
|
|
}
|
||
|
|
return null;
|
||
|
|
}
|
||
|
|
|
||
|
|
function listJobs(token) {
|
||
|
|
const response = http.get(`${BASE_URL}/jobs`, {
|
||
|
|
headers: {
|
||
|
|
Authorization: `Token ${token}`,
|
||
|
|
'Content-Type': 'application/json',
|
||
|
|
},
|
||
|
|
});
|
||
|
|
if (validateResponse(response, 200, 'List Jobs')) {
|
||
|
|
return response.json();
|
||
|
|
}
|
||
|
|
return null;
|
||
|
|
}
|
||
|
|
|
||
|
|
function getJobDetails(token, jobID) {
|
||
|
|
const response = http.get(`${BASE_URL}/jobs/${jobID}`, {
|
||
|
|
headers: {
|
||
|
|
Authorization: `Token ${token}`,
|
||
|
|
'Content-Type': 'application/json',
|
||
|
|
},
|
||
|
|
});
|
||
|
|
if (validateResponse(response, 200, 'Get Job Details')) {
|
||
|
|
return response.json();
|
||
|
|
}
|
||
|
|
return null;
|
||
|
|
}
|
||
|
|
|
||
|
|
export default { createJob, listJobs, getJobDetails };
|