add tests

test/conftest.py

@@ -227,3 +227,62 @@ def test_charging_structure_data():
                     nren=nren, year=year,
                     fee_type=fee_type)
             )
+
+
+@pytest.fixture
+def test_organization_data():
+    def _generate_sub_org_data():
+        for nren in ["nren" + str(i) for i in range(1, 50)]:
+            for year in range(2016, 2021):
+                yield {
+                    'nren': nren,
+                    'year': year,
+                    'name': 'sub_org' + str(random.randint(1, 100)),
+                    'role': random.choice(['technical centre', 'stuff', 'test123']),
+                }
+
+    def _generate_org_data():
+        for nren in ["nren" + str(i) for i in range(1, 50)]:
+            for year in range(2016, 2021):
+                yield {
+                    'nren': nren,
+                    'year': year,
+                    'name': 'org' + str(year)
+                }
+
+    with db.session_scope() as session:
+
+        org_data = list(_generate_org_data())
+        sub_org_data = list(_generate_sub_org_data())
+
+        nren_dict = {nren_name: model.NREN(name=nren_name)
+                     for nren_name in set(d['nren'] for d in [*org_data, *sub_org_data])}
+        session.add_all(nren_dict.values())
+
+        for org in org_data:
+            nren = nren_dict[org["nren"]]
+            year = org["year"]
+            name = org["name"]
+
+            session.add(
+                model.ParentOrganization(
+                    nren=nren,
+                    year=year,
+                    organization=name
+                )
+            )
+        for sub_org in sub_org_data:
+            nren = nren_dict[sub_org["nren"]]
+            year = sub_org["year"]
+            name = sub_org["name"]
+            role = sub_org["role"]
+
+            session.add(
+                model.SubOrganization(
+                    nren=nren,
+                    year=year,
+                    organization=name,
+                    role=role
+                )
+            )
+        session.commit()

test/test_organization.py

+import json
+import jsonschema
+from compendium_v2.routes.organization import ORGANIZATION_RESPONSE_SCHEMA
+
+
+def test_parentorganization_response(client, test_organization_data):
+    rv = client.get(
+        '/api/organization/parent',
+        headers={'Accept': ['application/json']})
+    assert rv.status_code == 200
+    result = json.loads(rv.data.decode('utf-8'))
+    jsonschema.validate(result, ORGANIZATION_RESPONSE_SCHEMA)
+    assert result
+
+def test_suborganization_response(client, test_organization_data):
+    rv = client.get(
+        '/api/organization/sub',
+        headers={'Accept': ['application/json']})
+    assert rv.status_code == 200
+    result = json.loads(rv.data.decode('utf-8'))
+    jsonschema.validate(result, ORGANIZATION_RESPONSE_SCHEMA)
+    assert result