test.py

import os

import pandas as pd
import pytest

from queries.process_all_db import (
    all_dbs,
    process_all_tech_db,
    process_all_tech_db_with_stats,
)
from utils.utils_vars import UtilsVars


class TestProcessAllDB:
    def setup_method(self):
        UtilsVars.all_db_dfs = []
        UtilsVars.final_all_database = None

    def test_all_dbs(self):
        filepath = r"C:\Users\David\Documents\PROJECTS\2023\PROJET 2023\DUMP\DUMP\NOVEMBRE\20241127_21145_27112024_Dump.xml.gz.xlsb"
        all_dbs(filepath)
        assert len(UtilsVars.all_db_dfs) == 7
        assert isinstance(UtilsVars.all_db_dfs[0], pd.DataFrame)

    def test_process_all_tech_db(self):
        filepath = r"C:\Users\David\Documents\PROJECTS\2023\PROJET 2023\DUMP\DUMP\NOVEMBRE\20241127_21145_27112024_Dump.xml.gz.xlsb"
        process_all_tech_db(filepath)
        assert UtilsVars.final_all_database is not None

    def test_process_all_tech_db_with_stats(self):
        filepath = r"C:\Users\David\Documents\PROJECTS\2023\PROJET 2023\DUMP\DUMP\NOVEMBRE\20241127_21145_27112024_Dump.xml.gz.xlsb"
        process_all_tech_db_with_stats(filepath)
        assert UtilsVars.final_all_database is not None

    # def test_all_dbs_empty_file(self):
    #     filepath = r"C:\Users\HP\Desktop\LTE\PROJET 2023\DUMP\2024\SEPTEMBRE\empty.xlsb"
    #     all_dbs(filepath)
    #     assert len(UtilsVars.all_db_dfs) == 0

    # def test_process_all_tech_db_empty_file(self):
    #     filepath = r"C:\Users\HP\Desktop\LTE\PROJET 2023\DUMP\2024\SEPTEMBRE\empty.xlsb"
    #     process_all_tech_db(filepath)
    #     assert UtilsVars.final_all_database is None

    # def test_process_all_tech_db_with_stats_empty_file(self):
    #     filepath = r"C:\Users\HP\Desktop\LTE\PROJET 2023\DUMP\2024\SEPTEMBRE\empty.xlsb"
    #     process_all_tech_db_with_stats(filepath)
    #     assert UtilsVars.final_all_database is None