def test_close(self): reader = Mock(channel=Mock()) tag = 1234 # test r = Receiver(reader) r.tag = tag r.close() # validation reader.channel.basic_cancel.assert_called_once_with(tag)
def test_close_exception(self): reader = Mock() reader.channel.basic_cancel.side_effect = ValueError tag = 1234 # test r = Receiver(reader) r.tag = tag r.close() # validation reader.channel.basic_cancel.assert_called_once_with(tag)