rse-grand-challenge/app/tests/challenges_tests/test_migrations.py at 752cfdd8af40d312aedc847277ce671363f96319 · DIAGNijmegen/rse-grand-challenge · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
import importlib

import pytest
from django.apps import apps
from django.contrib.auth.models import Group
from guardian.shortcuts import get_group_perms, get_user_perms, remove_perm

from tests.factories import ChallengeRequestFactory

_migration_0070 = importlib.import_module(
    "grandchallenge.challenges.migrations"
    ".0070_assign_review_perm_to_reviewers_group"
)
_migration_0073 = importlib.import_module(
    "grandchallenge.challenges.migrations"
    ".0073_set_average_size_job_output_for_tasks"
)


@pytest.mark.django_db
def test_assign_review_perm_to_reviewers_group(settings):
    challenge_request = ChallengeRequestFactory()
    reviewers_group = Group.objects.get(
        name=settings.CHALLENGES_REVIEWERS_GROUP_NAME
    )

    # The permission is assigned when the object is created; remove it to
    # simulate the pre-migration state where it was missing.
    assert "review_challengerequest" in get_group_perms(
        reviewers_group, challenge_request
    )
    remove_perm("review_challengerequest", reviewers_group, challenge_request)
    assert "review_challengerequest" not in get_group_perms(
        reviewers_group, challenge_request
    )

    _migration_0070.assign_review_perm_to_reviewers_group(
        apps, schema_editor=None
    )

    assert "review_challengerequest" in get_group_perms(
        reviewers_group, challenge_request
    )


@pytest.mark.django_db
def test_assign_change_perm_to_creators():
    challenge_request = ChallengeRequestFactory()

    # The permission is assigned when the object is created; remove it to
    # simulate the pre-migration state where it was missing.
    assert "change_challengerequest" in get_user_perms(
        challenge_request.creator, challenge_request
    )
    remove_perm(
        "change_challengerequest", challenge_request.creator, challenge_request
    )
    assert "change_challengerequest" not in get_user_perms(
        challenge_request.creator, challenge_request
    )

    _migration_0070.assign_change_perm_to_creators(apps, schema_editor=None)

    assert "change_challengerequest" in get_user_perms(
        challenge_request.creator, challenge_request
    )


@pytest.mark.django_db
@pytest.mark.parametrize(
    "task_ids, expected",
    [
        ([], []),
        ([1], [0]),
        ([1, 2, 3], [0, 0, 0]),
    ],
)
def test_set_average_size_job_output_mb_for_tasks(task_ids, expected):
    challenge_request = ChallengeRequestFactory(
        task_ids=task_ids,
        average_size_job_output_mb_for_tasks=[],
    )

    # The field is empty before the migration.
    assert challenge_request.average_size_job_output_mb_for_tasks == []

    _migration_0073.set_average_size_job_output_mb_for_tasks(
        apps, schema_editor=None
    )

    challenge_request.refresh_from_db()
    assert challenge_request.average_size_job_output_mb_for_tasks == expected