alighasami
3d5eaf9445
Some checks failed
Security: Sync from Public / sync-from-public (push) Has been cancelled
Test: Benchmark Nightly / build (push) Has been cancelled
Test: Benchmark Nightly / Notify Cats on failure (push) Has been cancelled
CI: Python / Checks (push) Has been cancelled
Test: Evals Python / Workflow Comparison Python (push) Has been cancelled
Util: Check Docs URLs / check-docs-urls (push) Has been cancelled
Test: Visual Storybook / Cloudflare Pages (push) Has been cancelled
Test: E2E Performance / build-and-test-performance (push) Has been cancelled
Test: Workflows Nightly / Run Workflow Tests (push) Has been cancelled
Util: Cleanup CI Docker Images / Delete stale CI images (push) Has been cancelled
Test: Benchmark Destroy Env / build (push) Has been cancelled
Util: Update Node Popularity / update-popularity (push) Has been cancelled
Test: E2E Coverage Weekly / Coverage Tests (push) Has been cancelled
45 lines
1.5 KiB
TypeScript
45 lines
1.5 KiB
TypeScript
import { setupBrokerTestServer } from '@test-integration/utils/task-broker-test-server';
|
|
|
|
describe('TaskBrokerServer', () => {
|
|
const { agent, server } = setupBrokerTestServer({
|
|
authToken: 'token',
|
|
mode: 'external',
|
|
});
|
|
|
|
beforeAll(async () => {
|
|
await server.start();
|
|
});
|
|
|
|
afterAll(async () => {
|
|
await server.stop();
|
|
});
|
|
|
|
describe('/healthz', () => {
|
|
it('should return 200', async () => {
|
|
await agent.get('/healthz').expect(200);
|
|
});
|
|
});
|
|
|
|
describe('/runners/_ws', () => {
|
|
it('should return 429 when too many requests are made', async () => {
|
|
await agent.post('/runners/_ws').send({}).expect(401);
|
|
await agent.post('/runners/_ws').send({}).expect(401);
|
|
await agent.post('/runners/_ws').send({}).expect(401);
|
|
await agent.post('/runners/_ws').send({}).expect(401);
|
|
await agent.post('/runners/_ws').send({}).expect(401);
|
|
await agent.post('/runners/_ws').send({}).expect(429);
|
|
});
|
|
});
|
|
|
|
describe('/runners/auth', () => {
|
|
it('should return 429 when too many requests are made', async () => {
|
|
await agent.post('/runners/auth').send({ token: 'invalid' }).expect(403);
|
|
await agent.post('/runners/auth').send({ token: 'invalid' }).expect(403);
|
|
await agent.post('/runners/auth').send({ token: 'invalid' }).expect(403);
|
|
await agent.post('/runners/auth').send({ token: 'invalid' }).expect(403);
|
|
await agent.post('/runners/auth').send({ token: 'invalid' }).expect(403);
|
|
await agent.post('/runners/auth').send({ token: 'invalid' }).expect(429);
|
|
});
|
|
});
|
|
});
|