import glob
import os
import subprocess
import unittest

from pathlib import Path

from hub.exports.energy_building_exports_factory import EnergyBuildingsExportsFactory
from hub.exports.exports_factory import ExportsFactory
from hub.imports.construction_factory import ConstructionFactory
from hub.imports.energy_systems_factory import EnergySystemsFactory
from hub.imports.geometry_factory import GeometryFactory
from hub.imports.results_factory import ResultFactory
from hub.imports.usage_factory import UsageFactory
from hub.helpers.dictionaries import Dictionaries

from costs.cost import Cost
from costs.constants import SKIN_RETROFIT, SKIN_RETROFIT_AND_SYSTEM_RETROFIT_AND_PV, SYSTEM_RETROFIT_AND_PV


class UnitTests(unittest.TestCase):
  def setUp(self) -> None:
    city_file = Path("./data/test.geojson").resolve()
    output_path = Path('./output/').resolve()
    city = GeometryFactory('geojson',
                           city_file,
                           height_field='citygml_me',
                           year_of_construction_field='ANNEE_CONS',
                           function_field='CODE_UTILI',
                           function_to_hub=Dictionaries().montreal_function_to_hub_function).city
    ConstructionFactory('nrcan', city).enrich()
    UsageFactory('nrcan', city).enrich()
    ExportsFactory('sra', city, output_path).export()
    sra_file = str((output_path / f'{city.name}_sra.xml').resolve())
    subprocess.run(['sra', sra_file])
    ResultFactory('sra', city, output_path).enrich()

    for building in city.buildings:
      building.energy_systems_archetype_name = 'system 1 gas pv'
    EnergySystemsFactory('montreal_custom', city).enrich()
    EnergyBuildingsExportsFactory('insel_monthly_energy_balance', city, output_path).export()
    _insel_files = glob.glob(f'{output_path}/*.insel')
    for insel_file in _insel_files:
      subprocess.run(['insel', str(insel_file)], stdout=subprocess.DEVNULL)
    ResultFactory('insel_monthly_energy_balance', city, output_path).enrich()
    self._city = city

  def test_current_status(self):
    for building in self._city.buildings:
      result = Cost(building).life_cycle
      self.assertIsNotNone(result)
      self.assertEqual(1, result.values[0])

  def test_scenario_1(self):
    for building in self._city.buildings:
      result = Cost(building, retrofit_scenario=SKIN_RETROFIT).life_cycle
      self.assertIsNotNone(result)

  def test_scenario_2(self):
    for building in self._city.buildings:
      result = Cost(building, retrofit_scenario=SYSTEM_RETROFIT_AND_PV).life_cycle
      self.assertIsNotNone(result)
      self.assertEqual(0, result.values[0])

  def test_scenario_3(self):
    for building in self._city.buildings:
      result = Cost(building, retrofit_scenario=SKIN_RETROFIT_AND_SYSTEM_RETROFIT_AND_PV).life_cycle
      self.assertIsNotNone(result)

  def tearDown(self):
    files = glob.glob('output/[!.]*')
    for file in files:
      os.unlink(file)