Updated test to make sure that tags are cleared on graceful shutdwon

eandersson · Feb 11, 2017 · f8832c5 · f8832c5
1 parent d173d1e
commit f8832c5
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/amqpstorm/tests/unit/channel/channel_tests.py b/amqpstorm/tests/unit/channel/channel_tests.py
@@ -227,8 +227,7 @@ def on_close_ok(_, frame_out):
     def test_channel_close_gracefully_with_queued_error(self):
         def on_close_ok(_, frame_out):
             if isinstance(frame_out, specification.Basic.Cancel):
-                channel.rpc.on_frame(specification.Basic.CancelOk())
-                return
+                raise AMQPChannelError("travis-ci")
             channel.rpc.on_frame(specification.Channel.CloseOk())
 
         channel = Channel(0, FakeConnection(on_write=on_close_ok), 360)