Philipp/aki_prj23_transparenzregister
1
0
Fork 0
mirror of https://github.com/fhswf/aki_prj23_transparenzregister.git synced 2025-06-22 19:33:54 +02:00
Code Issues Packages Projects Releases Wiki Activity

ignore types mypy

Browse Source
This commit is contained in:
TrisNol
2023-09-21 18:08:20 +02:00
parent 3a11a92687
commit d7f167a868
5 changed files with 40 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/aki_prj23_transparenzregister/models/company.py
View File

@ -60,6 +60,14 @@ class DistrictCourt:
name: str
city: str
def to_dict(self) -> dict:
"""Transform to dict.
Returns:
dict: Dictionary
"""
return asdict(self)
@dataclass
class CompanyID:

1
src/aki_prj23_transparenzregister/utils/data_extraction/unternehmensregister/transform.py
View File

@ -1,5 +1,4 @@
"""Transform raw Unternehmensregister export (*.xml) to processed .json files for loading."""
# mypy: ignore-errors
import dataclasses
import glob
import json

17
tests/models/company_test.py
View File

@ -3,19 +3,26 @@
from aki_prj23_transparenzregister.models.company import (
Capital,
CapitalTypeEnum,
Company,
CompanyID,
CompanyTypeEnum,
CurrencyEnum,
DistrictCourt,
Location,
)
def test_to_dict() -> None:
"""Tests if the version tag is entered."""
company_id = CompanyID("The Shire", "420")
district_court = DistrictCourt("abc", "abc")
company_id = CompanyID(district_court=district_court, hr_number="HRB 123")
location = Location(
city="Insmouth", house_number="19", street="Harbor", zip_code="1890"
)
capital = Capital(currency="BTC", type="Virtual assets", value=42)
capital = Capital(
currency=CurrencyEnum.DEUTSCHE_MARK, type=CapitalTypeEnum.GRUNDKAPITAL, value=42 # type: ignore
)
company = Company(
id=company_id,
last_update="Tomorrow",
@ -24,13 +31,13 @@ def test_to_dict() -> None:
relationships=[],
business_purpose="Blockchain and NFTs",
capital=capital,
company_type="Something",
company_type=CompanyTypeEnum.AG, # type: ignore
founding_date="Yesterday",
)
assert company.to_dict() == {
"id": {
"district_court": company_id.district_court,
"district_court": district_court.to_dict(),
"hr_number": company_id.hr_number,
},
"last_update": company.last_update,
@ -48,6 +55,6 @@ def test_to_dict() -> None:
"currency": capital.currency,
"type": capital.type,
},
"company_type": "Something",
"company_type": company.company_type,
"founding_date": "Yesterday",
}

38
tests/utils/data_extraction/unternehmensregister/transform_test.py
View File

@ -50,7 +50,7 @@ def test_parse_stakeholder_org_hidden_in_person() -> None:
"Rolle": {"Rollenbezeichnung": {"content": "Kommanditist(in)"}},
}
expected_result = CompanyToCompanyRelationship(
role=RelationshipRoleEnum.KOMMANDITIST,
role=RelationshipRoleEnum.KOMMANDITIST, # type: ignore
description="Some Company KG",
type=CompanyRelationshipEnum.COMPANY,
location=Location(**{"city": "Area 51"}),
@ -70,7 +70,7 @@ def test_parse_stakeholder_person() -> None:
"Rolle": {"Rollenbezeichnung": {"content": "Geschäftsleiter(in)"}},
}
expected_result = PersonToCompanyRelationship(
role=RelationshipRoleEnum.GESCHAEFTSLEITER,
role=RelationshipRoleEnum.GESCHAEFTSLEITER, # type: ignore
date_of_birth="1947-09-21",
name=PersonName(**{"firstname": "Stephen", "lastname": "King"}),
type=CompanyRelationshipEnum.PERSON,
@ -97,7 +97,7 @@ def test_parse_stakeholder_org() -> None:
}
expected_result = CompanyToCompanyRelationship(
description="Transparenzregister kG",
role=RelationshipRoleEnum.DIREKTOR,
role=RelationshipRoleEnum.DIREKTOR, # type: ignore
type=CompanyRelationshipEnum.COMPANY,
location=Location(
**{
@ -218,7 +218,7 @@ def test_map_rechtsform_from_name() -> None:
def test_map_capital_kg_single() -> None:
capital = Capital(
currency=CurrencyEnum.EURO, value=69000, type=CapitalTypeEnum.HAFTEINLAGE
currency=CurrencyEnum.EURO, value=69000, type=CapitalTypeEnum.HAFTEINLAGE # type: ignore
)
data = {
"XJustiz_Daten": {
@ -239,13 +239,13 @@ def test_map_capital_kg_single() -> None:
}
}
result = transform.map_capital(data, CompanyTypeEnum.KG)
result = transform.map_capital(data, CompanyTypeEnum.KG) # type: ignore
assert result == capital
def test_map_capital_kg_sum() -> None:
capital = Capital(
currency=CurrencyEnum.EURO, value=20000, type=CapitalTypeEnum.HAFTEINLAGE
currency=CurrencyEnum.EURO, value=20000, type=CapitalTypeEnum.HAFTEINLAGE # type: ignore
)
data = {
"XJustiz_Daten": {
@ -274,20 +274,20 @@ def test_map_capital_kg_sum() -> None:
}
}
result = transform.map_capital(data, CompanyTypeEnum.KG)
result = transform.map_capital(data, CompanyTypeEnum.KG) # type: ignore
assert result == capital
def test_map_capital_no_fachdaten() -> None:
data: dict = {"XJustiz_Daten": {"Fachdaten_Register": {}}}
result = transform.map_capital(data, CompanyTypeEnum.KG)
result = transform.map_capital(data, CompanyTypeEnum.KG) # type: ignore
assert result is None
def test_map_capital_gmbh() -> None:
capital = Capital(
currency=CurrencyEnum.DEUTSCHE_MARK, value=42, type=CapitalTypeEnum.STAMMKAPITAL
currency=CurrencyEnum.DEUTSCHE_MARK, value=42, type=CapitalTypeEnum.STAMMKAPITAL # type: ignore
)
data = {
"XJustiz_Daten": {
@ -306,13 +306,13 @@ def test_map_capital_gmbh() -> None:
}
}
result = transform.map_capital(data, CompanyTypeEnum.GMBH)
result = transform.map_capital(data, CompanyTypeEnum.GMBH) # type: ignore
assert result == capital
def test_map_capital_ag() -> None:
capital = Capital(
currency=CurrencyEnum.DEUTSCHE_MARK, value=42, type=CapitalTypeEnum.GRUNDKAPITAL
currency=CurrencyEnum.DEUTSCHE_MARK, value=42, type=CapitalTypeEnum.GRUNDKAPITAL # type: ignore
)
data = {
"XJustiz_Daten": {
@ -333,13 +333,13 @@ def test_map_capital_ag() -> None:
}
}
result = transform.map_capital(data, CompanyTypeEnum.SE)
result = transform.map_capital(data, CompanyTypeEnum.SE) # type: ignore
assert result == capital
def test_map_capital_personengesellschaft() -> None:
capital = Capital(
currency=CurrencyEnum.DEUTSCHE_MARK, value=42, type=CapitalTypeEnum.STAMMKAPITAL
currency=CurrencyEnum.DEUTSCHE_MARK, value=42, type=CapitalTypeEnum.STAMMKAPITAL # type: ignore
)
data = {
"XJustiz_Daten": {
@ -358,13 +358,13 @@ def test_map_capital_personengesellschaft() -> None:
}
}
result = transform.map_capital(data, CompanyTypeEnum.OHG)
result = transform.map_capital(data, CompanyTypeEnum.OHG) # type: ignore
assert result == capital
def test_map_capital_einzelkaufmann() -> None:
capital = Capital(
currency=CurrencyEnum.DEUTSCHE_MARK, value=42, type=CapitalTypeEnum.STAMMKAPITAL
currency=CurrencyEnum.DEUTSCHE_MARK, value=42, type=CapitalTypeEnum.STAMMKAPITAL # type: ignore
)
data = {
"XJustiz_Daten": {
@ -383,13 +383,13 @@ def test_map_capital_einzelkaufmann() -> None:
}
}
result = transform.map_capital(data, CompanyTypeEnum.EINZELKAUFMANN)
result = transform.map_capital(data, CompanyTypeEnum.EINZELKAUFMANN) # type: ignore
assert result is None
def test_map_capital_partial_null_values() -> None:
capital = Capital(
currency=CurrencyEnum.DEUTSCHE_MARK, value=42, type=CapitalTypeEnum.STAMMKAPITAL
currency=CurrencyEnum.DEUTSCHE_MARK, value=42, type=CapitalTypeEnum.STAMMKAPITAL # type: ignore
)
data = {
"XJustiz_Daten": {
@ -408,7 +408,7 @@ def test_map_capital_partial_null_values() -> None:
}
}
result = transform.map_capital(data, CompanyTypeEnum.OHG)
result = transform.map_capital(data, CompanyTypeEnum.OHG) # type: ignore
assert result is None
@ -572,7 +572,7 @@ def test_map_unternehmensregister_json( # noqa: PLR0913
mock_map_company_id: Mock,
) -> None:
expected_result = Company(
**{
**{ # type: ignore
"id": Mock(),
"name": Mock(),
"location": Mock(),

2
tests/utils/mongo/company_mongo_service_test.py
View File

@ -103,7 +103,7 @@ def test_insert(mock_mongo_connector: Mock, mock_collection: Mock) -> None:
mock_collection.insert_one.return_value = mock_result
assert (
service.insert(
Company(CompanyID("", ""), Location("Hier und Dort"), "", "", [])
Company(CompanyID("", ""), Location("Hier und Dort"), "", "", []) # type: ignore
)
== mock_result
)