def test_extract_multiple_tag_content_should_return_correct_content(self):
region_tag_helper = utils.RegionTagHelper()
expected_tag_content = '''
metadata_definition:
- name: 'sp_calculateOrder'
purpose: 'This stored procedure will calculate orders.'
inputs:
- name: 'in1'
type: 'string'
outputs:
- name: 'out1'
type: 'int'
'''.strip()
tags_with_content = \
'[GOOGLE_DATA_CATALOG_METADATA_DEFINITION_START] \n' + \
expected_tag_content + \
'\n[GOOGLE_DATA_CATALOG_METADATA_DEFINITION_END] \n'
expected_tag_content_2 = '''
metadata_definition:
- name: 'sp_other_cloud'
purpose: 'This stored procedure run in another cloud.'
inputs:
- name: 'in1'
type: 'string'
outputs:
- name: 'out1'
type: 'int'
'''.strip()
tags_with_content_2 = \
'[OTHER_CLOUD_DATA_CATALOG_METADATA_DEFINITION_START] \n' + \
expected_tag_content_2 + \
'\n[OTHER_CLOUD_DATA_CATALOG_METADATA_DEFINITION_END] \n'
content_string = tags_with_content + '\n' + tags_with_content_2
extracted_tag_content = region_tag_helper.extract_content(
'GOOGLE_DATA_CATALOG_METADATA_DEFINITION', content_string)
self.assertEqual(extracted_tag_content, expected_tag_content)
extracted_tag_content_2 = region_tag_helper.extract_content(
'OTHER_CLOUD_DATA_CATALOG_METADATA_DEFINITION', content_string)
self.assertEqual(extracted_tag_content_2, expected_tag_content_2)
def test_extract_tag_content_no_region_tags_should_return_none(self):
region_tag_helper = utils.RegionTagHelper()
content_string = '''
metadata_definition:
- name: 'sp_calculateOrder'
purpose: 'This stored procedure will calculate orders.'
inputs:
- name: 'in1'
type: 'string'
outputs:
- name: 'out1'
type: 'int'
'''.strip()
extracted_tag_content = region_tag_helper.extract_content(
'GOOGLE_DATA_CATALOG_METADATA_DEFINITION', content_string)
self.assertIsNone(extracted_tag_content)
def test_extract_tag_content_should_return_correct_content(self):
region_tag_helper = utils.RegionTagHelper()
expected_tag_content = '''
metadata_definition:
- name: 'sp_calculateOrder'
purpose: 'This stored procedure will calculate orders.'
inputs:
- name: 'in1'
type: 'string'
outputs:
- name: 'out1'
type: 'int'
'''.strip()
content_string = \
'[GOOGLE_DATA_CATALOG_METADATA_DEFINITION_START] \n' + \
expected_tag_content + \
'\n[GOOGLE_DATA_CATALOG_METADATA_DEFINITION_END] \n'
extracted_tag_content = region_tag_helper.extract_content(
'GOOGLE_DATA_CATALOG_METADATA_DEFINITION', content_string)
self.assertEqual(extracted_tag_content, expected_tag_content)