Merge pull request #81 from mcode/condition-context-util

Conditions from Context Util

Author: jafeltra

src/helpers/contextUtils.js

@@ -1,5 +1,5 @@
 const logger = require('./logger');
-const { getBundleResourcesByType } = require('./fhirUtils');
+const { getBundleEntriesByResourceType, getBundleResourcesByType } = require('./fhirUtils');
 
 function getPatientFromContext(mrn, context) {
   logger.debug('Getting patient from context');
@@ -11,6 +11,17 @@ function getPatientFromContext(mrn, context) {
   return patientInContext;
 }
 
+function getConditionEntriesFromContext(mrn, context) {
+  logger.debug('Getting conditions from context');
+  const conditionsInContext = getBundleEntriesByResourceType(context, 'Condition', {}, false);
+  if (conditionsInContext.length === 0) {
+    throw Error('Could not find any conditions in context; ensure that a ConditionExtractor is used earlier in your extraction configuration');
+  }
+  logger.debug(`Condition resources found in context. Found ${conditionsInContext.length} condition resources.`);
+  return conditionsInContext;
+}
+
 module.exports = {
+  getConditionEntriesFromContext,
   getPatientFromContext,
 };

src/index.js

@@ -62,7 +62,7 @@ const {
 } = require('./helpers/conditionUtils');
 const { getDiseaseStatusCode, getDiseaseStatusEvidenceCode, mEpochToDate } = require('./helpers/diseaseStatusUtils');
 const { formatDate, formatDateTime } = require('./helpers/dateUtils');
-const { getPatientFromContext } = require('./helpers/contextUtils');
+const { getConditionEntriesFromContext, getPatientFromContext } = require('./helpers/contextUtils');
 
 module.exports = {
   // CLI Related utilities
@@ -129,5 +129,6 @@ module.exports = {
   logOperationOutcomeInfo,
   mEpochToDate,
   // Context operations
+  getConditionEntriesFromContext,
   getPatientFromContext,
 };

test/helpers/contextUtils.test.js

@@ -1,4 +1,4 @@
-const { getPatientFromContext } = require('../../src/helpers/contextUtils');
+const { getConditionEntriesFromContext, getPatientFromContext } = require('../../src/helpers/contextUtils');
 
 const MOCK_PATIENT_MRN = '123';
 
@@ -26,3 +26,38 @@ describe('getPatientFromContext', () => {
     expect(() => getPatientFromContext(MOCK_PATIENT_MRN, {})).toThrow('Could not find a patient in context; ensure that a PatientExtractor is used earlier in your extraction configuration');
   });
 });
+
+describe('getConditionFromContext', () => {
+  const conditionResource = {
+    resourceType: 'Condition',
+    id: 'mCODEConditionExample01',
+  };
+  const conditionContext = {
+    resourceType: 'Bundle',
+    type: 'collection',
+    entry: [
+      {
+        fullUrl: 'context-url-1',
+        resource: conditionResource,
+      },
+      {
+        fullUrl: 'context-url-2',
+        resource: { ...conditionResource, id: 'mCODEConditionExample02' },
+      },
+    ],
+  };
+
+  test('Should return all Condition resources in context', () => {
+    const conditions = getConditionEntriesFromContext(MOCK_PATIENT_MRN, conditionContext);
+    expect(conditions).toHaveLength(2);
+    expect(conditions[0]).toEqual({
+      fullUrl: 'context-url-1',
+      resource: conditionResource,
+    });
+  });
+
+  test('Should throw an error if there are no conditions in context', () => {
+    expect(() => getConditionEntriesFromContext(MOCK_PATIENT_MRN, {}))
+      .toThrow('Could not find any conditions in context; ensure that a ConditionExtractor is used earlier in your extraction configuration');
+  });
+});