import unittest from axione_api.api import (LigneResult, EtageResult, BatimentResult, parse_response) class TestAxioneApiParser(unittest.TestCase): def test_parse_dummy_data_1(self): with open("fixtures/dummy-data-1.xml","r") as f: data = f.read() expectedLigne = LigneResult(actif="false", commercialisable="true", existant="true", raccordable="true", rompu="false", pbo="BA64_BIEM2 D 17-18", pto="SPTH-BIEM2-0197") expectedEtage = EtageResult(reference="RDC", nbLignesActives="2", nbLignesExistantes="3", nbLocauxFtth="3", lignes=[expectedLigne]) expectedBatiment = BatimentResult(etatBatiment="RACCORDABLE", identifiantImmeuble="IMM64-497143", referenceBatiment="RESIDENCE NORMAND PRINCE - BAT-C4", etages=[expectedEtage]) expected = [expectedBatiment] self.assertEqual(parse_response(data), expected) if __name__ == '__main__': unittest.main()