Fix for DSTU2/STU3 set_client_on_resource (fhir-crucible#144)

* fix setting client on dstu2 and stu3 references

* bump version

lib/fhir_client/client.rb

@@ -349,7 +349,9 @@ def set_client_on_resource(resource)
 
  resource.client = self
  resource.each_element do |element, _, _|
- element.client = self if element.is_a?(Reference) || element.respond_to?(:resourceType)
+ if element.is_a?(Reference) || element.is_a?(STU3::Reference) || element.is_a?(DSTU2::Reference) || element.respond_to?(:resourceType)
+ element.client = self
+ end
  end
  end
 

lib/fhir_client/version.rb

@@ -1,5 +1,5 @@
 module FHIR
  class Client
- VERSION = '5.0.0'.freeze
+ VERSION = '5.0.1'.freeze
  end
 end

test/unit/set_client_on_resource_test.rb

@@ -0,0 +1,46 @@
+require_relative '../test_helper'
+
+class SetClientOnResourceTest < Test::Unit::TestCase
+ def client
+ @client ||= FHIR::Client.new('abc')
+ end
+
+ def test_r4
+ condition = FHIR::Condition.new(
+ resourceType: 'Condition',
+ subject: {
+ reference: 'Patient/123'
+ }
+ )
+ client.set_client_on_resource(condition)
+
+ assert_equal(client, condition.client)
+ assert_equal(client, condition.subject.client)
+ end
+
+ def test_stu3
+ condition = FHIR::STU3::Condition.new(
+ resourceType: 'Condition',
+ subject: {
+ reference: 'Patient/123'
+ }
+ )
+ client.set_client_on_resource(condition)
+
+ assert_equal(client, condition.client)
+ assert_equal(client, condition.subject.client)
+ end
+
+ def test_dstu2
+ condition = FHIR::DSTU2::Condition.new(
+ resourceType: 'Condition',
+ patient: {
+ reference: 'Patient/123'
+ }
+ )
+ client.set_client_on_resource(condition)
+
+ assert_equal(client, condition.client)
+ assert_equal(client, condition.patient.client)
+ end
+end