lib/generate_patients.rb in cqm-models-3.0.0 vs lib/generate_patients.rb in cqm-models-3.0.1

@@ -1,22 +1,24 @@
 require('nokogiri')
 require_relative '../app/models/models'
 require('generate_types')
+require('generate_entities')
 
 module QDM
   # PatientGeneration module can be used to generate patients with dataElements that are populated
   # with every possible attribute that said type supports
   module PatientGeneration
     # Generates patient(s) with fully-loaded dataElements if new_patient_for_each_type is false then a
     # single patient will be returned that has every data element on it
-    def self.generate_exhastive_data_element_patients(new_patient_for_each_type = true, model_info_file = 'qdm-modelinfo-5.5.xml')
+    def self.generate_exhaustive_data_element_patients(new_patient_for_each_type = true, model_info_file = 'qdm-modelinfo-5.5.xml')
       datatypes = get_datatypes(File.join(File.dirname(__FILE__), "../modelinfo/#{model_info_file}"))
       patients = []
       cqm_patient = nil
       qdm_patient = nil
       datatypes.each do |type|
         next if type.include? 'PatientCharacteristic'
+
         # 1 Patient Per data element type containing negated and non-negated type (if negatable)
         if cqm_patient.nil? || qdm_patient.nil? || new_patient_for_each_type
           cqm_patient = QDM::BaseTypeGeneration.generate_cqm_patient(type)
           qdm_patient = QDM::BaseTypeGeneration.generate_qdm_patient
           # Add patient characteristics
@@ -31,11 +33,11 @@
           cqm_patient.qdmPatient = qdm_patient
         end
         data_element = generate_loaded_datatype(type)
         qdm_patient.dataElements.push(data_element)
         # if type is negatable, add a negated version to the patient
-        if data_element.fields.keys.include? 'negationRationale'
+        if data_element.fields.key?('negationRationale')
           negated_data_element = generate_loaded_datatype(type, true)
           qdm_patient.dataElements.push(negated_data_element)
         end
         patients.push(cqm_patient)
       end
@@ -51,10 +53,11 @@
         datatype_name = type.attributes['name'].value.split('.').last
         exclusion_array = %w[Component Identifier Patient ResultComponent FacilityLocation]
         # Store datatype and its attributes (reject irrelevant datatypes)
         next if datatype_name.include?('Negative') || datatype_name.include?('Positive') ||
                 datatype_name.include?('QDMBaseType') || (exclusion_array.include? datatype_name)
+
         datatypes.push(datatype_name)
       end
       datatypes
     end
 
@@ -62,14 +65,15 @@
       data_element = QDM.const_get(data_element_type).new
       fields = data_element.typed_attributes.keys
       fields.each do |field_name|
         # Ignore these fields, they are used by mongo
         next if %w[_id _type].include? field_name
-        if !data_element[field_name] || data_element[field_name] == []
-          populate_fields(field_name, data_element, negate_data_element)
-        end
+
+        populate_fields(field_name, data_element, negate_data_element) if !data_element[field_name] || data_element[field_name] == []
       end
+      QDM::EntityGeneration.generate_entities(data_element)
+      data_element.identifier = QDM::BaseTypeGeneration.generate_qdm_id if data_element.respond_to? 'identifier'
       data_element
     end
 
     def self.populate_fields(field_name, data_element, negate_data_element)
       # There are certain fields that we want to populate manually
@@ -79,20 +83,22 @@
         # Skip the setting code list id. It is not used in a patient's data_element, only in the measure's
       elsif field_name == 'negationRationale'
         data_element[field_name] = QDM::BaseTypeGeneration.generate_code_field if negate_data_element
       elsif field_name == 'components'
         data_element[field_name] = [QDM::BaseTypeGeneration.generate_component]
-      elsif field_name == 'result'
-        # TODO: Result can be MANY Integer, Decimal, Code, Quantity or Ratio randomize this
-        data_element[field_name] = QDM::BaseTypeGeneration.generate_code_field
-      elsif %w[diagnoses dataElementCodes].include? field_name
+      elsif field_name == 'dataElementCodes'
         # TODO: Populate dataElementCodes with codes specifically for data element type
+        # might need to build a large map of data element type to relevent codes
+        # should also be done for <data element type>.attribute -> relevant code for the coded attributes
         data_element[field_name] = [QDM::BaseTypeGeneration.generate_code_field]
+      elsif field_name == 'diagnoses'
+        data_element[field_name] = [QDM::BaseTypeGeneration.generate_diagnosis]
+      elsif field_name == 'result'
+        data_element[field_name] = QDM::BaseTypeGeneration.generate_result
       elsif field_name == 'facilityLocations'
-        # TODO: Randomize number of facility locations added
-        data_element[field_name] = [QDM::BaseTypeGeneration.generate_facility_location]
+        # create up to 5 facility locations
+        data_element[field_name] = (0..rand(5)).map { QDM::BaseTypeGeneration.generate_facility_location }
       elsif field_name == 'facilityLocation'
-        # TODO: Randomize number of facility locations added
         data_element[field_name] = QDM::BaseTypeGeneration.generate_facility_location
       elsif field_name == 'targetOutcome'
         # TODO: Randomize type of targetOutcome, use code for now
         data_element[field_name] = QDM::BaseTypeGeneration.generate_code_field
       else