add forgotten set_close_reason for channel

mosquito · Feb 20, 2023 · 647f0b8 · 647f0b8
1 parent 60bfc1f
commit 647f0b8
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/aiormq/channel.py b/aiormq/channel.py
@@ -397,6 +397,12 @@ async def _on_cancel_frame(
 
     async def _on_close_frame(self, frame: spec.Channel.Close) -> None:
         exc: BaseException = exception_by_code(frame)
+        self.set_close_reason(
+            frame.reply_code or -1,
+            frame.reply_text or "",
+            frame.class_id or -1,
+            frame.method_id or -1
+        )
         self.write_queue.put_nowait(
             ChannelFrame.marshall(
                 channel_number=self.number,