added getConditions fn

Author: Dtphelan1

src/helpers/contextUtils.js

@@ -1,6 +1,18 @@
+const _ = require('lodash');
 const logger = require('./logger');
 const { getBundleEntriesByResourceType, getBundleResourcesByType } = require('./fhirUtils');
 
+
+function getConditionsFromContext(mrn, context) {
+  logger.debug('Getting conditions from context');
+  const conditionsInContext = getBundleResourcesByType(context, 'Condition', {}, false);
+  if (_.isEmpty(conditionsInContext)) {
+    throw Error('Could not find conditions in context; ensure that a ConditionExtractor is used earlier in your extraction configuration');
+  }
+  logger.debug('Condition resources found in context.');
+  return conditionsInContext;
+}
+
 function getPatientFromContext(mrn, context) {
   logger.debug('Getting patient from context');
   const patientInContext = getBundleResourcesByType(context, 'Patient', {}, true);
@@ -38,6 +50,7 @@ function getEncountersFromContext(context) {
 
 module.exports = {
   getConditionEntriesFromContext,
+  getConditionsFromContext,
   getEncountersFromContext,
   getPatientFromContext,
 };

src/index.js

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

test/helpers/contextUtils.test.js

@@ -1,4 +1,4 @@
-const { getConditionEntriesFromContext, getEncountersFromContext, getPatientFromContext } = require('../../src/helpers/contextUtils');
+const { getConditionEntriesFromContext, getConditionsFromContext, getEncountersFromContext, getPatientFromContext } = require('../../src/helpers/contextUtils');
 
 const MOCK_PATIENT_MRN = '123';
 
@@ -27,6 +27,41 @@ describe('getPatientFromContext', () => {
   });
 });
 
+describe('getConditionsFromContext', () => {
+  const conditionResource = {
+    resourceType: 'Condition',
+    id: 'mCODEConditionExample01',
+  };
+  const conditionResource2 = {
+    resourceType: 'Condition',
+    id: 'mCODEConditionExample02',
+  };
+  const conditionContext = {
+    resourceType: 'Bundle',
+    type: 'collection',
+    entry: [
+      {
+        fullUrl: 'context-url-1',
+        resource: conditionResource,
+      },
+      {
+        fullUrl: 'context-url-2',
+        resource: conditionResource2,
+      },
+    ],
+  };
+  test('Should return Patient resource in context', () => {
+    const conditions = getConditionsFromContext(MOCK_PATIENT_MRN, conditionContext);
+    expect(conditions).toContain(conditionResource);
+    expect(conditions).toContain(conditionResource2);
+  });
+
+  test('Should throw an error if there is no patient in context', () => {
+    expect(() => getConditionsFromContext(MOCK_PATIENT_MRN, {}))
+      .toThrow('Could not find conditions in context; ensure that a ConditionExtractor is used earlier in your extraction configuration');
+  });
+});
+
 describe('getConditionFromContext', () => {
   const conditionResource = {
     resourceType: 'Condition',