lib/wamp/message_handler/invocation.rb in xconn-0.1.0 vs lib/wamp/message_handler/invocation.rb in xconn-0.1.1

@@ -14,10 +14,13 @@
       def alt_store_key
         "registration_#{message.registration_id}"
       end
 
       def invocation_response
-        Type::Invocation.new(args: message.args, kwargs: message.kwargs, details: message.details)
+        Type::Invocation.new(args: message.args, kwargs: message.kwargs, details: message.details).tap do |invocation|
+          invocation.connection = connection
+          invocation.request_id = message.request_id
+        end
       end
 
       private
 
       def send_yield_message(handler)