Merge pull request #55 from derekmwright/issue-54

Fix for issue #54 update to lib/wasabi/parser.rb
savonrb · Mar 2, 2015 · 17e37d1 · 17e37d1
2 parents bb89903 + 70dc788
commit 17e37d1
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/lib/wasabi/parser.rb b/lib/wasabi/parser.rb
@@ -257,8 +257,10 @@ def input_for(operation)
         end
 
         message = @messages[port_message_type]
-        port_message_part = message.element_children.find do |node|
-          soap_body_parts.nil? ? (node.name == 'part') : ( node.name == 'part' && node['name'] == soap_body_parts)
+        if !message.nil?
+          port_message_part = message.element_children.find do |node|
+            soap_body_parts.nil? ? (node.name == 'part') : ( node.name == 'part' && node['name'] == soap_body_parts)
+          end
         end
 
         if port_message_part && port_element = port_message_part.attribute('element')
@@ -315,8 +317,10 @@ def output_for(operation)
         end
 
         message = @messages[port_message_type]
-        port_message_part = message.element_children.find do |node|
-          soap_body_parts.nil? ? (node.name == 'part') : ( node.name == 'part' && node['name'] == soap_body_parts)
+        if !message.nil?
+          port_message_part = message.element_children.find do |node|
+            soap_body_parts.nil? ? (node.name == 'part') : ( node.name == 'part' && node['name'] == soap_body_parts)
+          end
         end
 
         if port_message_part && port_element = port_message_part.attribute('element')