def test_c(self): """Test the retrieval of records without a GAFIDENT field.""" record = {'DIEPTE': ' 1.17'} dbf = Mock() dbf.get_records = lambda: [record] self.attrs_retriever.open_database = Mock(return_value=dbf) attrs = self.attrs_retriever.as_dict(self.config) self.assertEqual({}, attrs)
def test_a(self): """Test the retrieval of a single bucket record.""" record = {'ID_GW': '3201-DGW-1', 'GEBIED_GW': '3201', 'OPPERVL': '2171871'} dbf = Mock() dbf.get_records = lambda: [record] self.attrs_retriever.open_database = Mock(return_value=dbf) attrs = self.attrs_retriever.as_dict(self.config) self.assertEqual({'3201-DGW-1': {'ID_GW': '3201-DGW-1', 'GEBIED_GW': '3201', 'OPPERVL': '2171871'}}, attrs)
def setUp(self): self.config = ConfigurationToValidate() self.config.area = Area() self.config.area.ident = '3201' self.attrs_retriever = AreaConfig() record = {'GAFIDENT': '3201', 'DIEPTE': ' 1.17'} dbf = Mock() dbf.get_records = lambda: [record] self.attrs_retriever.open_database = Mock(return_value=dbf)
def test_b(self): """Test the retrieval of two bucket records.""" records = [{'ID_GW': '3201-DGW-1', 'GEBIED_GW': '3201', 'OPPERVL': '2171871'}, {'ID_GW': '3201-DGW-2', 'GEBIED_GW': '3201', 'OPPERVL': '844617'}] dbf = Mock() dbf.get_records = lambda: records self.attrs_retriever.open_database = Mock(return_value=dbf) attrs = self.attrs_retriever.as_dict(self.config) self.assertEqual({'3201-DGW-1': {'ID_GW': '3201-DGW-1', 'GEBIED_GW': '3201', 'OPPERVL': '2171871'}, '3201-DGW-2': {'ID_GW': '3201-DGW-2', 'GEBIED_GW': '3201', 'OPPERVL': '844617'}}, attrs)