Medplum

// SPDX-FileCopyrightText: Copyright Orangebot, Inc. and Medplum contributors // SPDX-License-Identifier: Apache-2.0 import type { AgentLogsRequest, AgentLogsResponse, LogMessage, WithId } from '@medplum/core'; import { badRequest, ContentType } from '@medplum/core'; import type { Agent, Bundle, BundleEntry, OperationOutcome, Parameters, ParametersParameter } from '@medplum/fhirtypes'; import express from 'express'; import { randomUUID } from 'node:crypto'; import type { Server } from 'node:http'; import type { AddressInfo } from 'node:net'; import type { Response } from 'supertest'; import request from 'supertest'; import { initApp, shutdownApp } from '../../app'; import { loadTestConfig } from '../../config/loader'; import { initTestAuth } from '../../test.setup'; import type { MockAgentResponseHandle } from './utils/agenttestutils'; import { cleanupMockAgents, configMockAgents, mockAgentResponse } from './utils/agenttestutils'; const NUM_DEFAULT_AGENTS = 2; describe('Agent/$fetch-logs', () => { const app = express(); const agents = [] as WithId<Agent>[]; let server: Server; let port: number; let accessToken: string; beforeAll(async () => { const config = await loadTestConfig(); server = await initApp(app, config); accessToken = await initTestAuth({ membership: { admin: true } }); await new Promise<void>((resolve) => { server.listen(0, 'localhost', 8513, () => { port = (server.address() as AddressInfo).port; resolve(); }); }); const promises = Array.from({ length: NUM_DEFAULT_AGENTS }) as Promise<Response>[]; for (let i = 0; i < NUM_DEFAULT_AGENTS; i++) { promises[i] = request(app) .post('/fhir/R4/Agent') .set('Content-Type', ContentType.FHIR_JSON) .set('Authorization', 'Bearer ' + accessToken) .send({ resourceType: 'Agent', identifier: [{ system: 'https://example.com/agent', value: randomUUID() }], name: `Test Agent ${i + 1}`, status: 'active', }); } const responses = await Promise.all(promises); for (let i = 0; i < NUM_DEFAULT_AGENTS; i++) { expect(responses[i].status).toBe(201); agents[i] = responses[i].body; } configMockAgents(port); }); afterAll(async () => { cleanupMockAgents(); await shutdownApp(); }); test('Fetch logs for all agents', async () => { const handlePromises = [] as Promise<MockAgentResponseHandle>[]; const logs: LogMessage[] = [ { level: 'INFO', timestamp: new Date().toISOString(), msg: 'Test 1' }, { level: 'INFO', timestamp: new Date().toISOString(), msg: 'Test 2' }, { level: 'ERROR', timestamp: new Date().toISOString(), msg: 'An error occurred', error: new Error('This is an error').toString(), }, ]; for (let i = 0; i < agents.length; i++) { handlePromises[i] = mockAgentResponse<AgentLogsRequest, AgentLogsResponse>( agents[i], accessToken, 'agent:logs:request', { type: 'agent:logs:response', statusCode: 200, logs, } ); } const handles = await Promise.all(handlePromises); const res = await request(app) .get('/fhir/R4/Agent/$fetch-logs') .set('Authorization', 'Bearer ' + accessToken); expect(res.status).toBe(200); const bundle = res.body as Bundle<Parameters>; for (const agent of agents) { expectBundleToContainLogsEntry(bundle, agent, logs); } for (const handle of handles) { handle.cleanup(); } }); test('Fetch logs for Agent by ID', async () => { const logs: LogMessage[] = [ { level: 'INFO', timestamp: new Date().toISOString(), msg: 'Test 1' }, { level: 'INFO', timestamp: new Date().toISOString(), msg: 'Test 2' }, { level: 'ERROR', timestamp: new Date().toISOString(), msg: 'An error occurred', error: new Error('This is an error').toString(), }, ]; const { cleanup } = await mockAgentResponse<AgentLogsRequest, AgentLogsResponse>( agents[0], accessToken, 'agent:logs:request', { type: 'agent:logs:response', statusCode: 200, logs } ); const res = await request(app) .get(`/fhir/R4/Agent/${agents[0].id}/$fetch-logs`) .set('Authorization', 'Bearer ' + accessToken); expect(res.status).toBe(200); const params = res.body as Parameters; expect(params).toMatchObject<Parameters>({ resourceType: 'Parameters', parameter: expect.arrayContaining<ParametersParameter>([ expect.objectContaining<ParametersParameter>({ name: 'logs', valueString: logs.map((msg) => JSON.stringify(msg)).join('\n'), }), ]), }); cleanup(); }); test('Fetch logs -- non-integer limit', async () => { const res = await request(app) .get(`/fhir/R4/Agent/${agents[0].id}/$fetch-logs`) .query({ limit: 'true' }) .set('Authorization', 'Bearer ' + accessToken); expect(res.status).toBe(400); const outcome = res.body as OperationOutcome; expect(outcome).toMatchObject<OperationOutcome>( badRequest("Invalid value 'true' provided for integer parameter 'limit'") ); }); }); function expectBundleToContainLogsEntry(bundle: Bundle<Parameters>, agent: Agent, logs: LogMessage[]): void { const entries = bundle.entry as BundleEntry<Parameters>[]; expect(entries).toContainEqual({ resource: expect.objectContaining<Parameters>({ resourceType: 'Parameters', parameter: expect.arrayContaining<ParametersParameter>([ expect.objectContaining<ParametersParameter>({ name: 'agent', resource: expect.objectContaining<Agent>(agent), }), expect.objectContaining<ParametersParameter>({ name: 'result', resource: expect.objectContaining<Parameters>({ resourceType: 'Parameters', parameter: expect.arrayContaining<ParametersParameter>([ expect.objectContaining<ParametersParameter>({ name: 'logs', valueString: logs.map((msg) => JSON.stringify(msg)).join('\n'), }), ]), }), }), ]), }), }); }