def test_get_entity_of_tab(self):
# given
key_label_map = {
'Project': 'project',
'Donor': 'donor',
'Specimen from organism': 'specimen_from_organism'
}
fake_tabs_config = MagicMock(name='tabs_config')
fake_tabs_config.get_key_for_label = lambda key: key_label_map.get(key)
schema_template = MagicMock(name='schema_template')
schema_template.get_tabs_config = MagicMock(return_value=fake_tabs_config)
ingest_api = MagicMock(name='ingest_api')
template_manager = TemplateManager(schema_template, ingest_api)
# when:
entity = template_manager.get_concrete_entity_of_tab('Specimen from organism')
# then:
self.assertEqual('specimen_from_organism', entity)
def do_import(self, worksheet, template: TemplateManager):
row_template = template.create_row_template(worksheet)
self.concrete_entity = template.get_concrete_entity_of_tab(
worksheet.title)
return self._import_using_row_template(template, worksheet,
row_template)