def test_fixtureCreation(self):
with orm.db_session:
plantsNS = ns.loads("""\
plants:
- plant:
name: alcolea
codename: SCSOM04
description: la bonica planta
meters:
- meter:
name: '1234578'
inverters:
- inverter:
name: '5555'
- inverter:
name: '6666'
irradiationSensors:
- irradiationSensor:
name: alberto
temperatureModuleSensors:
- temperatureModuleSensor:
name: pol
temperatureAmbientSensors:
- temperatureAmbientSensor:
name: joana
""")
importPlants(self.pony.db, plantsNS)
#TODO test the whole fixture, not just the plant data
expectedPlantns = exportPlants(self.pony.db)
self.assertNsEqual(expectedPlantns, plantsNS)
def test_Plant_importPlants__manyPlants(self):
plantsns = self.samplePlantsNS()
importPlants(self.pony.db, plantsns)
resultPlantns = exportPlants(self.pony.db, skipEmpty=True)
self.assertNsEqual(plantsns, resultPlantns)
def test__importPlant_File__withMunicipalities(self):
fakePlantsYaml = 'fakeplant.yaml'
content = """\
municipalities:
- municipality:
name: Figueres
ineCode: '17066'
countryCode: ES
country: Spain
regionCode: '08'
region: Catalonia
provinceCode: '17'
province: Girona
plants:
- plant:
name: alcolea
codename: SCSOM04
description: la bonica planta
meters:
- meter:
name: '1234578'
inverters:
- inverter:
name: '5555'
- inverter:
name: '6666'
irradiationSensors:
- irradiationSensor:
name: alberto
temperatureAmbientSensors:
- temperatureAmbientSensor:
name: joana
temperatureModuleSensors:
- temperatureModuleSensor:
name: pol"""
p = Path(fakePlantsYaml)
with p.open("w", encoding="utf-8") as f:
f.write(content)
importPlantsFromFile(self.pony.db, fakePlantsYaml)
p.unlink()
with orm.db_session:
plantns = exportPlants(self.pony.db)
self.assertNsEqual(plantns, content)
def test_Plant_importPlants__ManyMunicipalities(self):
plantsns = ns()
municipalitiesns = ns.loads("""\
- municipality:
name: Figueres
ineCode: '17066'
- municipality:
name: Girona
ineCode: '17003'
""")
plantsns['municipalities'] = municipalitiesns
importPlants(self.pony.db, plantsns)
storedMunicipalitiesns = exportPlants(self.pony.db)
plantsns['plants'] = []
self.assertNsEqual(plantsns, storedMunicipalitiesns)
def test__importPlantCLI_File(self):
fakePlantYaml = 'fakeplant.yaml'
content = """\
plants:
- plant:
name: alcolea
codename: SCSOM04
description: la bonica planta
meters:
- meter:
name: '1234578'
inverters:
- inverter:
name: '5555'
- inverter:
name: '6666'
irradiationSensors:
- irradiationSensor:
name: alberto
temperatureAmbientSensors:
- temperatureAmbientSensor:
name: joana
temperatureModuleSensors:
- temperatureModuleSensor:
name: pol"""
p = Path(fakePlantYaml)
with p.open("w", encoding="utf-8") as f:
f.write(content)
runner = CliRunner()
result = runner.invoke(importPlantCLI, [fakePlantYaml])
print(result)
self.assertEqual(0, result.exit_code)
p.unlink()
with orm.db_session:
plantns = exportPlants(self.pony.db)
self.assertNsEqual(plantns, content)
def test_Plant_importPlants__manyPlants_ManyMunicipalities(self):
plantsns = self.samplePlantsNS()
# municipalitiesns = ns.loads("""\
# - municipality:
# name: Figueres
# ineCode: '17066'
# - municipality:
# name: Girona
# ineCode: '17003'
# """)
# plantsns['municipalities'] = municipalitiesns
importPlants(self.pony.db, plantsns)
# orm.flush()
resultPlantns = exportPlants(self.pony.db, skipEmpty=True)
self.assertNsEqual(plantsns, resultPlantns)
def test_Municipality__importMunicipality__OneMunicipality(self):
plantsns = ns()
municipalitiesns = ns.loads("""\
- municipality:
name: Figueres
ineCode: '17066'
countryCode: ES
country: Spain
regionCode: '08'
region: Catalonia
provinceCode: '17'
province: Girona
""")
plantsns['municipalities'] = municipalitiesns
importPlants(self.pony.db, plantsns)
storedMunicipalitiesns = exportPlants(self.pony.db)
plantsns['plants'] = []
self.assertNsEqual(plantsns, storedMunicipalitiesns)
def test__importPlant_File(self):
fakePlantsYaml = 'fakeplant.yaml'
content = """\
plants:
- plant:
name: alcolea
codename: SCSOM04
description: la bonica planta
meters:
- meter:
name: '1234578'
inverters:
- inverter:
name: '5555'
- inverter:
name: '6666'
irradiationSensors:
- irradiationSensor:
name: alberto
temperatureAmbientSensors:
- temperatureAmbientSensor:
name: joana
temperatureModuleSensors:
- temperatureModuleSensor:
name: pol"""
p = Path(fakePlantsYaml)
with p.open("w", encoding="utf-8") as f:
f.write(content)
importPlantsFromFile(self.pony.db, fakePlantsYaml)
p.unlink()
with orm.db_session:
plantns = exportPlants(self.pony.db)
self.assertNsEqual(plantns, content)
