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)