Skip to content

Commit

Permalink
refactor(tests): extract helper functions for worker and sequence tes…
Browse files Browse the repository at this point in the history
…ting

Move common test utility functions to a shared _helpers.ts file to improve code reusability
and reduce duplication in e2e tests
  • Loading branch information
jumski committed Jan 9, 2025
1 parent f96726c commit 0195930
Show file tree
Hide file tree
Showing 2 changed files with 39 additions and 30 deletions.
37 changes: 37 additions & 0 deletions pkgs/supaworker/tests/e2e/_helpers.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,37 @@
import { sql } from '../sql.ts';
import { delay } from 'jsr:@std/async';

export async function seqLastValue(): Promise<number> {
const seqResult = await sql`SELECT last_value::integer FROM test_seq`;
return seqResult[0].last_value;
}

export async function waitForSeqValue(
value: number,
message: string = 'Polling...'
): Promise<number> {
let lastVal = 0;
while (lastVal < value) {
console.log(`${message} current value:`, lastVal);
await delay(1000);
lastVal = await seqLastValue();
}
return lastVal;
}

export async function fetchWorkers(workerName: string) {
return await sql`SELECT * FROM supaworker.workers`;
}

export async function startWorker(workerName: string, seconds: number = 5) {
await sql`SELECT supaworker.spawn(${workerName})`;

let workers = await fetchWorkers(workerName);
console.log('Waiting for worker to spawn...');

while (workers.length === 0) {
await delay(500);
workers = await fetchWorkers(workerName);
}
console.log('Worker spawned!');
}
32 changes: 2 additions & 30 deletions pkgs/supaworker/tests/e2e/worker_spawns_next_worker.test.ts
Original file line number Diff line number Diff line change
@@ -1,28 +1,6 @@
import { sql } from '../sql.ts';
import { assertEquals, assertGreaterOrEqual } from 'jsr:@std/assert';
import { delay } from 'jsr:@std/async';

async function seqLastValue(): Promise<number> {
const seqResult = await sql`SELECT last_value::integer FROM test_seq`;
return seqResult[0].last_value;
}

async function fetchWorkers(workerName: string) {
return await sql`SELECT * FROM supaworker.workers`;
}

async function startWorker(workerName: string, seconds: number = 5) {
await sql`SELECT supaworker.spawn(${workerName})`;

let workers = await fetchWorkers(workerName);
console.log('Waiting for worker to spawn...');

while (workers.length === 0) {
await delay(500);
workers = await fetchWorkers(workerName);
}
console.log('Worker spawned!');
}
import { fetchWorkers, startWorker, waitForSeqValue } from './_helpers.ts';

Deno.test('should spawn next worker when CPU clock limit hits', async () => {
await startWorker('increment-sequence');
Expand All @@ -44,13 +22,7 @@ Deno.test('should spawn next worker when CPU clock limit hits', async () => {
)
)`;

let lastVal = 0;
while (lastVal < MESSAGES_TO_SEND) {
let w = await fetchWorkers('increment-sequence');
console.log('Polling... current value:', lastVal, w);
await delay(1000);
lastVal = await seqLastValue();
}
const lastVal = await waitForSeqValue(MESSAGES_TO_SEND);

assertGreaterOrEqual(
lastVal,
Expand Down

0 comments on commit 0195930

Please sign in to comment.