|
|
|
import assert from 'assert';
|
|
|
|
import { writeFileSync } from 'fs';
|
|
|
|
import autocannon from 'autocannon';
|
|
|
|
import {
|
|
|
|
after, before, describe, it,
|
|
|
|
} from 'node:test';
|
|
|
|
import * as supertest from 'supertest';
|
|
|
|
|
|
|
|
const Root = {
|
|
|
|
username: 'root',
|
|
|
|
password: '123456',
|
|
|
|
creditionals: null,
|
|
|
|
};
|
|
|
|
|
|
|
|
describe('App', () => {
|
|
|
|
let agent: supertest.SuperAgentTest;
|
|
|
|
before(async () => {
|
|
|
|
const init = Date.now();
|
|
|
|
await new Promise((resolve) => {
|
|
|
|
process.send = ((send) => (data) => {
|
|
|
|
console.log('send', data);
|
|
|
|
if (data === 'ready') {
|
|
|
|
agent = supertest.agent(require('hydrooj').httpServer);
|
|
|
|
resolve(null);
|
|
|
|
}
|
|
|
|
return send?.(data) || false;
|
|
|
|
})(process.send);
|
|
|
|
});
|
|
|
|
console.log('Application inited in %d ms', Date.now() - init);
|
|
|
|
}, { timeout: 30000 });
|
|
|
|
|
|
|
|
const routes = ['/', '/api', '/p', '/contest', '/homework', '/user/1', '/training'];
|
|
|
|
routes.forEach((route) => it(`GET ${route}`, () => agent.get(route).expect(200)));
|
|
|
|
|
|
|
|
it('API user', async () => {
|
|
|
|
await agent.get('/api?{user(id:1){uname}}').expect({ data: { user: { uname: 'Hydro' } } });
|
|
|
|
await agent.get('/api?{user(id:2){uname}}').expect({ data: { user: null } });
|
|
|
|
});
|
|
|
|
|
|
|
|
it('Create User', async () => {
|
|
|
|
const redirect = await agent.post('/register')
|
|
|
|
.send({ mail: 'test@example.com' })
|
|
|
|
.expect(302)
|
|
|
|
.then((res) => res.headers.location);
|
|
|
|
await agent.post(redirect)
|
|
|
|
.send({ uname: Root.username, password: Root.password, verifyPassword: Root.password })
|
|
|
|
.expect(302);
|
|
|
|
});
|
|
|
|
|
|
|
|
it('Login', async () => {
|
|
|
|
const cookie = await agent.post('/login')
|
|
|
|
.send({ uname: Root.username, password: Root.password })
|
|
|
|
.expect(302)
|
|
|
|
.then((res) => res.headers['set-cookie']);
|
|
|
|
Root.creditionals = cookie;
|
|
|
|
});
|
|
|
|
|
|
|
|
it('API registered user', async () => {
|
|
|
|
await agent.get('/api?{user(id:2){uname}}').expect({ data: { user: { uname: 'root' } } });
|
|
|
|
});
|
|
|
|
|
|
|
|
// TODO add more tests
|
|
|
|
|
|
|
|
if (process.env.BENCHMARK) {
|
|
|
|
const results: Record<string, autocannon.Result> = {};
|
|
|
|
routes.forEach((route) => it(`Performance test ${route}`, { timeout: 60000 }, async () => {
|
|
|
|
await global.Hydro.model.system.set('limit.global', 99999);
|
|
|
|
const result = await autocannon({ url: `http://localhost:8888${route}` });
|
|
|
|
assert(result.errors === 0, `test ${route} returns errors`);
|
|
|
|
results[route] = result;
|
|
|
|
}));
|
|
|
|
}
|
|
|
|
|
|
|
|
after(() => {
|
|
|
|
if (process.env.BENCHMARK) {
|
|
|
|
const metrics = Object.entries(([k, v]) => ({
|
|
|
|
name: `Benchmark - ${k} - Req/sec`,
|
|
|
|
unit: 'Req/sec',
|
|
|
|
value: v.requests.average,
|
|
|
|
}));
|
|
|
|
writeFileSync('./benchmark.json', JSON.stringify(metrics, null, 2));
|
|
|
|
}
|
|
|
|
setTimeout(() => process.exit(0), 1000);
|
|
|
|
});
|
|
|
|
});
|