Medplum

// SPDX-FileCopyrightText: Copyright Orangebot, Inc. and Medplum contributors // SPDX-License-Identifier: Apache-2.0 import { ContentType } from '@medplum/core'; import type { Agent, OperationOutcome, OperationOutcomeIssue, Parameters } from '@medplum/fhirtypes'; import { randomUUID } from 'crypto'; import express from 'express'; import request from 'supertest'; import type { AgentInfo } from '../../agent/utils'; import { AgentConnectionState } from '../../agent/utils'; import { initApp, shutdownApp } from '../../app'; import { loadTestConfig } from '../../config/loader'; import { getRedis } from '../../redis'; import { initTestAuth } from '../../test.setup'; const app = express(); let accessToken: string; let agent: Agent; describe('Agent Status', () => { beforeAll(async () => { const config = await loadTestConfig(); await initApp(app, config); accessToken = await initTestAuth(); const res1 = await 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', status: 'active', }); expect(res1.status).toBe(201); agent = res1.body as Agent; }); afterAll(async () => { await shutdownApp(); }); test('Get agent status', async () => { const res1 = await request(app) .get(`/fhir/R4/Agent/${agent.id}/$status`) .set('Authorization', 'Bearer ' + accessToken); expect(res1.status).toBe(200); const parameters1 = res1.body as Parameters; expect(parameters1.resourceType).toBe('Parameters'); expect(parameters1.parameter).toHaveLength(2); expect(parameters1.parameter?.find((p) => p.name === 'status')?.valueCode).toBe(AgentConnectionState.UNKNOWN); expect(parameters1.parameter?.find((p) => p.name === 'version')?.valueString).toBe('unknown'); // Emulate a connection await getRedis().set( `medplum:agent:${agent.id}:info`, JSON.stringify({ status: AgentConnectionState.CONNECTED, version: '3.1.4', lastUpdated: new Date().toISOString(), }), 'EX', 60 ); const res2 = await request(app) .get(`/fhir/R4/Agent/${agent.id}/$status`) .set('Authorization', 'Bearer ' + accessToken); expect(res2.status).toBe(200); const parameters2 = res2.body as Parameters; expect(parameters2.resourceType).toBe('Parameters'); expect(parameters2.parameter).toHaveLength(3); expect(parameters2.parameter?.find((p) => p.name === 'status')?.valueCode).toBe(AgentConnectionState.CONNECTED); expect(parameters2.parameter?.find((p) => p.name === 'version')?.valueString).toBe('3.1.4'); expect(parameters2.parameter?.find((p) => p.name === 'lastUpdated')?.valueInstant).toBeTruthy(); }); test('Get agent status by ID -- invalid AgentInfo from Redis', async () => { await getRedis().set( `medplum:agent:${agent.id}:info`, JSON.stringify({ version: '3.1.4', lastUpdated: new Date().toISOString(), }), 'EX', 60 ); const res = await request(app) .get(`/fhir/R4/Agent/${agent.id}/$status`) .set('Authorization', 'Bearer ' + accessToken); expect(res.status).toBe(400); expect(res.body).toMatchObject<OperationOutcome>({ resourceType: 'OperationOutcome', issue: [ expect.objectContaining({ severity: 'error', code: 'invalid', details: { text: expect.stringContaining('Invalid agent info: ') }, }), ], }); await getRedis().set( `medplum:agent:${agent.id}:info`, JSON.stringify({ status: AgentConnectionState.UNKNOWN, version: 'unknown', lastUpdated: new Date().toISOString(), } satisfies AgentInfo), 'EX', 60 ); }); test('Get agent statuses -- no ID or identifier', async () => { const res = await request(app) .get('/fhir/R4/Agent/$status') .set('Authorization', 'Bearer ' + accessToken) .set('Authorization', 'Bearer ' + accessToken); expect(res.status).toBe(400); expect(res.body).toMatchObject<OperationOutcome>({ resourceType: 'OperationOutcome', issue: expect.arrayContaining<OperationOutcomeIssue>([ expect.objectContaining<OperationOutcomeIssue>({ severity: 'error', code: 'invalid', details: { text: 'Must specify agent ID or identifier' }, }), ]), }); }); });