Sets Message.channel on delivery. Closes #12

1 files changed, 5 insertions, 3 deletions

diff --git a/amqp/method_framing.py b/amqp/method_framing.py
index a272213..cee05ba 100644
--- a/amqp/method_framing.py
+++ b/amqp/method_framing.py
@@ -46,10 +46,10 @@ _CONTENT_METHODS = [
 class _PartialMessage(object):
     """Helper class to build up a multi-frame method."""
 
-    def __init__(self, method_sig, args):
+    def __init__(self, method_sig, args, channel):
         self.method_sig = method_sig
         self.args = args
-        self.msg = Message()
+        self.msg = Message(channel=channel)
         self.body_parts = []
         self.body_received = 0
         self.body_size = None
@@ -147,7 +147,9 @@ class MethodReader(object):
             #
             # Save what we've got so far and wait for the content-header
             #
-            self.partial_messages[channel] = _PartialMessage(method_sig, args)
+            self.partial_messages[channel] = _PartialMessage(
+                method_sig, args, channel,
+            )
             self.expected_types[channel] = 2
         else:
             self._quick_put((channel, method_sig, args, None))