us_core_test_kit 0.7.0 → 0.7.2

data/lib/us_core_test_kit/generated/v7.0.0-ballot/service_request/metadata.yml

@@ -328,20 +328,22 @@
 - :id: us_core_v700_ballot_service_request_reference_resolution_test
   :file_name: service_request_reference_resolution_test.rb
 :granular_scope_tests:
-- :id: us_core_v700_ballot_ServiceRequest_patient_granular_scope_test
-  :file_name: service_request_patient_granular_scope_test.rb
-- :id: us_core_v700_ballot_ServiceRequest__id_granular_scope_test
-  :file_name: service_request_id_granular_scope_test.rb
-- :id: us_core_v700_ballot_ServiceRequest_patient_category_authored_granular_scope_test
-  :file_name: service_request_patient_category_authored_granular_scope_test.rb
-- :id: us_core_v700_ballot_ServiceRequest_patient_category_granular_scope_test
-  :file_name: service_request_patient_category_granular_scope_test.rb
-- :id: us_core_v700_ballot_ServiceRequest_patient_code_authored_granular_scope_test
-  :file_name: service_request_patient_code_authored_granular_scope_test.rb
-- :id: us_core_v700_ballot_ServiceRequest_patient_code_granular_scope_test
-  :file_name: service_request_patient_code_granular_scope_test.rb
-- :id: us_core_v700_ballot_ServiceRequest_patient_status_granular_scope_test
-  :file_name: service_request_patient_status_granular_scope_test.rb
+- :id: us_core_v700_ballot_ServiceRequest_patient_granular_scope_search_test
+  :file_name: service_request_patient_granular_scope_search_test.rb
+- :id: us_core_v700_ballot_ServiceRequest__id_granular_scope_search_test
+  :file_name: service_request_id_granular_scope_search_test.rb
+- :id: us_core_v700_ballot_ServiceRequest_patient_category_authored_granular_scope_search_test
+  :file_name: service_request_patient_category_authored_granular_scope_search_test.rb
+- :id: us_core_v700_ballot_ServiceRequest_patient_category_granular_scope_search_test
+  :file_name: service_request_patient_category_granular_scope_search_test.rb
+- :id: us_core_v700_ballot_ServiceRequest_patient_code_authored_granular_scope_search_test
+  :file_name: service_request_patient_code_authored_granular_scope_search_test.rb
+- :id: us_core_v700_ballot_ServiceRequest_patient_code_granular_scope_search_test
+  :file_name: service_request_patient_code_granular_scope_search_test.rb
+- :id: us_core_v700_ballot_ServiceRequest_patient_status_granular_scope_search_test
+  :file_name: service_request_patient_status_granular_scope_search_test.rb
+- :id: us_core_v700_ballot_ServiceRequest_granular_scope_read_test
+  :file_name: service_request_granular_scope_read_test.rb
 :id: us_core_v700_ballot_service_request
 :file_name: service_request_group.rb
 :delayed_references:

data/lib/us_core_test_kit/generated/v7.0.0-ballot/service_request_granular_scope1_group.rb

@@ -1,10 +1,11 @@
-require_relative './granular_scope_tests/service_request/service_request_patient_granular_scope_test'
-require_relative './granular_scope_tests/service_request/service_request_id_granular_scope_test'
-require_relative './granular_scope_tests/service_request/service_request_patient_category_authored_granular_scope_test'
-require_relative './granular_scope_tests/service_request/service_request_patient_category_granular_scope_test'
-require_relative './granular_scope_tests/service_request/service_request_patient_code_authored_granular_scope_test'
-require_relative './granular_scope_tests/service_request/service_request_patient_code_granular_scope_test'
-require_relative './granular_scope_tests/service_request/service_request_patient_status_granular_scope_test'
+require_relative './granular_scope_tests/service_request/service_request_patient_granular_scope_search_test'
+require_relative './granular_scope_tests/service_request/service_request_id_granular_scope_search_test'
+require_relative './granular_scope_tests/service_request/service_request_patient_category_authored_granular_scope_search_test'
+require_relative './granular_scope_tests/service_request/service_request_patient_category_granular_scope_search_test'
+require_relative './granular_scope_tests/service_request/service_request_patient_code_authored_granular_scope_search_test'
+require_relative './granular_scope_tests/service_request/service_request_patient_code_granular_scope_search_test'
+require_relative './granular_scope_tests/service_request/service_request_patient_status_granular_scope_search_test'
+require_relative './granular_scope_tests/service_request/service_request_granular_scope_read_test'
 
 module USCoreTestKit
   module USCoreV700_BALLOT
@@ -24,13 +25,14 @@ based on the following granular scopes:
       run_as_group
 
     
-      test from: :us_core_v700_ballot_ServiceRequest_patient_granular_scope_test
-      test from: :us_core_v700_ballot_ServiceRequest__id_granular_scope_test
-      test from: :us_core_v700_ballot_ServiceRequest_patient_category_authored_granular_scope_test
-      test from: :us_core_v700_ballot_ServiceRequest_patient_category_granular_scope_test
-      test from: :us_core_v700_ballot_ServiceRequest_patient_code_authored_granular_scope_test
-      test from: :us_core_v700_ballot_ServiceRequest_patient_code_granular_scope_test
-      test from: :us_core_v700_ballot_ServiceRequest_patient_status_granular_scope_test
+      test from: :us_core_v700_ballot_ServiceRequest_patient_granular_scope_search_test
+      test from: :us_core_v700_ballot_ServiceRequest__id_granular_scope_search_test
+      test from: :us_core_v700_ballot_ServiceRequest_patient_category_authored_granular_scope_search_test
+      test from: :us_core_v700_ballot_ServiceRequest_patient_category_granular_scope_search_test
+      test from: :us_core_v700_ballot_ServiceRequest_patient_code_authored_granular_scope_search_test
+      test from: :us_core_v700_ballot_ServiceRequest_patient_code_granular_scope_search_test
+      test from: :us_core_v700_ballot_ServiceRequest_patient_status_granular_scope_search_test
+      test from: :us_core_v700_ballot_ServiceRequest_granular_scope_read_test
     end
   end
 end

data/lib/us_core_test_kit/generated/v7.0.0-ballot/service_request_granular_scope2_group.rb

@@ -1,10 +1,11 @@
-require_relative './granular_scope_tests/service_request/service_request_patient_granular_scope_test'
-require_relative './granular_scope_tests/service_request/service_request_id_granular_scope_test'
-require_relative './granular_scope_tests/service_request/service_request_patient_category_authored_granular_scope_test'
-require_relative './granular_scope_tests/service_request/service_request_patient_category_granular_scope_test'
-require_relative './granular_scope_tests/service_request/service_request_patient_code_authored_granular_scope_test'
-require_relative './granular_scope_tests/service_request/service_request_patient_code_granular_scope_test'
-require_relative './granular_scope_tests/service_request/service_request_patient_status_granular_scope_test'
+require_relative './granular_scope_tests/service_request/service_request_patient_granular_scope_search_test'
+require_relative './granular_scope_tests/service_request/service_request_id_granular_scope_search_test'
+require_relative './granular_scope_tests/service_request/service_request_patient_category_authored_granular_scope_search_test'
+require_relative './granular_scope_tests/service_request/service_request_patient_category_granular_scope_search_test'
+require_relative './granular_scope_tests/service_request/service_request_patient_code_authored_granular_scope_search_test'
+require_relative './granular_scope_tests/service_request/service_request_patient_code_granular_scope_search_test'
+require_relative './granular_scope_tests/service_request/service_request_patient_status_granular_scope_search_test'
+require_relative './granular_scope_tests/service_request/service_request_granular_scope_read_test'
 
 module USCoreTestKit
   module USCoreV700_BALLOT
@@ -25,13 +26,14 @@ based on the following granular scopes:
       run_as_group
 
     
-      test from: :us_core_v700_ballot_ServiceRequest_patient_granular_scope_test
-      test from: :us_core_v700_ballot_ServiceRequest__id_granular_scope_test
-      test from: :us_core_v700_ballot_ServiceRequest_patient_category_authored_granular_scope_test
-      test from: :us_core_v700_ballot_ServiceRequest_patient_category_granular_scope_test
-      test from: :us_core_v700_ballot_ServiceRequest_patient_code_authored_granular_scope_test
-      test from: :us_core_v700_ballot_ServiceRequest_patient_code_granular_scope_test
-      test from: :us_core_v700_ballot_ServiceRequest_patient_status_granular_scope_test
+      test from: :us_core_v700_ballot_ServiceRequest_patient_granular_scope_search_test
+      test from: :us_core_v700_ballot_ServiceRequest__id_granular_scope_search_test
+      test from: :us_core_v700_ballot_ServiceRequest_patient_category_authored_granular_scope_search_test
+      test from: :us_core_v700_ballot_ServiceRequest_patient_category_granular_scope_search_test
+      test from: :us_core_v700_ballot_ServiceRequest_patient_code_authored_granular_scope_search_test
+      test from: :us_core_v700_ballot_ServiceRequest_patient_code_granular_scope_search_test
+      test from: :us_core_v700_ballot_ServiceRequest_patient_status_granular_scope_search_test
+      test from: :us_core_v700_ballot_ServiceRequest_granular_scope_read_test
     end
   end
 end

data/lib/us_core_test_kit/generated/v7.0.0-ballot/us_core_test_suite.rb

@@ -5,6 +5,7 @@ require_relative '../../custom_groups/v4.0.0/clinical_notes_guidance_group'
 require_relative '../../custom_groups/data_absent_reason_group'
 require_relative '../../custom_groups/smart_app_launch_group'
 require_relative '../../custom_groups/v7.0.0-ballot/smart_granular_scopes_group'
+require_relative '../../custom_groups/v7.0.0-ballot/screening_assessment_group'
 require_relative '../../provenance_validator'
 require_relative '../../us_core_options'
 
@@ -207,10 +208,11 @@ module USCoreTestKit
         group from: :us_core_v700_ballot_related_person
         group from: :us_core_v700_ballot_specimen
         group from: :us_core_v400_clinical_notes_guidance
+        group from: :us_core_v700_ballot_screening_assessment
         group from: :us_core_311_data_absent_reason
       end
 
-      group from: :'us_core_v700_ballot_smart_granular_scopes',
+      group from: :us_core_v700_ballot_smart_granular_scopes,
             required_suite_options: USCoreOptions::SMART_2_REQUIREMENT
       
     end

data/lib/us_core_test_kit/generator/granular_scope_group_generator.rb

@@ -4,18 +4,11 @@ require_relative '../custom_groups/smart_scopes_constants'
 module USCoreTestKit
   class Generator
     class GranularScopeGroupGenerator
-      include SmartScopesConstants
-
       class << self
         def generate(ig_metadata, base_output_dir)
           return unless ['6', '7'].include? ig_metadata.ig_version[1]
 
           [1, 2].each do |group_number|
-            scopes =
-              SmartScopesConstants
-                .const_get("SMART_GRANULAR_SCOPES_GROUP#{group_number}")[ig_metadata.reformatted_version]
-            next if scopes.blank?
-
             new(ig_metadata, base_output_dir, group_number).generate
           end
         end

data/lib/us_core_test_kit/generator/granular_scope_read_test_generator.rb

@@ -0,0 +1,102 @@
+require_relative 'naming'
+require_relative 'special_cases'
+require_relative '../custom_groups/smart_scopes_constants'
+
+module USCoreTestKit
+  class Generator
+    class GranularScopeReadTestGenerator
+      class << self
+        def generate(ig_metadata, base_output_dir)
+          
+          return unless ['6', '7'].include? ig_metadata.ig_version[1]
+          
+          scopes =
+            SmartScopesConstants::SMART_GRANULAR_SCOPES_GROUP1[ig_metadata.reformatted_version] +
+            (SmartScopesConstants::SMART_GRANULAR_SCOPES_GROUP2[ig_metadata.reformatted_version] || [])
+
+          SmartScopesConstants::SMART_GRANULAR_SCOPE_RESOURCES.each do |resource_type|
+            group = ig_metadata.groups.find { |group| group.resource == resource_type }
+
+            next if scopes.none? { |scope| scope.start_with? "patient/#{group.resource}" }
+
+          new(group, base_output_dir).generate
+          end
+        end
+      end
+
+      attr_accessor :group_metadata, :base_output_dir, :group_number
+
+      def initialize(group_metadata, base_output_dir)
+        self.group_metadata = group_metadata
+        self.base_output_dir = base_output_dir
+      end
+
+      def template
+        @template ||= File.read(File.join(__dir__, 'templates', 'granular_scope_read_test.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(base_output_dir, 'granular_scope_tests', resource_type.underscore)
+      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_#{group_metadata.reformatted_version}_#{resource_type}_granular_scope_read_test"
+      end
+
+      def class_name
+        "#{resource_type}GranularScopeReadTest"
+      end
+
+      def module_name
+        "USCore#{group_metadata.reformatted_version.upcase}"
+      end
+
+      def resource_type
+        group_metadata.resource
+      end
+
+      def path_for_value(path)
+        path == 'class' ? 'local_class' : path
+      end
+
+      def array_of_strings(array)
+        quoted_strings = array.map { |element| "'#{element}'" }
+        "[#{quoted_strings.join(', ')}]"
+      end
+
+      def generate
+        FileUtils.mkdir_p(output_file_directory)
+        File.open(output_file_name, 'w') { |f| f.write(output) }
+
+        group_metadata.add_granular_scope_test(
+          id: test_id,
+          file_name: base_output_file_name
+        )
+      end
+
+      def description
+        <<~DESCRIPTION.gsub(/\n{3,}/, "\n\n")
+          This test attempts #{resource_type} reads 
+          and verifies that the results have been
+          filtered based on the granted granular scopes.
+        DESCRIPTION
+      end
+    end
+  end
+end

data/lib/us_core_test_kit/generator/granular_scope_resource_type_group_generator.rb

@@ -16,7 +16,7 @@ module USCoreTestKit
                 .const_get("SMART_GRANULAR_SCOPES_GROUP#{group_number}")[ig_metadata.reformatted_version]
 
             groups.each do |group_metadata|
-              next if scopes.blank? || scopes.none? { |scope| scope.start_with? "patient/#{group_metadata.resource}" }
+              next if scopes.none? { |scope| scope.start_with? "patient/#{group_metadata.resource}" }
 
               new(GroupMetadata.new(group_metadata.to_hash), ig_metadata, base_output_dir, group_number).generate
             end

data/lib/us_core_test_kit/generator/granular_scope_test_generator.rb

@@ -11,12 +11,12 @@ module USCoreTestKit
 
           scopes =
             SmartScopesConstants::SMART_GRANULAR_SCOPES_GROUP1[ig_metadata.reformatted_version] +
-            (SmartScopesConstants::SMART_GRANULAR_SCOPES_GROUP2[ig_metadata.reformatted_version] || [])
+            SmartScopesConstants::SMART_GRANULAR_SCOPES_GROUP2[ig_metadata.reformatted_version]
 
           SmartScopesConstants::SMART_GRANULAR_SCOPE_RESOURCES.each do |resource_type|
             group = ig_metadata.groups.find { |group| group.resource == resource_type }
 
-            next if scopes.blank? || scopes.none? { |scope| scope.start_with? "patient/#{group.resource}" }
+            next if scopes.none? { |scope| scope.start_with? "patient/#{group.resource}" }
 
             group.searches
               .each { |search| new(group, search, base_output_dir).generate }
@@ -57,7 +57,7 @@ module USCoreTestKit
       end
 
       def test_id
-        "us_core_#{group_metadata.reformatted_version}_#{resource_type}_#{search_identifier}_granular_scope_test"
+        "us_core_#{group_metadata.reformatted_version}_#{resource_type}_#{search_identifier}_granular_scope_search_test"
       end
 
       def search_identifier
@@ -69,7 +69,7 @@ module USCoreTestKit
       end
 
       def class_name
-        "#{resource_type}#{search_title}GranularScopeTest"
+        "#{resource_type}#{search_title}GranularScopeSearchTest"
       end
 
       def module_name

data/lib/us_core_test_kit/generator/suite_generator.rb

@@ -75,7 +75,7 @@ module USCoreTestKit
         end
       end
 
-      def test_granular_scopes?
+      def us_core_6_and_above?
         ig_metadata.ig_version[1].to_i > 5
       end
 
@@ -129,6 +129,14 @@ module USCoreTestKit
       def granular_scopes_id
         "us_core_#{ig_metadata.reformatted_version}_smart_granular_scopes"
       end
+
+      def screening_assessment_file_name
+        "../../custom_groups/#{ig_metadata.ig_version}/screening_assessment_group"
+      end
+
+      def screening_assessment_id
+        "us_core_#{ig_metadata.reformatted_version}_screening_assessment"
+      end
     end
   end
 end

data/lib/us_core_test_kit/generator.rb

@@ -6,6 +6,7 @@ require_relative 'generator/ig_metadata_extractor'
 require_relative 'generator/granular_scope_group_generator'
 require_relative 'generator/granular_scope_resource_type_group_generator'
 require_relative 'generator/granular_scope_test_generator'
+require_relative 'generator/granular_scope_read_test_generator'
 require_relative 'generator/group_generator'
 require_relative 'generator/must_support_test_generator'
 require_relative 'generator/provenance_revinclude_search_test_generator'
@@ -104,6 +105,7 @@ module USCoreTestKit
 
     def generate_granular_scope_tests
       GranularScopeTestGenerator.generate(ig_metadata, base_output_dir)
+      GranularScopeReadTestGenerator.generate(ig_metadata, base_output_dir)
     end
 
     def generate_groups

data/lib/us_core_test_kit/granular_scope.rb

@@ -0,0 +1,63 @@
+module USCoreTestKit
+  module GranularScope
+
+    def granular_scopes
+      @granular_scopes ||=
+        received_scopes
+          .split(' ')
+          .select do |scope|
+            (scope.start_with?("patient/#{resource_type}") || scope.start_with?("user/#{resource_type}")) &&
+              scope.include?('?')
+          end
+    end
+
+    def granular_scope_search_params
+      @granular_scope_search_params ||=
+        granular_scopes.map do |scope|
+          _, granular_scope = scope.split('?')
+          name, value = granular_scope.split('=')
+
+          {
+            name:,
+            value:
+          }
+        end
+    end
+
+    def resource_specific_granular_scope_search_params
+      @resource_specific_granular_scope_search_params ||=
+        granular_scopes.select {|scope| scope.include?(resource_type)}.map do |scope|
+          _, granular_scope = scope.split('?')
+          name, value = granular_scope.split('=')
+          _, value = value.split('|')
+          {
+            name:,
+            value:
+          }
+      end
+    end
+
+    def previous_request_resources
+      first_request = previous_requests.first
+      next_page_url = nil
+      hash = Hash.new { |hash, key| hash[key] = [] }
+      previous_requests.each_with_object(hash) do |request, request_resource_hash|
+        request_resources =
+          if request.status == 200
+            request.resource.entry.map(&:resource).select { |resource| resource.resourceType == resource_type }
+          else
+            []
+          end
+
+        first_request = request if request.url != next_page_url
+
+        request_resource_hash[first_request].concat(request_resources)
+
+        next if request.resource&.resourceType != 'Bundle'
+
+        next_page_url = request.resource&.link&.find { |link| link.relation == 'next' }&.url
+      end
+    end
+
+  end
+end

data/lib/us_core_test_kit/granular_scope_read_test.rb

@@ -0,0 +1,71 @@
+require_relative 'resource_search_param_checker'
+require_relative 'granular_scope'
+
+module USCoreTestKit
+  module GranularScopeReadTest
+    extend Forwardable
+    include ResourceSearchParamChecker
+    include GranularScope
+
+    def_delegators 'self.class', :metadata
+    def self.included(klass)
+      klass.input(:received_scopes)
+      klass.attr_accessor :previous_requests
+    end
+
+    def run_scope_read_test
+      assert granular_scopes.present?, "No granular scopes were received for #{resource_type} resources"
+      resource_scopes = granular_scopes.select {|scope| scope.include?(resource_type)}
+
+      load_previous_requests
+      skip_if previous_requests.blank?,
+              "No #{resource_type} reads found"
+      previous_resources_for_reads = previous_request_resources.values.flatten
+
+      resource_specific_granular_scope_search_params.each do |scope| 
+        current_scope = granular_scopes.find {|granular| granular.include?(scope[:name]) && granular.include?(scope[:value])}
+
+        resource_matching_scope = previous_resources_for_reads.find do |prev_resource| 
+          resource_matches_param?(prev_resource, scope[:name], scope[:value])
+        end
+        skip_if resource_matching_scope.nil?, "Unable to find any resources to match scope #{current_scope}"
+
+        fhir_read resource_type, resource_matching_scope.id
+
+        assert_response_status(200)
+        assert_resource_type(resource_type)
+      end
+
+      nonmatching_resource = previous_resources_for_reads.find do |prev_resource|
+        resource_specific_granular_scope_search_params.none? do |scope| 
+          resource_matches_param?(prev_resource, scope[:name], scope[:value])
+        end
+      end
+      if nonmatching_resource
+        fhir_read resource_type, nonmatching_resource.id
+        assert (response && response[:status]) != 200, "Server incorrectly responded with a successful status, read should fail due to scopes."
+        assert (resource && resource.resourceType != resource_type), "Server incorrectly returned a #{resource_type}, read should fail due to scopes"
+      else
+        info "Unable to find a resource that does not match scopes."
+      end
+    end
+    
+    def load_previous_requests
+      params = metadata.searches.first[:names]
+      search_params_as_hash = params.each_with_object({}) do |name, hash|
+        hash[name] = nil
+      end
+      @previous_requests ||=
+        load_tagged_requests(search_params_tag(search_params_as_hash))
+          .sort_by { |request| request.index }
+    end
+
+    def unescape_search_value(value)
+      value&.gsub('\\,', ',')
+    end
+
+    def search_params_tag(params)
+      "#{resource_type}?#{params.keys.join('&')}"
+    end
+  end
+end

data/lib/us_core_test_kit/{granular_scope_checker.rb → granular_scope_search_test.rb}

@@ -1,11 +1,17 @@
+require_relative 'resource_search_param_checker'
+require_relative 'granular_scope'
+
 module USCoreTestKit
-  module GranularScopeChecker
+  module GranularScopeSearchTest
+    include ResourceSearchParamChecker
+    include GranularScope
+
     def self.included(klass)
       klass.input(:received_scopes)
       klass.attr_accessor :previous_requests
     end
 
-    def run_scope_check_test
+    def run_scope_search_test
       assert granular_scopes.present?, "No granular scopes were received for #{resource_type} resources"
 
       load_previous_requests
@@ -47,51 +53,6 @@ module USCoreTestKit
       end
     end
 
-    def previous_request_resources
-      first_request = previous_requests.first
-      next_page_url = nil
-      hash = Hash.new { |hash, key| hash[key] = [] }
-      previous_requests.each_with_object(hash) do |request, request_resource_hash|
-        request_resources =
-          if request.status == 200
-            request.resource.entry.map(&:resource).select { |resource| resource.resourceType == resource_type }
-          else
-            []
-          end
-
-        first_request = request if request.url != next_page_url
-
-        request_resource_hash[first_request].concat(request_resources)
-
-        next if request.resource&.resourceType != 'Bundle'
-
-        next_page_url = request.resource&.link&.find { |link| link.relation == 'next' }&.url
-      end
-    end
-
-    def granular_scopes
-      @granular_scopes ||=
-        received_scopes
-          .split(' ')
-          .select do |scope|
-            (scope.start_with?("patient/#{resource_type}") || scope.start_with?("user/#{resource_type}")) &&
-              scope.include?('?')
-          end
-    end
-
-    def granular_scope_search_params
-      @granular_scope_search_params ||=
-        granular_scopes.map do |scope|
-          _, granular_scope = scope.split('?')
-          name, value = granular_scope.split('=')
-
-          {
-            name:,
-            value:
-          }
-        end
-    end
-
     def load_previous_requests
       search_params_as_hash = search_param_names.each_with_object({}) do |name, hash|
         hash[name] = nil

data/lib/us_core_test_kit/must_support_test.rb

@@ -115,12 +115,13 @@ module USCoreTestKit
               .map { |ex| ex[:url] }
 
             value_found = find_a_value_at(resource, path) do |value|
-              if value.respond_to?(:extension) && ms_extension_urls.present?
-                urls = value.extension.map(&:url)
+              if value.instance_of?(USCoreTestKit::PrimitiveType) && ms_extension_urls.present?
+                urls = value.extension&.map(&:url)
                 has_ms_extension = (urls & ms_extension_urls).present?
               end
 
               unless has_ms_extension
+                value = value.value if value.instance_of?(USCoreTestKit::PrimitiveType)
                 value_without_extensions =
                   value.respond_to?(:to_hash) ? value.to_hash.reject { |key, _| key == 'extension' } : value
               end

data/lib/us_core_test_kit/primitive_type.rb

@@ -0,0 +1,5 @@
+module USCoreTestKit
+  class PrimitiveType < FHIR::Element
+    attr_accessor :value
+  end
+end

data/lib/us_core_test_kit/provenance_validator.rb

@@ -19,11 +19,10 @@ module USCoreTestKit
 
       failed_provenance =
         find_a_value_at(resource, 'agent') do |agent|
-          ['Practitioner', 'Device'].any? { |resource_type| agent.who.reference&.include?(resource_type) } &&
-          agent.onBehalfOf.nil?
+          ['Practitioner', 'Device'].any? { |resource_type| agent.who&.reference&.include?(resource_type) } &&
+            agent.onBehalfOf.nil?
         end
 
-
       if failed_provenance.present?
         validation_messages << {
           type: 'error',
@@ -34,4 +33,4 @@ module USCoreTestKit
       validation_messages
     end
   end
-end
+end