def test_procesar_csv_carga_reemplaza_anterior(db, usr_unidad_basica,
carga_inicial):
cant_mesas_ok, cant_mesas_parcialmente_ok, errores = CSVImporter(
PATH_ARCHIVOS_TEST + 'info_resultados_ok.csv',
usr_unidad_basica).procesar()
assert cant_mesas_ok == 1
assert cant_mesas_parcialmente_ok == 0
cargas_totales = Carga.objects.filter(tipo=Carga.TIPOS.total)
assert cargas_totales.count() == 2
ids_viejos = list(cargas_totales.values_list('id', flat=True))
# Vuelvo a cargar.
cant_mesas_ok, cant_mesas_parcialmente_ok, errores = CSVImporter(
PATH_ARCHIVOS_TEST + 'info_resultados_ok.csv',
usr_unidad_basica).procesar()
assert cant_mesas_ok == 1
assert cant_mesas_parcialmente_ok == 0
cargas_totales = Carga.objects.filter(tipo=Carga.TIPOS.total)
# No aumentó.
assert cargas_totales.count() == 2
cargas_repetidas = Carga.objects.filter(id__in=ids_viejos)
# No quedó ninguna de las viejas.
assert cargas_repetidas.count() == 0
def importar_ahora_at_once(self, file):
csvimporter = CSVImporter(Path(file), self.usr.user, self.debug)
errores = csvimporter.procesar()
for cant_mesas_ok, cant_mesas_parcialmente_ok, error in errores:
print("Error: ", error)
print(
f"{cant_mesas_ok} mesas ok, {cant_mesas_parcialmente_ok} mesas parcialmente ok. "
)
def importar_ahora(self, file):
csvimporter = CSVImporter(Path(file), self.usr.user, self.debug)
errores = csvimporter.procesar_parcialmente()
try:
for cant_mesas_ok, cant_mesas_parcialmente_ok, error in errores:
print("Error: ", error)
except TypeError:
cant_mesas_ok, cant_mesas_parcialmente_ok, error = errores
print("Errro: ", error)
print(
f"{cant_mesas_ok} mesas ok, {cant_mesas_parcialmente_ok} mesas parcialmente ok. "
)
def test_procesar_csv_informacion_valida_genera_resultados_salvo_totales(
db, usr_unidad_basica, carga_inicial):
cant_mesas_ok, cant_mesas_parcialmente_ok, errores = CSVImporter(
PATH_ARCHIVOS_TEST + 'info_resultados_ok.csv',
usr_unidad_basica).procesar()
assert cant_mesas_ok == 1
assert cant_mesas_parcialmente_ok == 0
cargas_totales = Carga.objects.filter(tipo=Carga.TIPOS.total)
assert cargas_totales.count() == 0
votos_carga_total = VotoMesaReportado.objects.filter(
carga__in=cargas_totales).all()
assert votos_carga_total.count() == 0
cargas_parciales = Carga.objects.filter(tipo=Carga.TIPOS.parcial)
# Hay una sola categoría no prioritaria.
assert cargas_parciales.count() == len(CATEGORIAS) - 1
for parcial in cargas_parciales:
assert parcial.origen == 'csv'
votos_carga_parcial = VotoMesaReportado.objects.filter(
carga__in=cargas_parciales).all()
# Cada cat tiene 2 partidos + total + blancos + nulos + sobres = 6
assert votos_carga_parcial.count() == (len(CATEGORIAS) - 1) * 6
def test_falta_total_de_votos(db, usr_unidad_basica, carga_inicial):
cant_mesas_ok, cant_mesas_parcialmente_ok, errores = CSVImporter(
PATH_ARCHIVOS_TEST + 'falta_total_votos.csv',
usr_unidad_basica).procesar()
assert cant_mesas_ok == 0
assert cant_mesas_parcialmente_ok == 0
assert f"Faltan las opciones: ['{Opcion.total_votos().nombre}'] en la mesa" in errores
assert Carga.objects.count() == 0
def test_caracteres_alfabeticos_en_votos(db, usr_unidad_basica, carga_inicial):
cant_mesas_ok, cant_mesas_parcialmente_ok, errores = CSVImporter(
PATH_ARCHIVOS_TEST + 'valores_texto_en_votos.csv',
usr_unidad_basica).procesar()
assert cant_mesas_ok == 0
assert cant_mesas_parcialmente_ok == 1
assert 'Los resultados deben ser números enteros positivos.' in errores
assert Carga.objects.count() == 0
def test_validar_csv_mesas_invalidas(db, usr_unidad_basica):
cant_mesas_ok, cant_mesas_parcialmente_ok, errores = CSVImporter(
PATH_ARCHIVOS_TEST + 'mesas_invalidas.csv',
usr_unidad_basica).procesar()
assert 'No existe mesa' in errores
assert cant_mesas_ok == 0
assert cant_mesas_parcialmente_ok == 0
assert Carga.objects.count() == 0
def test_procesar_csv_resultados_negativos(db, usr_unidad_basica,
carga_inicial):
cant_mesas_ok, cant_mesas_parcialmente_ok, errores = CSVImporter(
PATH_ARCHIVOS_TEST + 'info_resultados_negativos.csv',
usr_unidad_basica).procesar()
assert cant_mesas_ok == 0
assert cant_mesas_parcialmente_ok == 0
assert 'Los resultados deben ser números enteros positivos' in errores
assert Carga.objects.count() == 0
def test_falta_jpc_en_carga_total(db, usr_unidad_basica, carga_inicial):
cant_mesas_ok, cant_mesas_parcialmente_ok, errores = CSVImporter(
PATH_ARCHIVOS_TEST + 'falta_jpc_carga_total.csv',
usr_unidad_basica).procesar()
assert cant_mesas_ok == 0
assert cant_mesas_parcialmente_ok == 1
assert "Los resultados para la carga total de la categoría Intendentes, Concejales y Consejeros Escolares deben estar completos. " \
"Faltan las opciones: ['JpC'] en la mesa" in errores
assert Carga.objects.count() == len(CATEGORIAS) - 1
def test_acumula_errores(db, usr_unidad_basica, carga_inicial):
cant_mesas_ok, cant_mesas_parcialmente_ok, errores = CSVImporter(
PATH_ARCHIVOS_TEST + 'acumula_errores.csv',
usr_unidad_basica).procesar()
assert cant_mesas_ok == 0
assert cant_mesas_parcialmente_ok == 0
assert 'Los resultados deben ser números enteros positivos.' in errores
assert "Faltan las opciones: ['JpC'] en la mesa" in errores
assert Carga.objects.count() == 0
def test_falta_jpc_en_carga_parcial(db, usr_unidad_basica, carga_inicial):
settings.OPCIONES_CARGAS_TOTALES_COMPLETAS = False
cant_mesas_ok, cant_mesas_parcialmente_ok, errores = CSVImporter(
PATH_ARCHIVOS_TEST + 'falta_jpc_carga_parcial.csv',
usr_unidad_basica).procesar()
assert cant_mesas_ok == 0
assert cant_mesas_parcialmente_ok == 1
assert "Faltan las opciones: ['JpC'] en la mesa" in errores
assert Carga.objects.count() == 1
def test_procesar_csv_opciones_no_encontradas(db, usr_unidad_basica,
carga_inicial):
cant_mesas_ok, cant_mesas_parcialmente_ok, errores = CSVImporter(
PATH_ARCHIVOS_TEST + 'opciones_invalidas.csv',
usr_unidad_basica).procesar()
assert cant_mesas_ok == 0
assert cant_mesas_parcialmente_ok == 0
assert 'El número de lista C2019 no fue encontrado' in errores
assert Carga.objects.count() == 0
def test_procesar_csv_otros_separadores(db, usr_unidad_basica, carga_inicial):
cant_mesas_ok, cant_mesas_parcialmente_ok, errores = CSVImporter(
PATH_ARCHIVOS_TEST +
'info_resultados_ok_separados_por_punto_y_coma.csv',
usr_unidad_basica).procesar()
assert cant_mesas_ok == 1
assert cant_mesas_parcialmente_ok == 0
cargas_totales = Carga.objects.filter(tipo=Carga.TIPOS.total)
assert cargas_totales.count() == 2
cant_mesas_ok, cant_mesas_parcialmente_ok, errores = CSVImporter(
PATH_ARCHIVOS_TEST + 'info_resultados_ok_separados_por_tab.csv',
usr_unidad_basica).procesar()
assert cant_mesas_ok == 1
assert cant_mesas_parcialmente_ok == 0
cargas_totales = Carga.objects.filter(tipo=Carga.TIPOS.total)
assert cargas_totales.count() == 2
def test_validar_csv_fiscal_sin_permisos_suficientes(db):
user = UserFactory()
FiscalFactory(user=user)
Group.objects.create(name='unidades basicas')
g_visualizadores = Group.objects.create(name='visualizadores')
user.groups.add(g_visualizadores)
with pytest.raises(PermisosInvalidosError) as e:
CSVImporter(PATH_ARCHIVOS_TEST + 'info_resultados_negativos.csv',
user).validar()
assert 'Su usuario no tiene los permisos necesarios' in str(e.value)
def test_procesar_csv_hace_importacion_parcial(db, usr_unidad_basica,
carga_inicial):
cant_mesas_ok, cant_mesas_parcialmente_ok, errores = CSVImporter(
PATH_ARCHIVOS_TEST + 'info_resultados_ok_mas_error.csv',
usr_unidad_basica).procesar()
assert cant_mesas_ok == 1
assert cant_mesas_parcialmente_ok == 0
assert 'No existe mesa 4013' in errores
cargas_totales = Carga.objects.filter(tipo=Carga.TIPOS.total)
assert cargas_totales.count() == 2
def worker_import_file(self, tarea):
"""
Hace la importación de un archivo propiamente dicha.
"""
path = self.determinar_path(tarea)
if not path:
mensaje = f"archivo {tarea.csv_file.name} no encontrado."
self.logger.error("[%d] Tarea %s abortada: %s",
self.thread_local.worker_id, tarea, mensaje)
tarea.errores = mensaje
tarea.save_errores()
tarea.fin_procesamiento(0, 0)
return
csvimporter = CSVImporter(path, tarea.fiscal.user, self.debug)
errores = csvimporter.procesar_parcialmente()
i = 0
if not tarea.errores:
tarea.errores = ''
try:
for cant_mesas_ok, cant_mesas_parcialmente_ok, error in errores:
if not error:
continue
tarea.errores = tarea.errores + error
i += 1
if i == 20:
# Cada 20 errores grabamos.
i = 0
tarea.save_errores(cant_mesas_ok,
cant_mesas_parcialmente_ok)
except TypeError:
cant_mesas_ok, cant_mesas_parcialmente_ok, error = errores
tarea.errores = tarea.errores + error
i += 1
# Si quedaron errores sin grabar los grabamos:
if i > 0:
tarea.save_errores()
tarea.fin_procesamiento(cant_mesas_ok, cant_mesas_parcialmente_ok)
def test_procesar_csv_acepta_metadata_opcional(db, usr_unidad_basica,
carga_inicial):
cant_mesas_ok, cant_mesas_parcialmente_ok, errores = CSVImporter(
PATH_ARCHIVOS_TEST + 'info_resultados_ok_con_metadata_optativa.csv',
usr_unidad_basica).procesar()
assert cant_mesas_ok == 1
assert cant_mesas_parcialmente_ok == 0
cargas_totales = Carga.objects.filter(tipo=Carga.TIPOS.total)
assert cargas_totales.count() == 2
cargas_parciales = Carga.objects.filter(tipo=Carga.TIPOS.parcial)
assert cargas_parciales.count() == len(CATEGORIAS) - 1
def test_procesar_csv_sanitiza_ok(db, usr_unidad_basica, carga_inicial):
fdt = Opcion.objects.get(nombre='FdT')
fdt.codigo = '136'
fdt.save()
cant_mesas_ok, cant_mesas_parcialmente_ok, errores = CSVImporter(
PATH_ARCHIVOS_TEST + 'info_resultados_ok_con_sanitizar.csv',
usr_unidad_basica).procesar()
assert cant_mesas_ok == 1
assert cant_mesas_parcialmente_ok == 0
cargas_totales = Carga.objects.filter(tipo=Carga.TIPOS.total)
# Debería haber 2 cargas total: Int (que no es prio), y presi, que es prio pero tiene
# además opción no prioritaria.
assert cargas_totales.count() == 2
def test_procesar_csv_categorias_faltantes_en_archivo(db, usr_unidad_basica):
CategoriaGeneralFactory()
d1 = DistritoFactory(numero=1)
s1 = SeccionFactory(numero=50, distrito=d1)
c1 = CircuitoFactory(numero='2', seccion=s1)
m = MesaFactory(numero='4012',
lugar_votacion__circuito=c1,
electores=100,
circuito=c1)
o2 = OpcionFactory(codigo='Todes')
o3 = OpcionFactory(codigo='Juntos')
c = CategoriaFactory(opciones=[o2, o3], nombre='Otra categoria')
MesaCategoriaFactory(mesa=m, categoria=c)
cant_mesas_ok, cant_mesas_parcialmente_ok, errores = CSVImporter(
PATH_ARCHIVOS_TEST + 'info_resultados_negativos.csv',
usr_unidad_basica).procesar()
assert cant_mesas_ok == 0
assert cant_mesas_parcialmente_ok == 0
assert 'Faltan datos en el archivo de la siguiente categoría' in errores
assert Carga.objects.count() == 0
def test_procesar_csv_informacion_valida_copia_parciales_a_totales(
db, usr_unidad_basica, carga_inicial):
cant_mesas_ok, cant_mesas_parcialmente_ok, errores = CSVImporter(
PATH_ARCHIVOS_TEST + 'info_resultados_copia_parciales_a_totales.csv',
usr_unidad_basica).procesar()
assert cant_mesas_parcialmente_ok == 0
assert cant_mesas_ok == 1
cargas_totales = Carga.objects.filter(tipo=Carga.TIPOS.total).all()
cargas_parciales = Carga.objects.filter(tipo=Carga.TIPOS.parcial).all()
# Todo lo que está en carga total también está en carga parcial para la misma categoría.
for carga_parcial in cargas_parciales:
cargas_totales_misma_mc = cargas_totales.filter(
mesa_categoria=carga_parcial.mesa_categoria)
if cargas_totales_misma_mc.count() == 0:
continue
carga_total_misma_mc = cargas_totales_misma_mc.first()
for voto in carga_parcial.reportados.all():
assert VotoMesaReportado.objects.filter(
carga=carga_total_misma_mc,
votos=voto.votos,
opcion=voto.opcion).exists()
def test_validar_csv_columnas_duplicadas(usr_unidad_basica):
with pytest.raises(ColumnasInvalidasError):
CSVImporter(PATH_ARCHIVOS_TEST + 'columnas_duplicadas.csv',
usr_unidad_basica).validar()
assert Carga.objects.count() == 0
def test_validar_csv_fiscal_no_encontrado(db):
user = UserFactory()
with pytest.raises(PermisosInvalidosError) as e:
CSVImporter(PATH_ARCHIVOS_TEST + 'info_resultados_negativos.csv',
user).validar()
assert 'Fiscal no encontrado' in str(e.value)