def test_close(self): connection = Mock() channel = Mock() # test sender = Sender(None) sender.connection = connection sender.channel = channel sender.is_open = Mock(return_value=True) sender.close() # validation channel.close.assert_called_once_with() self.assertFalse(connection.close.called)
def test_repair(self, connection): url = 'test-url' # test sender = Sender(url) sender.close = Mock() sender.repair() # validation sender.close.assert_called_once_with() sender.connection.close.assert_called_once_with() connection.return_value.open.assert_called_once_with() connection.return_value.channel.assert_called_once_with() self.assertEqual(sender.channel, connection.return_value.channel.return_value)