Skip to content

conftest#

data_urls#

Return urls related to dataset data.

Source code in tests/unit/apps/conftest.py 
16
17
18
19
20
21
22
23
24
25
26
27
28
29
@pytest.fixture
def data_urls():
    """Return urls related to dataset data."""

    def f(dataset: Union[Dataset, UUID, str]):
        dataset_id = dataset.id if isinstance(dataset, Dataset) else dataset
        kwargs = {"dataset_id": dataset_id}
        return {
            "dataset": reverse("dataset-detail", kwargs={"pk": dataset_id}),
            "files": reverse("dataset-files-list", kwargs=kwargs),
            "directories": reverse("dataset-directories-list", kwargs=kwargs),
        }

    return f

file_tree#

Source code in tests/unit/apps/conftest.py 
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
@pytest.fixture
def file_tree() -> dict:
    return create_project_with_files(
        file_paths=[
            "/dir1/file.csv",
            "/dir2/a.txt",
            "/dir2/b.txt",
            "/dir2/c.txt",
            "/dir2/subdir/file1.txt",
            "/dir2/subdir/file2.txt",
            "/dir3/file.pdf",
            "/rootfile.txt",
        ],
        file_args={"*": {"size": 1024}},
    )

deep_file_tree#

Source code in tests/unit/apps/conftest.py 
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
@pytest.fixture
def deep_file_tree() -> dict:
    return create_project_with_files(
        file_paths=[
            "/dir1/file.csv",
            "/dir1/sub/file.csv",
            "/dir2/a.txt",
            "/dir2/subdir1/file1.txt",
            "/dir2/subdir1/file2.txt",
            "/dir2/subdir1/file3.txt",
            "/dir2/subdir2/file1.txt",
            "/dir2/subdir2/file2.txt",
            "/dir2/subdir2/file3.txt",
            "/dir2/subdir2/subsub/subsubsub1/file.txt",
            "/dir2/subdir2/subsub/subsubsub2/file1.txt",
            "/dir2/subdir2/subsub/subsubsub2/file2.txt",
            "/dir3/sub1/file.txt",
            "/dir3/sub2/file.txt",
            "/rootfile.txt",
        ],
        file_args={"*": {"size": 1024}},
    )

dataset_with_files#

Source code in tests/unit/apps/conftest.py 
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
@pytest.fixture
def dataset_with_files(data_catalog, file_tree):
    dataset = factories.DatasetFactory(data_catalog=data_catalog, persistent_identifier="somepid")
    factories.DatasetActorFactory(roles=["creator", "publisher"], dataset=dataset)
    storage = next(iter(file_tree["files"].values())).storage
    file_set = factories.FileSetFactory(dataset=dataset, storage=storage)
    file_set.files.set(
        [
            file_tree["files"]["/dir1/file.csv"],
            file_tree["files"]["/dir2/a.txt"],
            file_tree["files"]["/dir2/b.txt"],
            file_tree["files"]["/dir2/subdir/file1.txt"],
        ]
    )
    return dataset

TokenUsers#

Source code in tests/unit/apps/conftest.py 
90
91
92
93
@dataclass
class TokenUsers:
    user: get_user_model()
    token: str

end_users#

Source code in tests/unit/apps/conftest.py 
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
@pytest.fixture
def end_users(faker, fairdata_users_group):
    user1 = get_user_model().objects.create(
        username=faker.simple_profile()["username"], password=faker.password(), organization="test"
    )
    user2 = get_user_model().objects.create(
        username=faker.simple_profile()["username"], password=faker.password(), organization="test"
    )
    user3 = get_user_model().objects.create(
        username=faker.simple_profile()["username"], password=faker.password(), organization="test"
    )
    fairdata_users_group.user_set.add(user1, user2, user3)
    instance1, token1 = AuthToken.objects.create(user=user1)
    instance2, token2 = AuthToken.objects.create(user=user2)
    instance3, token3 = AuthToken.objects.create(user=user3)
    return (
        TokenUsers(user=user1, token=token1),
        TokenUsers(user=user2, token=token2),
        TokenUsers(user=user3, token=token3),
    )

service_user#

Source code in tests/unit/apps/conftest.py 
118
119
120
121
122
123
124
125
126
@pytest.fixture
def service_user(faker):
    group, _ = Group.objects.get_or_create(name="service")
    user = get_user_model().objects.create(
        username=faker.simple_profile()["username"], password=faker.password()
    )
    instance, token = AuthToken.objects.create(user=user)
    user.groups.add(group)
    return TokenUsers(user=user, token=token)

ida_service_user#

Source code in tests/unit/apps/conftest.py 
129
130
131
132
133
@pytest.fixture
def ida_service_user(service_user):
    group_ida, _ = Group.objects.get_or_create(name="ida")
    service_user.user.groups.add(group_ida)
    return service_user

update_request_client_auth_token#

Source code in tests/unit/apps/conftest.py 
136
137
138
139
140
141
142
@pytest.fixture
def update_request_client_auth_token():
    def _update_request_client_auth_token(client, token):
        client.headers.update({"Authorization": f"Bearer {token}"})
        return client

    return _update_request_client_auth_token