lib/eancom/edifact/segments/unh.rb in eancom-1.3.0 vs lib/eancom/edifact/segments/unh.rb in eancom-1.4.0

@@ -9,11 +9,11 @@
         message_reference_number:,
         message_type:,
         message_version_number:,
         message_release_number:,
         controlling_agency:,
-        association_assigned_code:,
+        association_assigned_code: nil,
         code_list_directory_version_number: nil
       )
 
         @tag = tag
         @message_reference_number = message_reference_number
@@ -25,17 +25,19 @@
         @code_list_directory_version_number = code_list_directory_version_number
         super(tag: tag || TAG)
       end
 
       def to_json_hash
-        {
+        hash = {
           message_reference_number: @message_reference_number,
           message_type: @message_type,
           message_version_number: @message_version_number,
           message_release_number: @message_release_number,
           controlling_agency: @controlling_agency,
-          association_assigned_code: @association_assigned_code
         }
+        hash[:association_assigned_code] = @association_assigned_code if @association_assigned_code
+        hash[:code_list_directory_version_number] = @code_list_directory_version_number if @code_list_directory_version_number
+        hash
       end
 
       def segment_type
         TYPE
       end