def test_listen(self):
with mock.patch('yowsup_celery.tasks.YowsupTask.stack', new_callable=mock.PropertyMock) as mock_stack:
mock_stack.return_value = mock.MagicMock(listening=False)
mock_stack.return_value.stack = mock.MagicMock()
tasks.listen()
mock_stack.return_value.asynloop.assert_called_once_with()
def test_only_one_listen(self):
with mock.patch('yowsup_celery.tasks.YowsupTask.stack', new_callable=mock.PropertyMock) as mock_stack:
mock_stack.return_value = mock.MagicMock(listening=True)
mock_stack.return_value.stack = mock.MagicMock()
tasks.listen()
self.assertEqual(0, mock_stack.asynloop.call_count)
