Function create_project_with_context

shared/tests/test_utils/data_mocking.py:796–865

This function will create mock data for a project and all the necessary context for any unit testing to be performed. context_data will be a dictionary with all the context for creating the project. The idea is that you'll be able to specify users, labels, tasks, etc... and the func

(context_data, session)

def create_project_with_context(context_data, session):
    """
    This function will create mock data for a project and all the necessary context for any unit testing
    to be performed.
    context_data will be a dictionary with all the context for creating the project.
    The idea is that you'll be able to specify users, labels, tasks, etc... and the function will make sure all the
    data is mocked properly.

    Example: For my test I need a project with 2 users, 1 admin and another with view permissions. I also need 3 labels.
    The context_data should look something like this:
    {
    'project_name': 'My test project',
    'users': [
        {'name': 'john', permissions: 'admin'}
        {'name': 'maria', permissions: 'view'}
    },
    'labels': [
        {'name': 'catlabel', 'type': 'box'}
        {'name': 'dogabel2', 'type': 'box'}
    ]

    The function will return a similar data structure with the ID's on the test database for further querying
    inside the test cases.

    :param context_data:
    :return:
    """

    random_name = get_random_string(8)
    project_string_id = context_data.get('project_string_id', random_name)
    project_name = context_data.get('project_name', random_name)

    default_project_limit = 10
    user = register_user(
        {'username': f"project_owner_{project_string_id}",
         'email': f"test{project_string_id}@test.com",
         'project_roles': ['admin'],
         'password': 'diffgram123'},
        session
    )
    member = Member(kind = 'human')
    session.add(member)
    session.flush()
    user.member = member

    project = Project.new(
        session = session,
        name = project_name,
        project_string_id = project_string_id,
        goal = 'Test stuff',
        member_created = member,
        user = user
    )
    user_list = []
    for user in context_data['users']:
        if user.get('project_string_id') is None:
            user['project_string_id'] = random_name
        new_user = register_user(user, session)