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)