altera get_fields para ser público (#17)

imopay_wrapper/models/base.py

@@ -17,7 +17,7 @@ def _init_nested_fields(self):
         pass
 
     @classmethod
-    def __get_fields(cls):
+    def get_fields(cls):
         """
         Método para retornar todos os campos!
         """
@@ -30,7 +30,7 @@ def __get_field(self, name):
         """
         try:
             # noinspection PyUnresolvedReferences
-            return self.__get_fields()[name]
+            return self.get_fields()[name]
         except KeyError as e:
             raise AttributeError(f"Não existe o campo {name} em {self}") from e
 
@@ -104,7 +104,7 @@ def from_dict(cls, data: Union[dict, Any]):
 
         missing_fields = {
             field_name
-            for field_name in cls.__get_fields().keys()
+            for field_name in cls.get_fields().keys()
             if field_name not in data.keys()
         }
 
@@ -116,7 +116,7 @@ def from_dict(cls, data: Union[dict, Any]):
 
     def to_dict(self):
         data = {}
-        for field_name, field in self.__get_fields().items():
+        for field_name, field in self.get_fields().items():
             value = getattr(self, field_name)
 
             if self.__is_empty_value(value):

tests/models/test_base.py

@@ -178,11 +178,7 @@ def test_get_field_1(self):
 
         mocked_field = MagicMock()
 
-        obj = MagicMock(
-            _BaseImopayObj__get_fields=MagicMock(
-                return_value={field_name: mocked_field}
-            )
-        )
+        obj = MagicMock(get_fields=MagicMock(return_value={field_name: mocked_field}))
 
         expected = mocked_field
 
@@ -205,9 +201,7 @@ def test_get_field_2(self):
 
         mocked_field = MagicMock()
 
-        obj = MagicMock(
-            _BaseImopayObj__get_fields=MagicMock(return_value={"bar": mocked_field})
-        )
+        obj = MagicMock(get_fields=MagicMock(return_value={"bar": mocked_field}))
 
         with self.assertRaises(AttributeError):
             BaseImopayObj._BaseImopayObj__get_field(obj, field_name)