Skip to content

test_organization#

organization_tree#

Source code in tests/unit/apps/actors/api/test_organization.py 
 9
10
11
12
13
14
15
16
17
@pytest.fixture
def organization_tree():
    OrganizationFactory.reset_sequence()
    org0 = OrganizationFactory.create()
    org1 = OrganizationFactory.create()
    org1_2 = OrganizationFactory.create(parent=org1)
    org1_2_3 = OrganizationFactory.create(parent=org1_2)
    org1_2_4 = OrganizationFactory.create(parent=org1_2)
    org1_5 = OrganizationFactory.create(parent=org1)

get_code_trees#

Convert org tree into nested dict, key from code, value from children

Source code in tests/unit/apps/actors/api/test_organization.py 
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
def get_code_trees(orgs):
    """Convert org tree into nested dict, key from code, value from children"""

    codes = {}
    for org in orgs:
        # list parents first
        while parent := org.get("parent"):
            if not parent.get("children"):
                parent["children"] = [org]
            org = parent

        def recurse(node):
            node_codes = {}
            children = node.get("children", [])
            for child in children:
                node_codes[child["code"]] = recurse(child)
            return node_codes

        codes[org["code"]] = recurse(org)
    return codes

check_org_trees#

Check that response for organization contains correct organizations based on codes.

Source code in tests/unit/apps/actors/api/test_organization.py 
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
@pytest.fixture
def check_org_trees(organization_tree):
    """Check that response for organization contains correct organizations based on codes."""

    def func(org_code, expected_code_tree):
        client = APIClient()
        code_trees = {}
        if org_code:
            # single org
            org_id = Organization.available_objects.get(code=org_code).id
            resp = client.get(
                f"{reverse('organization-detail', args=[org_id])}", {"expand_children": True}
            )
            assert resp.status_code == 200
            code_trees = get_code_trees([resp.data])
        else:
            # list of orgs
            resp = client.get(f"{reverse('organization-list')}", {"expand_children": True})
            assert resp.status_code == 200
            code_trees = get_code_trees(resp.data["results"])
        assert code_trees == expected_code_tree

    return func

test_get_org_list#

Source code in tests/unit/apps/actors/api/test_organization.py 
67
68
69
@pytest.mark.django_db
def test_get_org_list(check_org_trees):
    check_org_trees(None, {"0": {}, "1": {"1-2": {"1-2-3": {}, "1-2-4": {}}, "1-5": {}}})

test_get_main_org#

Source code in tests/unit/apps/actors/api/test_organization.py 
72
73
74
@pytest.mark.django_db
def test_get_main_org(check_org_trees):
    check_org_trees("1", {"1": {"1-2": {"1-2-3": {}, "1-2-4": {}}, "1-5": {}}})

test_get_sub_org#

Source code in tests/unit/apps/actors/api/test_organization.py 
77
78
79
@pytest.mark.django_db
def test_get_sub_org(check_org_trees):
    check_org_trees("1-2", {"1": {"1-2": {"1-2-3": {}, "1-2-4": {}}}})

test_get_sub_sub_org#

Source code in tests/unit/apps/actors/api/test_organization.py 
82
83
84
@pytest.mark.django_db
def test_get_sub_sub_org(check_org_trees):
    check_org_trees("1-2-3", {"1": {"1-2": {"1-2-3": {}}}})

test_get_org_list_no_include_suborganizations#

Source code in tests/unit/apps/actors/api/test_organization.py 
87
88
89
90
91
92
93
94
@pytest.mark.django_db
def test_get_org_list_no_include_suborganizations(organization_tree, client):
    resp = client.get(
        reverse("organization-list"),
        {"pagination": False},
    )
    assert resp.status_code == 200
    assert len(resp.data) == 2

test_get_org_list_include_suborganizations#

Source code in tests/unit/apps/actors/api/test_organization.py 
 97
 98
 99
100
101
102
103
104
@pytest.mark.django_db
def test_get_org_list_include_suborganizations(organization_tree, client):
    resp = client.get(
        reverse("organization-list"),
        {"pagination": False, "include_suborganizations": True},
    )
    assert resp.status_code == 200
    assert len(resp.data) == 6