This commit is contained in:
dal 2025-07-21 16:27:15 -06:00
parent d53826a2f1
commit cf2ec2b02a
No known key found for this signature in database
GPG Key ID: 16F4B0E1E9F61122
3 changed files with 25 additions and 5 deletions

View File

@ -60,12 +60,25 @@ vi.mock('@trigger.dev/sdk/v3', () => ({
})),
}));
// Mock Braintrust
vi.mock('braintrust', () => ({
initLogger: vi.fn(() => ({
flush: vi.fn().mockResolvedValue(undefined),
})),
currentSpan: vi.fn(() => ({
log: vi.fn(),
})),
wrapTraced: vi.fn((fn) => fn),
}));
describe('messagePostProcessingTask', () => {
let mockDb: any;
let mockWorkflowRun: any;
beforeEach(() => {
vi.clearAllMocks();
// Mock BRAINTRUST_KEY for unit tests
vi.stubEnv('BRAINTRUST_KEY', 'test-braintrust-key');
mockDb = {
update: vi.fn().mockReturnThis(),
set: vi.fn().mockReturnThis(),
@ -234,6 +247,8 @@ describe('messagePostProcessingTask', () => {
sent: true,
messageTs: 'msg-ts-456',
threadTs: 'thread-ts-456',
integrationId: 'int-123',
channelId: 'C123456',
});
vi.mocked(helpers.buildWorkflowInput).mockReturnValue({
conversationHistory: undefined,
@ -290,6 +305,8 @@ describe('messagePostProcessingTask', () => {
sent: true,
messageTs: 'msg-ts-123',
threadTs: 'thread-ts-123',
integrationId: 'int-123',
channelId: 'C123456',
});
vi.mocked(helpers.buildWorkflowInput).mockReturnValue({
conversationHistory: undefined,
@ -365,6 +382,8 @@ describe('messagePostProcessingTask', () => {
sent: true,
messageTs: 'msg-ts-123',
threadTs: 'thread-ts-123',
integrationId: 'int-123',
channelId: 'C123456',
});
vi.mocked(helpers.buildWorkflowInput).mockReturnValue({
conversationHistory: undefined,

View File

@ -1,2 +1,3 @@
dist*
node_modules*
node_modules*
/dist/

View File

@ -75,7 +75,7 @@ importers:
specifier: ^8.0.0
version: 8.0.0
typescript:
specifier: ^5
specifier: ^5.8.3
version: 5.8.3
devDependencies:
'@biomejs/biome':
@ -17439,14 +17439,14 @@ snapshots:
msw: 2.10.4(@types/node@20.19.4)(typescript@5.8.3)
vite: 6.3.5(@types/node@20.19.4)(jiti@2.4.2)(lightningcss@1.30.1)(sass@1.89.2)(terser@5.43.1)(tsx@4.20.3)(yaml@2.8.0)
'@vitest/mocker@3.2.4(msw@2.10.4(@types/node@24.0.10)(typescript@5.8.3))(vite@6.3.5(@types/node@24.0.10)(jiti@2.4.2)(lightningcss@1.30.1)(sass@1.89.2)(terser@5.43.1)(tsx@4.20.3)(yaml@2.8.0))':
'@vitest/mocker@3.2.4(msw@2.10.4(@types/node@24.0.10)(typescript@5.8.3))(vite@6.3.5(@types/node@20.19.4)(jiti@2.4.2)(lightningcss@1.30.1)(sass@1.89.2)(terser@5.43.1)(tsx@4.20.3)(yaml@2.8.0))':
dependencies:
'@vitest/spy': 3.2.4
estree-walker: 3.0.3
magic-string: 0.30.17
optionalDependencies:
msw: 2.10.4(@types/node@24.0.10)(typescript@5.8.3)
vite: 6.3.5(@types/node@24.0.10)(jiti@2.4.2)(lightningcss@1.30.1)(sass@1.89.2)(terser@5.43.1)(tsx@4.20.3)(yaml@2.8.0)
vite: 6.3.5(@types/node@20.19.4)(jiti@2.4.2)(lightningcss@1.30.1)(sass@1.89.2)(terser@5.43.1)(tsx@4.20.3)(yaml@2.8.0)
'@vitest/pretty-format@2.0.5':
dependencies:
@ -24159,7 +24159,7 @@ snapshots:
dependencies:
'@types/chai': 5.2.2
'@vitest/expect': 3.2.4
'@vitest/mocker': 3.2.4(msw@2.10.4(@types/node@24.0.10)(typescript@5.8.3))(vite@6.3.5(@types/node@24.0.10)(jiti@2.4.2)(lightningcss@1.30.1)(sass@1.89.2)(terser@5.43.1)(tsx@4.20.3)(yaml@2.8.0))
'@vitest/mocker': 3.2.4(msw@2.10.4(@types/node@24.0.10)(typescript@5.8.3))(vite@6.3.5(@types/node@20.19.4)(jiti@2.4.2)(lightningcss@1.30.1)(sass@1.89.2)(terser@5.43.1)(tsx@4.20.3)(yaml@2.8.0))
'@vitest/pretty-format': 3.2.4
'@vitest/runner': 3.2.4
'@vitest/snapshot': 3.2.4