Skip to content

test_index_organizations#

test_index_organizations_from_file#

Source code in tests/unit/apps/actors/commands/test_index_organizations.py 
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
@pytest.mark.django_db
@override_settings(**test_settings)
def test_index_organizations_from_file():
    call_command("index_organizations", "--cached")
    orgs = Organization.available_objects.values(
        "pref_label__en", "parent__pref_label__en", "code", "url", "in_scheme"
    )

    expected_orgs = [
        {
            "pref_label__en": "Aalto University",
            "parent__pref_label__en": None,
            "code": "10076",
        },
        {
            "pref_label__en": "Geological Survey of Finland",
            "parent__pref_label__en": None,
            "code": "5040011",
        },
        {
            "pref_label__en": "LUT University",
            "parent__pref_label__en": None,
            "code": "01914",
        },
        {
            "pref_label__en": "ARTS Infra",
            "parent__pref_label__en": "Aalto University",
            "code": "10076-A855",
        },
        {
            "pref_label__en": "Aalto Common Items",
            "parent__pref_label__en": "Aalto University",
            "code": "10076-U920",
        },
        {
            "pref_label__en": "Aalto Nanofab",
            "parent__pref_label__en": "Aalto University",
            "code": "10076-T409",
        },
        {
            "pref_label__en": "Aalto Studios",
            "parent__pref_label__en": "Aalto University",
            "code": "10076-A850",
        },
        {
            "pref_label__en": "Alueellinen geotieto ALG",
            "parent__pref_label__en": "Geological Survey of Finland",
            "code": "5040011-504030014",
        },
    ]
    for org in expected_orgs:
        org["in_scheme"] = test_settings["ORGANIZATION_SCHEME"]
        org["url"] = f"{test_settings['ORGANIZATION_BASE_URI']}{org['code']}"

    assert list(orgs) == expected_orgs

test_index_organizations_from_api#

Source code in tests/unit/apps/actors/commands/test_index_organizations.py 
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
@pytest.mark.django_db
@override_settings(
    **{
        **test_settings,
        "ORGANIZATION_DATA_FILE": None,
        "ORGANIZATION_FETCH_API_URL": "https://referenceorganizations.com/_search",
    }
)
def test_index_organizations_from_api(requests_mock):
    with open(f"{test_data_path}test_orgs.json", "rb") as datafile:
        requests_mock.get(
            "https://referenceorganizations.com/_search",
            content=datafile.read(),
        )

    call_command("index_organizations")
    orgs = Organization.available_objects.values(
        "pref_label__en", "parent__pref_label__en", "code", "url", "in_scheme"
    )

    expected_orgs = [
        {
            "pref_label__en": "Diaconia University of Applied Sciences",
            "parent__pref_label__en": None,
            "code": "02623",
        },
        {
            "pref_label__en": "Innovaatiot ja kumppanuudet tulosalue",
            "parent__pref_label__en": "Diaconia University of Applied Sciences",
            "code": "02623-100",
        },
        {
            "pref_label__en": "Sub org with space",
            "parent__pref_label__en": "Diaconia University of Applied Sciences",
            "code": "02623-105 123",
        },
    ]
    for org in expected_orgs:
        org["in_scheme"] = test_settings["ORGANIZATION_SCHEME"]
        org["url"] = f"{test_settings['ORGANIZATION_BASE_URI']}{quote(org['code'])}"

    assert list(orgs) == expected_orgs

test_index_organizations_deprecation#

Reference data organizations removed from source data should be deprecated.

Source code in tests/unit/apps/actors/commands/test_index_organizations.py 
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
@pytest.mark.django_db
@override_settings(**test_settings)
def test_index_organizations_deprecation():
    """Reference data organizations removed from source data should be deprecated."""
    dep1 = Organization.objects.create(
        pref_label={"en": "This will be deprecated"},
        url=f"{test_settings['ORGANIZATION_BASE_URI']}deprecateme",
        in_scheme=test_settings["ORGANIZATION_SCHEME"],
        is_reference_data=True,
    )
    dep2 = Organization.objects.create(
        pref_label={"en": "This too"},
        url=f"{test_settings['ORGANIZATION_BASE_URI']}deprecatemetoo",
        in_scheme=test_settings["ORGANIZATION_SCHEME"],
        is_reference_data=True,
    )
    # Deprecated entry should be undeprecated when it is found in the data source
    undep = Organization.objects.create(
        url=f"{test_settings['ORGANIZATION_BASE_URI']}10076",
        deprecated="2022-01-02T11:22:33Z",
        pref_label={"en": "Aalto"},
        in_scheme=test_settings["ORGANIZATION_SCHEME"],
        is_reference_data=True,
    )

    call_command("index_organizations", "--cached")
    deprecated_orgs = Organization.available_objects.filter(deprecated__isnull=False)
    assert set(deprecated_orgs) == {dep1, dep2}

    undep.refresh_from_db()
    assert undep.deprecated is None