Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
47 changes: 47 additions & 0 deletions worker/src/testing/__tests__/test-utils.test.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,47 @@
import { describe, expect, it, vi } from 'bun:test';
import { createMockLogCollector, createMockTrace } from '../test-utils';

describe('worker test utilities', () => {
it('only mirrors trace and log events to console when test diagnostics are enabled', async () => {
const previousDebugValue = process.env.SENTRIS_DEBUG_TESTS;
const logSpy = vi.spyOn(console, 'log').mockImplementation(() => {});

try {
delete process.env.SENTRIS_DEBUG_TESTS;

createMockTrace().record({
type: 'NODE_STARTED',
nodeRef: 'node-1',
message: 'started',
});
await createMockLogCollector().append({
level: 'info',
message: 'hello',
});

expect(logSpy).not.toHaveBeenCalled();

process.env.SENTRIS_DEBUG_TESTS = '1';

createMockTrace().record({
type: 'NODE_COMPLETED',
nodeRef: 'node-2',
message: 'finished',
});
await createMockLogCollector().append({
level: 'warn',
message: 'careful',
});

expect(logSpy).toHaveBeenCalledWith('TRACE:', 'NODE_COMPLETED', 'node-2', 'finished');
expect(logSpy).toHaveBeenCalledWith('LOG:', 'warn', 'careful');
} finally {
logSpy.mockRestore();
if (previousDebugValue === undefined) {
delete process.env.SENTRIS_DEBUG_TESTS;
} else {
process.env.SENTRIS_DEBUG_TESTS = previousDebugValue;
}
}
});
});
12 changes: 10 additions & 2 deletions worker/src/testing/test-utils.ts
Original file line number Diff line number Diff line change
@@ -1,6 +1,10 @@
import { vi } from 'bun:test';
import { ExecutionContext, ISecretsService } from '@sentris/component-sdk';

function shouldMirrorTestDiagnostics(): boolean {
return process.env.SENTRIS_DEBUG_TESTS === '1';
}

export function createMockExecutionContext(
overrides: Partial<ExecutionContext> = {},
): ExecutionContext {
Expand Down Expand Up @@ -49,7 +53,9 @@ export function createMockTrace(): any {
return {
record: vi.fn().mockImplementation((event) => {
events.push(event);
console.log('TRACE:', event.type, event.nodeRef, event.message);
if (shouldMirrorTestDiagnostics()) {
console.log('TRACE:', event.type, event.nodeRef, event.message);
}
}),
flush: vi.fn().mockResolvedValue(undefined),
setRunMetadata: vi.fn(),
Expand All @@ -63,7 +69,9 @@ export function createMockLogCollector(): any {
return {
append: vi.fn().mockImplementation((log) => {
logs.push(log);
console.log('LOG:', log.level, log.message);
if (shouldMirrorTestDiagnostics()) {
console.log('LOG:', log.level, log.message);
}
return Promise.resolve();
}),
flush: vi.fn().mockResolvedValue(undefined),
Expand Down
Loading