us_core_test_kit 0.1.0

data/lib/us_core_test_kit/generator/provenance_revinclude_search_test_generator.rb

@@ -0,0 +1,186 @@
+require_relative 'naming'
+require_relative 'special_cases'
+
+module USCoreTestKit
+  class Generator
+    class ProvenanceRevincludeSearchTestGenerator
+      class << self
+        def generate(ig_metadata)
+          ig_metadata.groups
+            .reject { |group| SpecialCases.exclude_resource? group.resource }
+            .select { |group| group.revincludes.include? 'Provenance:target' }
+            .each { |group| new(group, group.searches.first).generate }
+        end
+      end
+
+      attr_accessor :group_metadata, :search_metadata
+
+      def initialize(group_metadata, search_metadata)
+        self.group_metadata = group_metadata
+        self.search_metadata = search_metadata
+      end
+
+      def template
+        @template ||= File.read(File.join(__dir__, 'templates', 'provenance_revinclude_search.rb.erb'))
+      end
+
+      def output
+        @output ||= ERB.new(template).result(binding)
+      end
+
+      def base_output_file_name
+        "#{class_name.underscore}.rb"
+      end
+
+      def output_file_directory
+        File.join(__dir__, '..', 'generated', profile_identifier)
+      end
+
+      def output_file_name
+        File.join(output_file_directory, base_output_file_name)
+      end
+
+      def profile_identifier
+        Naming.snake_case_for_profile(group_metadata)
+      end
+
+      def test_id
+        "us_core_311_#{profile_identifier}_#{search_identifier}_search_test"
+      end
+
+      def search_identifier
+        'provenance_revinclude'
+      end
+
+      def search_title
+        search_identifier.camelize
+      end
+
+      def class_name
+        "#{Naming.upper_camel_case_for_profile(group_metadata)}#{search_title}SearchTest"
+      end
+
+      def resource_type
+        group_metadata.resource
+      end
+
+      def search_params
+        @search_params ||=
+          search_metadata[:names].map do |name|
+            {
+              name: name,
+              path: search_definition(name)[:path]
+            }
+          end
+      end
+
+      def first_search?
+        group_metadata.searches.first == search_metadata
+      end
+
+      def fixed_value_search?
+        search_metadata[:names] != ['patient'] &&
+          !group_metadata.delayed? && resource_type != 'Patient'
+      end
+
+      def fixed_value_search_param_name
+        (search_metadata[:names] - [:patient]).first
+      end
+
+      def search_param_name_string
+        search_metadata[:names].join(' + ') + ' + revInclude:Provenance:target'
+      end
+
+      def needs_patient_id?
+        search_metadata[:names].include?('patient') ||
+          (resource_type == 'Patient' && search_metadata[:names].include?('_id'))
+      end
+
+      def search_param_names
+        search_params.map { |param| param[:name] }
+      end
+
+      def search_param_names_array
+        array_of_strings(search_param_names)
+      end
+
+      def path_for_value(path)
+        path == 'class' ? 'local_class' : path
+      end
+
+      def required_comparators_for_param(name)
+        search_definition(name)[:comparators].select { |_comparator, expectation| expectation == 'SHALL' }
+      end
+
+      def required_comparators
+        @required_comparators ||=
+          search_param_names.each_with_object({}) do |name, comparators|
+            required_comparators = required_comparators_for_param(name)
+            comparators[name] = required_comparators if required_comparators.present?
+          end
+      end
+
+      # def patient_id_param?(param)
+      #   param[:name] == 'patient' ||
+      #     (resource_type == 'Patient' && param[:name] == '_id')
+      # end
+
+      def search_definition(name)
+        group_metadata.search_definitions[name.to_sym]
+      end
+
+      def saves_delayed_references?
+        first_search? && group_metadata.delayed_references.present?
+      end
+
+      def possible_status_search?
+        !search_metadata[:names].include?('status') && group_metadata.search_definitions.key?(:status)
+      end
+
+      def token_search_params
+        @token_search_params ||=
+          search_param_names.select do |name|
+            ['Identifier', 'CodeableConcept', 'Coding'].include? group_metadata.search_definitions[name.to_sym][:type]
+          end
+      end
+
+      def token_search_params_string
+        array_of_strings(token_search_params)
+      end
+
+      def required_comparators_string
+        array_of_strings(required_comparators.keys)
+      end
+
+      def array_of_strings(array)
+        quoted_strings = array.map { |element| "'#{element}'" }
+        "[#{quoted_strings.join(', ')}]"
+      end
+
+      def search_properties
+        {}.tap do |properties|
+          properties[:fixed_value_search] = 'true' if fixed_value_search?
+          properties[:resource_type] = "'#{resource_type}'"
+          properties[:search_param_names] = search_param_names_array
+          properties[:possible_status_search] = 'true' if possible_status_search?
+        end
+      end
+
+      def search_test_properties_string
+        search_properties
+          .map { |key, value| "#{' ' * 8}#{key}: #{value}" }
+          .join(",\n")
+      end
+
+      def generate
+        FileUtils.mkdir_p(output_file_directory)
+        File.open(output_file_name, 'w') { |f| f.write(output) }
+
+        group_metadata.add_test(
+          id: test_id,
+          file_name: base_output_file_name
+        )
+      end
+    end
+  end
+end

data/lib/us_core_test_kit/generator/read_test_generator.rb

@@ -0,0 +1,89 @@
+require_relative 'naming'
+require_relative 'special_cases'
+
+module USCoreTestKit
+  class Generator
+    class ReadTestGenerator
+      class << self
+        def generate(ig_metadata)
+          ig_metadata.groups
+            .reject { |group| SpecialCases.exclude_resource? group.resource }
+            .select { |group| read_interaction(group).present? }
+            .each { |group| new(group).generate }
+        end
+
+        def read_interaction(group_metadata)
+          group_metadata.interactions.find { |interaction| interaction[:code] == 'read' }
+        end
+      end
+
+      attr_accessor :group_metadata
+
+      def initialize(group_metadata)
+        self.group_metadata = group_metadata
+      end
+
+      def template
+        @template ||= File.read(File.join(__dir__, 'templates', 'read.rb.erb'))
+      end
+
+      def output
+        @output ||= ERB.new(template).result(binding)
+      end
+
+      def base_output_file_name
+        "#{class_name.underscore}.rb"
+      end
+
+      def output_file_directory
+        File.join(__dir__, '..', 'generated', profile_identifier)
+      end
+
+      def output_file_name
+        File.join(output_file_directory, base_output_file_name)
+      end
+
+      def read_interaction
+        self.class.read_interaction(group_metadata)
+      end
+
+      def profile_identifier
+        Naming.snake_case_for_profile(group_metadata)
+      end
+
+      def test_id
+        "us_core_311_#{profile_identifier}_read_test"
+      end
+
+      def class_name
+        "#{Naming.upper_camel_case_for_profile(group_metadata)}ReadTest"
+      end
+
+      def resource_type
+        group_metadata.resource
+      end
+
+      def resource_collection_string
+        if group_metadata.delayed? && resource_type != 'Provenance'
+          "scratch.dig(:references, '#{resource_type}')"
+        else
+          'all_scratch_resources'
+        end
+      end
+
+      def conformance_expectation
+        read_interaction[:expectation]
+      end
+
+      def generate
+        FileUtils.mkdir_p(output_file_directory)
+        File.open(output_file_name, 'w') { |f| f.write(output) }
+
+        group_metadata.add_test(
+          id: test_id,
+          file_name: base_output_file_name
+        )
+      end
+    end
+  end
+end

data/lib/us_core_test_kit/generator/reference_resolution_test_generator.rb

@@ -0,0 +1,72 @@
+require_relative 'naming'
+require_relative 'special_cases'
+
+module USCoreTestKit
+  class Generator
+    class ReferenceResolutionTestGenerator
+      class << self
+        def generate(ig_metadata)
+          ig_metadata.groups
+            .reject { |group| SpecialCases.exclude_resource? group.resource }
+            .each { |group| new(group).generate }
+        end
+      end
+
+      attr_accessor :group_metadata
+
+      def initialize(group_metadata)
+        self.group_metadata = group_metadata
+      end
+
+      def template
+        @template ||= File.read(File.join(__dir__, 'templates', 'reference_resolution.rb.erb'))
+      end
+
+      def output
+        @output ||= ERB.new(template).result(binding)
+      end
+
+      def base_output_file_name
+        "#{class_name.underscore}.rb"
+      end
+
+      def output_file_directory
+        File.join(__dir__, '..', 'generated', profile_identifier)
+      end
+
+      def output_file_name
+        File.join(output_file_directory, base_output_file_name)
+      end
+
+      def profile_identifier
+        Naming.snake_case_for_profile(group_metadata)
+      end
+
+      def test_id
+        "us_core_311_#{profile_identifier}_reference_resolution_test"
+      end
+
+      def class_name
+        "#{Naming.upper_camel_case_for_profile(group_metadata)}ReferenceResolutionTest"
+      end
+
+      def resource_type
+        group_metadata.resource
+      end
+
+      def resource_collection_string
+        'scratch_resources[:all]'
+      end
+
+      def generate
+        FileUtils.mkdir_p(output_file_directory)
+        File.open(output_file_name, 'w') { |f| f.write(output) }
+
+        group_metadata.add_test(
+          id: test_id,
+          file_name: base_output_file_name
+        )
+      end
+    end
+  end
+end

data/lib/us_core_test_kit/generator/resource_list_generator.rb

@@ -0,0 +1,46 @@
+module USCoreTestKit
+  class Generator
+    class ResourceListGenerator
+      class << self
+        def generate(ig_metadata)
+          @ig_metadata = ig_metadata
+
+          FileUtils.mkdir_p(output_file_directory)
+          File.open(output_file_name, 'w') { |f| f.write(output) }
+        end
+
+        def resource_list
+          @ig_metadata.groups.map(&:resource).uniq
+        end
+
+        def resource_list_string
+          resource_list.map { |resource| "      '#{resource}'" }.join(",\n")
+        end
+
+        def read_interaction(group_metadata)
+          group_metadata.interactions.find { |interaction| interaction[:code] == 'read' }
+        end
+
+        def template
+          @template ||= File.read(File.join(__dir__, 'templates', 'resource_list.rb.erb'))
+        end
+
+        def output
+          @output ||= ERB.new(template).result(binding)
+        end
+
+        def base_output_file_name
+          "resource_list.rb"
+        end
+
+        def output_file_directory
+          File.join(__dir__, '..', 'generated')
+        end
+
+        def output_file_name
+          File.join(output_file_directory, base_output_file_name)
+        end
+      end
+    end
+  end
+end

data/lib/us_core_test_kit/generator/search_definition_metadata_extractor.rb

@@ -0,0 +1,181 @@
+module USCoreTestKit
+  class Generator
+    class SearchDefinitionMetadataExtractor
+      attr_accessor :ig_resources, :name, :resource, :profile_elements
+
+      def initialize(name, ig_resources, resource, profile_elements)
+        self.name = name
+        self.ig_resources = ig_resources
+        self.resource = resource
+        self.profile_elements = profile_elements
+      end
+
+      def search_definition
+        @search_definition ||=
+          {
+            path: path,
+            full_path: full_path,
+            comparators: comparators,
+            values: values,
+            type: type,
+            contains_multiple: contains_multiple?,
+            multiple_or: multiple_or_expectation,
+            chain: chain
+          }.compact
+      end
+
+      def param
+        @param ||= ig_resources.search_param_by_resource_and_name(resource, name)
+      end
+
+      def param_hash
+        param.source_hash
+      end
+
+      def full_path
+        @full_path ||=
+          begin
+            path = param.expression.gsub(/.where\((.*)/, '')
+            path = path[1..-2] if path.start_with?('(') && path.end_with?(')')
+            as_type = path.scan(/[. ]as[( ]([^)]*)[)]?/).flatten.first
+            path.gsub!(/[. ]as[( ]([^)]*)[)]?/, as_type.upcase_first) if as_type.present?
+            path
+          end
+      end
+
+      def path
+        @path ||= full_path.gsub("#{resource}.", '')
+      end
+
+      def profile_element
+        @profile_element ||=
+          profile_elements.find { |element| element.id == full_path }
+      end
+
+      def comparator_expectation_extensions
+        @comparator_expectation_extensions ||= param_hash['_comparator'] || []
+      end
+
+      def support_expectation(extension)
+        extension['extension'].first['valueCode']
+      end
+
+      def comparator_expectation(extension)
+        if extension.nil?
+          'MAY'
+        else
+          support_expectation(extension)
+        end
+      end
+
+      def comparators
+        {}.tap do |comparators|
+          param.comparator&.each_with_index do |comparator, index|
+            comparators[comparator.to_sym] = comparator_expectation(comparator_expectation_extensions[index])
+          end
+        end
+      end
+
+      def type
+        if profile_element.present?
+          profile_element.type.first.code
+        else
+          # search is a variable type, eg. Condition.onsetDateTime - element
+          # in profile def is Condition.onset[x]
+          param.type
+        end
+      end
+
+      def contains_multiple?
+        profile_element&.max == '*'
+      end
+
+      def chain_extensions
+        param_hash['_chain']
+      end
+
+      def chain_expectations
+        chain_extensions.map { |extension| support_expectation(extension) }
+      end
+
+      def chain
+        return nil if param.chain.blank?
+
+        param.chain
+          .zip(chain_expectations)
+          .map { |chain, expectation| { chain: chain, expectation: expectation } }
+      end
+
+      def multiple_or_expectation
+        param_hash['_multipleOr']['extension'].first['valueCode']
+      end
+
+      def values
+        (
+          values_from_slices +
+          values_from_fixed_codes +
+          values_from_pattern_codeable_concept +
+          values_from_value_set_binding
+        ).uniq.presence || values_from_resource_metadata
+      end
+
+      def slices
+        return [] unless contains_multiple?
+
+        profile_elements.select do |element|
+          element.path == full_path &&
+            element.sliceName.present? &&
+            element.patternCodeableConcept.present?
+        end
+      end
+
+      def values_from_slices
+        slices.map do |slice|
+          slice.patternCodeableConcept.coding.first.code
+        end
+      end
+
+      def values_from_fixed_codes
+        return [] unless type == 'CodeableConcept'
+
+        profile_elements
+          .select { |element| element.path == "#{profile_element.path}.coding.code" && element.fixedCode.present? }
+          .map { |element| element.fixedCode }
+      end
+
+      def values_from_pattern_codeable_concept
+        return [] if type != 'CodeableConcept' || profile_element.patternCodeableConcept.blank?
+
+        [profile_element.patternCodeableConcept.coding.first.code]
+      end
+
+      def value_set_binding
+        profile_element&.binding
+      end
+
+      def value_set
+        ig_resources.value_set_by_url(value_set_binding&.valueSet)
+      end
+
+      def bound_systems
+        value_set&.compose&.include&.reject { |code| code.concept.nil? }
+      end
+
+      def values_from_value_set_binding
+        return [] if bound_systems.blank?
+
+        bound_systems.flat_map { |system| system.concept.map { |code| code.code } }.uniq
+      end
+
+      def fhir_metadata
+        FHIR.const_get(resource)::METADATA[path.gsub("#{resource}.", '')]
+      end
+
+      def values_from_resource_metadata
+        return [] if fhir_metadata&.dig('valid_codes').blank?
+
+        fhir_metadata['valid_codes'].values.flatten
+      end
+    end
+  end
+end

data/lib/us_core_test_kit/generator/search_metadata_extractor.rb

@@ -0,0 +1,74 @@
+require_relative 'search_definition_metadata_extractor'
+
+module USCoreTestKit
+  class Generator
+    class SearchMetadataExtractor
+      COMBO_EXTENSION_URL =
+        'http://hl7.org/fhir/StructureDefinition/capabilitystatement-search-parameter-combination'.freeze
+
+      attr_accessor :resource_capabilities, :ig_resources, :resource, :profile_elements
+
+      def initialize(resource_capabilities, ig_resources, resource, profile_elements)
+        self.resource_capabilities = resource_capabilities
+        self.ig_resources = ig_resources
+        self.resource = resource
+        self.profile_elements = profile_elements
+      end
+
+      def searches
+        @searches ||= basic_searches + combo_searches
+      end
+
+      def conformance_expectation(search_param)
+        search_param.extension.first.valueCode # TODO: fix expectation extension finding
+      end
+
+      def no_search_params?
+        resource_capabilities.searchParam.blank?
+      end
+
+      def basic_searches
+        return [] if no_search_params?
+
+        resource_capabilities.searchParam
+          .select { |search_param| ['SHALL', 'SHOULD'].include? conformance_expectation(search_param) }
+          .map do |search_param|
+            {
+              names: [search_param.name],
+              expectation: conformance_expectation(search_param)
+            }
+          end
+      end
+
+      def search_extensions
+        resource_capabilities.extension
+      end
+
+      def combo_searches
+        return [] if search_extensions.blank?
+
+        search_extensions
+          .select { |extension| extension.url == COMBO_EXTENSION_URL }
+          .select { |extension| ['SHALL', 'SHOULD'].include? conformance_expectation(extension) }
+          .map do |extension|
+            names = extension.extension.select { |param| param.valueString.present? }.map(&:valueString)
+            {
+              expectation: conformance_expectation(extension),
+              names: names
+            }
+          end
+      end
+
+      def search_param_names
+        searches.flat_map { |search| search[:names] }.uniq
+      end
+
+      def search_definitions
+        search_param_names.each_with_object({}) do |name, definitions|
+          definitions[name.to_sym] =
+            SearchDefinitionMetadataExtractor.new(name, ig_resources, resource, profile_elements).search_definition
+        end
+      end
+    end
+  end
+end