Python delay Examples

Example #1

File: test_tasks.py Project: holandes22/rafee

def test_pull_all_repos(gm_mock):
    with patch('rafee.repositories.tasks.get_dst_path'):
        git_manager = gm_mock.return_value
        RepositoryFactory()
        RepositoryFactory()
        result = pull_all_repos.delay()
        assert [] == result.result['errors']
        assert 2 == git_manager.pull.call_count

Example #2

File: test_tasks.py Project: holandes22/rafee

def test_pull_all_repos_return_error_info(gm_mock):
    with patch('rafee.repositories.tasks.get_dst_path'):
        git_manager = gm_mock.return_value
        RepositoryFactory()
        repo2 = RepositoryFactory()
        repo3 = RepositoryFactory()
        error2 = ValueError('Fake msg2')
        error3 = IOError()
        git_manager.pull.side_effect = [Mock(), error2, error3]
        result = pull_all_repos.delay()
        expected = [
            {'repo': repo2.id, 'message': str(error2), 'error': 'ValueError'},
            {'repo': repo3.id, 'message': str(error3), 'error': 'IOError'},
        ]
        assert expected == result.result['errors']