"""Testing apps/ingest.py."""
import json
import tempfile
from unittest.mock import Mock, patch

from aki_prj23_transparenzregister.apps import ingest


def test_import() -> None:
    assert ingest


def test_load_schedule() -> None:
    with tempfile.TemporaryDirectory() as temp_dir:
        data = {"test": "test"}
        path = f"{temp_dir}/schedule.json"
        with open(path, "w") as file:
            json.dump({"test": "test"}, file)
        assert ingest.load_schedule(path) == data


def test_load_scheduler_no_result() -> None:
    assert ingest.load_schedule("./hello_there.json") == {}


def test_save_schedule() -> None:
    with tempfile.TemporaryDirectory() as temp_dir:
        data = {"test": "test"}
        path = f"{temp_dir}/schedule.json"

        ingest.save_schedule(data, path)
        with open(path) as file:
            assert json.load(file) == data


@patch("aki_prj23_transparenzregister.apps.ingest.find_missing_companies_main")
@patch("aki_prj23_transparenzregister.apps.ingest.enrich_company_financials_main")
def test_main(mock_financials: Mock, mock_find_missing: Mock) -> None:
    ingest.main(Mock())
    assert mock_financials.called
    assert mock_find_missing.called