def test_listen_required_not_needed(self): with mock.patch('yowsup_celery.tasks.YowsupTask.stack', new_callable=mock.PropertyMock) as mock_stack, \ mock.patch('yowsup_celery.tasks.YowsupTask.facade', new_callable=mock.PropertyMock) as mock_facade: mock_stack.return_value = mock.MagicMock(listening=True) mock_stack.return_value.stack = mock.MagicMock() tasks.connect() self.assertEqual(0, mock_stack.asynloop.call_count) mock_facade.return_value.connect.assert_called_once_with()
def test_listen_required(self): with mock.patch('yowsup_celery.tasks.YowsupTask.stack', new_callable=mock.PropertyMock) as mock_stack, \ mock.patch('yowsup_celery.tasks.YowsupTask.facade', new_callable=mock.PropertyMock) as mock_facade, \ mock.patch('celery.app.task.Context', new_callable=mock.MagicMock) as mock_request, \ mock.patch('yowsup_celery.tasks.connect.apply_async', new_callable=tasks.listen), \ mock.patch('yowsup_celery.tasks.connect.retry', new_callable=tasks.connect): mock_request.return_value = mock.MagicMock(delivery_info={'routing_key': 'any_queue'}) mock_stack.return_value = mock.MagicMock(listening=False) mock_stack.return_value.stack = mock.MagicMock() tasks.connect() mock_stack.return_value.asynloop.assert_called_once_with() mock_facade.return_value.connect.assert_called_once_with()