class ConfigurationSpecRetrieverTestSuite(TestCase): def setUp(self): self.retriever = ConfigurationSpecRetriever() self.retriever.retrieve_area_codes = Mock(return_value=['3201']) self.retriever.retrieve_meta_info = Mock( return_value={'user': '******'}) def test_a(self): """Test the construction of a single ConfigurationSpec.""" dir_name = '/path/to/configuration/directory' config_type = 'waterbalans' config_specs = self.retriever.retrieve(dir_name, config_type) self.assertEqual(1, len(config_specs)) self.assertEqual('3201', config_specs[0]['area_code']) def test_b(self): """Test retrieve_area_codes_from_dbf is called correctly.""" dir_name = '/path/to/configuration/directory' config_type = 'waterbalans' self.retriever.retrieve(dir_name, config_type) args, kwargs = self.retriever.retrieve_area_codes.call_args self.assertEqual( '/path/to/configuration/directory/aanafvoer_waterbalans.dbf', args[0]) def test_c(self): """Test retrieve_area_codes_from_dbf is called correctly.""" dir_name = '/path/to/configuration/directory' config_type = 'esf1' self.retriever.retrieve(dir_name, config_type) args, kwargs = self.retriever.retrieve_area_codes.call_args self.assertEqual('/path/to/configuration/directory/aanafvoer_esf1.dbf', args[0]) def test_d(self): """Test the retrieval of the 'user' attribute.""" dir_name = '/path/to/configuration/directory' config_type = 'waterbalans' config_specs = self.retriever.retrieve(dir_name, config_type) self.assertEqual(1, len(config_specs)) self.assertEqual('Pieter Swinkels', config_specs[0]['user']) def test_e(self): """Test retrieva_meta_info receives the right parameters.""" dir_name = '/path/to/configuration/directory' config_type = 'waterbalans' self.retriever.retrieve(dir_name, config_type) args, kwargs = self.retriever.retrieve_meta_info.call_args self.assertEqual('/path/to/configuration/directory/description.txt', args[0])
def setUp(self): self.retriever = ConfigurationSpecRetriever() self.retriever.retrieve_area_codes = Mock(return_value=['3201']) self.retriever.retrieve_meta_info = Mock( return_value={'user': '******'})