Medplum

// SPDX-FileCopyrightText: Copyright Orangebot, Inc. and Medplum contributors // SPDX-License-Identifier: Apache-2.0 import type { AgentError, AgentTransmitResponse, AgentUpgradeRequest, AgentUpgradeResponse, WithId, } from '@medplum/core'; import { ContentType, allOk, badRequest, serverError } from '@medplum/core'; import type { Agent, Bundle, OperationOutcome, OperationOutcomeIssue, Parameters } 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, expectBundleToContainOutcome, mockAgentResponse, } from './utils/agenttestutils'; const NUM_DEFAULT_AGENTS = 2; describe('Agent/$upgrade', () => { 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', 8516, () => { 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('Upgrade all agents', async () => { const handlePromises = [] as Promise<MockAgentResponseHandle>[]; for (let i = 0; i < agents.length; i++) { handlePromises[i] = mockAgentResponse<AgentUpgradeRequest, AgentUpgradeResponse>( agents[i], accessToken, 'agent:upgrade:request', { type: 'agent:upgrade:response', statusCode: 200 } ); } const handles = await Promise.all(handlePromises); const res = await request(app) .get('/fhir/R4/Agent/$upgrade') .set('Authorization', 'Bearer ' + accessToken); expect(res.status).toBe(200); const bundle = res.body as Bundle<Parameters>; for (const agent of agents) { expectBundleToContainOutcome(bundle, agent, allOk); } for (const handle of handles) { handle.cleanup(); } }); test('Upgrade all agents -- set timeout', async () => { const handlePromises = [] as Promise<MockAgentResponseHandle>[]; for (let i = 0; i < agents.length; i++) { handlePromises[i] = mockAgentResponse<AgentUpgradeRequest, AgentUpgradeResponse>( agents[i], accessToken, 'agent:upgrade:request', { type: 'agent:upgrade:response', statusCode: 200 } ); } const handles = await Promise.all(handlePromises); const res = await request(app) .get('/fhir/R4/Agent/$upgrade?timeout=1000') .set('Authorization', 'Bearer ' + accessToken); expect(res.status).toBe(200); const bundle = res.body as Bundle<Parameters>; for (const agent of agents) { expectBundleToContainOutcome(bundle, agent, allOk); } for (const handle of handles) { handle.cleanup(); } }); test('Upgrade all agents -- set timeout greater than max', async () => { const handlePromises = [] as Promise<MockAgentResponseHandle>[]; for (let i = 0; i < agents.length; i++) { handlePromises[i] = mockAgentResponse<AgentUpgradeRequest, AgentUpgradeResponse>( agents[i], accessToken, 'agent:upgrade:request', { type: 'agent:upgrade:response', statusCode: 200 } ); } const handles = await Promise.all(handlePromises); const res = await request(app) .get('/fhir/R4/Agent/$upgrade?timeout=100000') .set('Authorization', 'Bearer ' + accessToken); expect(res.status).toBe(200); const bundle = res.body as Bundle<Parameters>; for (const agent of agents) { expectBundleToContainOutcome(bundle, agent, allOk); } for (const handle of handles) { handle.cleanup(); } }); test('Upgrade agent by ID', async () => { const { cleanup } = await mockAgentResponse<AgentUpgradeRequest, AgentUpgradeResponse>( agents[0], accessToken, 'agent:upgrade:request', { type: 'agent:upgrade:response', statusCode: 200 } ); const res = await request(app) .get(`/fhir/R4/Agent/${agents[0].id}/$upgrade`) .set('Authorization', 'Bearer ' + accessToken); expect(res.status).toBe(200); const outcome = res.body as OperationOutcome; expect(outcome).toMatchObject<OperationOutcome>(allOk); cleanup(); }); test('Upgrade all agents -- invalid timeout', async () => { const handlePromises = [] as Promise<MockAgentResponseHandle>[]; for (let i = 0; i < agents.length; i++) { handlePromises[i] = mockAgentResponse<AgentUpgradeRequest, AgentUpgradeResponse>( agents[i], accessToken, 'agent:upgrade:request', { type: 'agent:upgrade:response', statusCode: 200 } ); } const handles = await Promise.all(handlePromises); const res = await request(app) .get('/fhir/R4/Agent/$upgrade?timeout=INVALID') .set('Authorization', 'Bearer ' + accessToken); expect(res.status).toBe(400); expect(res.body).toMatchObject<Partial<OperationOutcome>>({ resourceType: 'OperationOutcome', issue: expect.arrayContaining<OperationOutcomeIssue>([ { severity: 'error', code: 'invalid', details: { text: `Invalid value 'INVALID' provided for integer parameter 'timeout'` }, }, ]), }); for (const handle of handles) { handle.cleanup(); } }); test('Agent error during upgrade', async () => { // Multi agent example const handlePromises = [] as Promise<MockAgentResponseHandle>[]; for (let i = 0; i < agents.length; i++) { handlePromises[i] = mockAgentResponse<AgentUpgradeRequest, AgentUpgradeResponse | AgentError>( agents[i], accessToken, 'agent:upgrade:request', i === 0 ? { type: 'agent:error', body: 'Something is broken' } : { type: 'agent:upgrade:response', statusCode: 200 } ); } const handles = await Promise.all(handlePromises); let res = await request(app) .get('/fhir/R4/Agent/$upgrade') .set('Authorization', 'Bearer ' + accessToken); expect(res.status).toBe(200); const bundle = res.body as Bundle<Parameters>; for (let i = 0; i < agents.length; i++) { if (i === 0) { expectBundleToContainOutcome(bundle, agents[0], badRequest('Something is broken')); continue; } expectBundleToContainOutcome(bundle, agents[i], allOk); } // Agent by ID res = await request(app) .get(`/fhir/R4/Agent/${agents[0].id}/$upgrade`) .set('Authorization', 'Bearer ' + accessToken); expect(res.status).toBe(400); const outcome = res.body as OperationOutcome; expect(outcome).toMatchObject(badRequest('Something is broken')); for (const handle of handles) { handle.cleanup(); } }); test('Invalid response from agent upgrade', async () => { // Multi agent example const handlePromises = [] as Promise<MockAgentResponseHandle>[]; for (let i = 0; i < agents.length; i++) { handlePromises[i] = mockAgentResponse<AgentUpgradeRequest, AgentUpgradeResponse | AgentTransmitResponse>( agents[i], accessToken, 'agent:upgrade:request', i === 0 ? { type: 'agent:transmit:response', remote: '8.8.8.8', contentType: ContentType.PING, body: 'PING' } : { type: 'agent:upgrade:response', statusCode: 200 } ); } const handles = await Promise.all(handlePromises); let res = await request(app) .get('/fhir/R4/Agent/$upgrade') .set('Authorization', 'Bearer ' + accessToken); expect(res.status).toBe(200); const bundle = res.body as Bundle<Parameters>; for (let i = 0; i < agents.length; i++) { if (i === 0) { expectBundleToContainOutcome(bundle, agents[0], serverError(new Error('Invalid response received from agent'))); continue; } expectBundleToContainOutcome(bundle, agents[i], allOk); } // Single agent by ID res = await request(app) .get(`/fhir/R4/Agent/${agents[0].id}/$upgrade`) .set('Authorization', 'Bearer ' + accessToken); expect(res.status).toBe(500); const outcome = res.body as OperationOutcome; expect(outcome).toMatchObject<OperationOutcome>(serverError(new Error('Invalid response received from agent'))); for (const handle of handles) { handle.cleanup(); } }); test('Agent force upgrade', async () => { // Multi agent example const handlePromises = [] as Promise<MockAgentResponseHandle>[]; for (let i = 0; i < agents.length; i++) { handlePromises[i] = mockAgentResponse<AgentUpgradeRequest, AgentUpgradeResponse | AgentError>( agents[i], accessToken, 'agent:upgrade:request', { type: 'agent:upgrade:response', statusCode: 200 } ); } const handles = await Promise.all(handlePromises); let res = await request(app) .get('/fhir/R4/Agent/$upgrade') .query({ force: true }) .set('Authorization', 'Bearer ' + accessToken); expect(res.status).toBe(200); const bundle = res.body as Bundle<Parameters>; for (const agent of agents) { expectBundleToContainOutcome(bundle, agent, allOk); } // Agent by ID res = await request(app) .get(`/fhir/R4/Agent/${agents[0].id}/$upgrade`) .query({ force: true }) .set('Authorization', 'Bearer ' + accessToken); expect(res.status).toBe(200); const outcome = res.body as OperationOutcome; expect(outcome).toMatchObject(allOk); for (const handle of handles) { handle.cleanup(); } }); });