def test_get_concrete_type_of_worksheet_invalid_format(self):
# given:
schema_template = MagicMock(name='schema_template')
manager = TemplateManager(schema_template,
MagicMock(name='ingest_api'))
# when:
raised_exception = None
try:
manager.get_concrete_type('- does not match format -')
except InvalidTabName as exception:
raised_exception = exception
# then:
self.assertIsNotNone(raised_exception)
self.assertEqual('- does not match format -',
raised_exception.tab_name)
def test_get_concrete_type_of_regular_worksheet(self):
# given
schema_template = MagicMock(name='schema_template')
schema_template.get_tab_key = MagicMock(return_value='user_profile')
manager = TemplateManager(schema_template,
MagicMock(name='ingest_api'))
# expect:
self.assertEqual('user_profile',
manager.get_concrete_type('User Profile'))
schema_template.get_tab_key.assert_called_with('User Profile')
def test_get_concrete_type_of_module_worksheet(self):
# given:
schema_template = MagicMock(name='schema_template')
schema_template.get_tab_key = MagicMock(return_value='product')
manager = TemplateManager(schema_template,
MagicMock(name='ingest_api'))
# expect:
self.assertEqual('product',
manager.get_concrete_type('Product - Barcodes'))
# and:
schema_template.get_tab_key.assert_called_with('Product')
def test_get_concrete_type_of_regular_worksheet(self):
# given
schema_template = MagicMock(name='schema_template')
schema_template.lookup_metadata_schema_name_given_title = MagicMock(
return_value='user_profile')
manager = TemplateManager(schema_template,
MagicMock(name='mock_ingest_api'))
# expect:
self.assertEqual('user_profile',
manager.get_concrete_type('User Profile'))
schema_template.lookup_metadata_schema_name_given_title.assert_called_with(
'User Profile')