fix old test without family info

prefeitura-rio · Jun 26, 2024 · 11404da · 11404da
1 parent bb9cc51
commit 11404da
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/api/app/routers/entities_mrg.py b/api/app/routers/entities_mrg.py
@@ -265,7 +265,7 @@ async def create_or_update_professionals(
             if cbo.get("id_cbo_familia") not in occupation_family_ids:
                 new_occupation_families.append(
                     OccupationFamily(
-                        code=cbo.get("id_cbo_familia", cbo.get("id_cbo")[:4]),
+                        code=cbo.get("id_cbo_familia"),
                         name=cbo.get("cbo_familia")
                     )
                 )

diff --git a/api/tests/test_mrg.py b/api/tests/test_mrg.py
@@ -212,7 +212,9 @@ async def test_create_or_update_mrgprofessionals(
                 "nome": "PESSOA QUALQUER",
                 "cbo": [{
                     "id_cbo": "111415",
-                    "cbo": "Teste"
+                    "cbo": "Teste",
+                    "id_cbo_familia": "1114",
+                    "cbo_familia": "TESTE"
                 }],
                 "conselho": [{
                     "id_registro_conselho": "1111111",