test_formato_dbf : initial tests

Signed-off-by: SONIABHISHEK121 <[email protected]>

tests/test_formato_dbf.py

@@ -0,0 +1,211 @@
+"""
+Pruebas para el módulo `formato_dbf` del paquete `pyafipws`.
+
+Este módulo contiene pruebas para las funciones `definir_campos`, `dar_nombre_campo`, `leer` y `escribir` en el módulo `formato_dbf`.
+Las pruebas utilizan datos simulados y fixtures para verificar el comportamiento esperado de estas funciones.
+
+La función `test_definir_campos` prueba la función `definir_campos`, que define los nombres de los campos y los formatos de los campos para un archivo DBF.
+
+La función `test_dar_nombre_campo` prueba la función `dar_nombre_campo`, que genera un nombre de campo abreviado a partir de un nombre de campo largo.
+
+La función `test_leer` prueba la función `leer`, que lee datos de uno o más archivos DBF y devuelve un diccionario que contiene los datos.
+
+La función `test_escribir` prueba la función `escribir`, que escribe datos en un archivo DBF.
+
+La función `test_ayuda` prueba la función `ayuda`, que imprime un mensaje de ayuda describiendo el formato del archivo DBF.
+"""
+import os
+import pytest
+from pyafipws.formatos import formato_dbf
+from unittest.mock import Mock
+
+# Datos de prueba y fixtures
+TEST_DATA_DIR = "tests/data"
+TEST_DBF_FILES = {
+    "encabezado": os.path.join(TEST_DATA_DIR, "encabeza.dbf"),
+    "detalle": os.path.join(TEST_DATA_DIR, "detalle.dbf"),
+    # Agregar más archivos DBF de prueba según sea necesario
+}
+
+# Prueba de definir_campos
+@pytest.mark.dontusefix
+def test_definir_campos():
+    test_formato = [
+        ("campo1", 10, formato_dbf.N),
+        ("campo2", 20, formato_dbf.A)
+    ]
+    claves, campos = formato_dbf.definir_campos(test_formato)
+
+    # Asegurar la salida esperada
+    assert claves == ["campo1", "campo2"]
+    assert campos[0] == "campo1 N(10,0)"
+    assert campos[1] == "campo2 C(20)"
+
+# Prueba de definir_campos con formato vacío
+@pytest.mark.dontusefix
+def test_definir_campos_empty():
+    claves, campos = formato_dbf.definir_campos([])
+    assert claves == []
+    assert campos == []
+
+# Prueba de definir_campos con nombres de campos largos
+@pytest.mark.dontusefix
+def test_definir_campos_long_field():
+    test_formato = [
+        ("campo1", 300, formato_dbf.N),  # Longitud del campo > 250
+        ("campo2", 20, formato_dbf.A)
+    ]
+    claves, campos = formato_dbf.definir_campos(test_formato)
+    assert campos[0] == "campo1 M"  # Campo de tipo Memo
+
+# Prueba de definir_campos con campo decimal
+@pytest.mark.dontusefix
+def test_definir_campos_decimal_field():
+    test_formato = [
+        ("campo1", (10, 2), formato_dbf.N),  # Campo numérico con 2 decimales
+        ("campo2", 20, formato_dbf.A)
+    ]
+    claves, campos = formato_dbf.definir_campos(test_formato)
+    assert campos[0] == "campo1 N(10,0)"  # Campo numérico con 2 decimales
+
+# Prueba de definir_campos con campo de caracteres
+@pytest.mark.dontusefix
+def test_definir_campos_character_field():
+    test_formato = [
+        ("campo1", 10, formato_dbf.A),  # Campo de caracteres
+        ("campo2", 20, formato_dbf.A)
+    ]
+    claves, campos = formato_dbf.definir_campos(test_formato)
+    assert campos[0] == "campo1 C(10)"  # Campo de caracteres
+
+# Prueba de definir_campos con campo numérico
+@pytest.mark.dontusefix
+def test_definir_campos_numeric_field():
+    test_formato = [
+        ("campo1", 10, formato_dbf.N),  # Campo numérico
+        ("campo2", 20, formato_dbf.A)
+    ]
+    claves, campos = formato_dbf.definir_campos(test_formato)
+    assert campos[0] == "campo1 N(10,0)"  # Campo numérico
+
+# Prueba de dar_nombre_campo
+@pytest.mark.dontusefix
+@pytest.mark.parametrize("clave, expected", [
+    ("campo_largo", "campolargo"),
+    ("Dato_adicional1", "datoadic01"),
+    ("campo_corto", "campocorto"),
+])
+def test_dar_nombre_campo(clave, expected):
+    assert formato_dbf.dar_nombre_campo(clave) == expected
+
+@pytest.mark.dontusefix
+def test_leer_empty_dbf_files(monkeypatch):
+    # Simular la clase dbf.Table para devolver un objeto mock vacío
+    mock_table = MockTable()
+    monkeypatch.setattr("pyafipws.formatos.formato_dbf.dbf.Table", lambda *args, **kwargs: mock_table)
+
+    # Llamar a leer y asegurar la salida
+    output = formato_dbf.leer(TEST_DBF_FILES, TEST_DATA_DIR)
+
+    # Si los archivos DBF están vacíos, la salida debe ser un diccionario vacío
+    assert output == {}
+
+@pytest.mark.dontusefix
+def test_leer_non_empty_dbf_files(monkeypatch):
+    # Simular la clase dbf.Table para devolver un objeto mock con datos
+    mock_table = MockTable()
+    mock_table.records = [MockRecord({"campo1": 123, "campo2": "test_value"})]
+    monkeypatch.setattr("pyafipws.formatos.formato_dbf.dbf.Table", lambda *args, **kwargs: mock_table)
+
+    # Llamar a leer y asegurar la salida
+    output = formato_dbf.leer(TEST_DBF_FILES, TEST_DATA_DIR)
+
+    # Si los archivos DBF no están vacíos, asegurar que la salida no está vacía
+    assert len(output) > 0
+    first_key = list(output.keys())[0]
+    assert "detalles" in output[first_key]
+
+# Prueba de leer con múltiples archivos DBF
+@pytest.mark.dontusefix
+def test_leer_multiple_dbf_files(monkeypatch):
+    # Simular la clase dbf.Table para devolver un objeto mock con datos
+    mock_table = MockTable()
+    mock_table.records = [
+        MockRecord({"campo1": 123, "campo2": "test_value"}),
+        MockRecord({"campo1": 456, "campo2": "another_value"})
+    ]
+    monkeypatch.setattr("pyafipws.formatos.formato_dbf.dbf.Table", lambda *args, **kwargs: mock_table)
+
+    # Llamar a leer y asegurar la salida
+    output = formato_dbf.leer(TEST_DBF_FILES, TEST_DATA_DIR)
+
+    # Asegurar que la salida es un diccionario con múltiples claves
+    assert len(output) == 2
+    assert "detalles" in output["encabezado"]
+    assert "detalles" in output["detalle"]
+
+@pytest.mark.dontusefix
+def test_escribir(tmp_path, monkeypatch):
+    # Crear un directorio temporal para pruebas
+    test_dir = tmp_path / "test_dbf"
+    test_dir.mkdir()
+
+    # Simular la clase dbf.Table para rastrear los datos escritos
+    written_data = []
+    mock_table = MockTable(written_data)
+    monkeypatch.setattr("pyafipws.formatos.formato_dbf.dbf.Table", lambda *args, **kwargs: mock_table)
+
+    # Definir datos de prueba y llamar a escribir
+    test_data = [{
+        "id": 1,
+        "detalles": [{"campo1": 123, "campo2": "test_value"}],
+        "ivas": [],
+        "tributos": [],
+        "permisos": [],
+        "cbtes_asoc": [],
+        "datos": []
+    }]
+    formato_dbf.escribir(test_data, {"encabezado": "encabeza.dbf"}, str(test_dir))
+
+    # Asegurar que los datos fueron escritos en el archivo DBF
+    assert len(written_data) == 2
+    assert written_data[0] == {'tiporeg': 0, 'webservice': b'', 'fechacbte': b'', 'tipocbte': 0, 'puntovta': 0, 'cbtenro': 0, 'tipoexpo': 0, 'permisoexi': b'', 'paisdstcmp': 0, 'nombreclie': b'', 'tipodoc': 0, 'nrodoc': 0, 'domicilioc': b'', 'idimpositi': b'', 'imptotal': 0, 'imptotconc': 0, 'impneto': 0, 'imptoliq': 0, 'imptoliqnr': 0, 'impopex': 0, 'imptoperc': 0, 'impiibb': 0, 'imptopercm': 0, 'impinterno': 0, 'imptrib': 0, 'monedaid': b'', 'monedactz': 0, 'obscomerci': b'', 'obsgeneral': b'', 'formapago': b'', 'incoterms': b'', 'incotermsd': b'', 'idiomacbte': b'', 'zona': b'', 'fechavencp': b'', 'prestaserv': 0, 'fechaservd': b'', 'fechaservh': b'', 'cae': b'', 'fechavto': b'', 'resultado': b'', 'reproceso': b'', 'motivosobs': b'', 'id': 1, 'telefonocl': b'', 'localidadc': b'', 'provinciac': b'', 'formatoid': 0, 'email': b'', 'pdf': b'', 'errcode': b'', 'errmsg': b'', 'datoadic01': b'', 'datoadic02': b'', 'datoadic03': b'', 'datoadic04': b'', 'descuento': 0, 'cbtdesde': 0, 'cbthasta': 0, 'concepto': 0, 'nousar': 0, 'impiva': 0, 'emisiontip': b'', 'impsubtota': 0, 'cativa': 0, 'tipocodaut': b''}
+    assert written_data[1] == {'id': 1, 'tiporeg': 0, 'codigo': b'', 'qty': 0, 'umed': 0, 'precio': 0, 'importe': 0, 'ivaid': 0, 'ds': b'', 'ncm': b'', 'sec': b'', 'bonif': 0, 'impiva': 0, 'despacho': b'', 'umtx': 0, 'codmtx': 0, 'datoa': b'', 'datob': b'', 'datoc': b'', 'datod': b'', 'datoe': b''}
+
+# Prueba de ayuda (se puede capturar la salida y asegurar el contenido esperado)
+@pytest.mark.dontusefix
+def test_ayuda(capsys):
+    formato_dbf.ayuda()
+    captured = capsys.readouterr()
+    assert "Formato DBF:" in captured.out
+    assert "Encabezado" in captured.out
+    assert "Detalle Item" in captured.out
+
+class MockRecord:
+    def __init__(self, data):
+        self.data = data
+
+    def scatter_fields(self):
+        return self.data
+
+class MockTable:
+    def __init__(self, records=None):
+        self.records = records or []
+        self.current_index = 0
+
+    def __iter__(self):
+        return self
+
+    def __next__(self):
+        if self.current_index >= len(self.records):
+            raise StopIteration
+        record = self.records[self.current_index]
+        self.current_index += 1
+        return record
+
+    def append(self, data):
+        record = MockRecord(data)
+        self.records.append(record)
+
+    def close(self):
+        pass