class BucketConfigTestSuite(TestCase):
def setUp(self):
self.config = ConfigurationToValidate()
self.config.area = Area()
self.config.area.ident = '3201'
self.attrs_retriever = BucketConfig()
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 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)
def setUp(self):
self.config = ConfigurationToValidate()
self.config.area = Area()
self.config.area.ident = '3201'
self.attrs_retriever = BucketConfig()
