def __map_datacatalog_relationships(cls, assembled_entries_dict):
all_assembled_entries = []
for assembled_entries_data in assembled_entries_dict.values():
all_assembled_entries.extend(assembled_entries_data)
prepare.EntryRelationshipMapper().fulfill_tag_fields(
all_assembled_entries)
def test_fulfill_tag_fields_should_resolve_app_custom_prop_def_mapping(
self):
definition_id = 'test_definition'
definition_entry = self.__make_fake_entry(
definition_id, 'custom_property_definition')
definition_tag = self.__make_fake_tag(
string_fields=(('id', definition_id), ))
app_id = 'test_app'
app_entry = self.__make_fake_entry(app_id, 'app')
string_fields = ('id', app_id), ('property_definition_id',
definition_id)
app_tag = self.__make_fake_tag(string_fields=string_fields)
definition_assembled_entry = commons_prepare.AssembledEntryData(
definition_id, definition_entry, [definition_tag])
app_assembled_entry = commons_prepare.AssembledEntryData(
app_id, app_entry, [app_tag])
prepare.EntryRelationshipMapper().fulfill_tag_fields(
[definition_assembled_entry, app_assembled_entry])
self.assertEqual(
'https://console.cloud.google.com/datacatalog/'
'fake_entries/test_definition',
app_tag.fields['property_definition_entry'].string_value)
def test_fulfill_tag_fields_should_resolve_app_stream_mapping(self):
stream_id = 'test_stream'
stream_entry = self.__make_fake_entry(stream_id, 'stream')
stream_tag = self.__make_fake_tag(string_fields=(('id', stream_id), ))
app_id = 'test_app'
app_entry = self.__make_fake_entry(app_id, 'app')
string_fields = ('id', app_id), ('stream_id', stream_id)
app_tag = self.__make_fake_tag(string_fields=string_fields)
stream_assembled_entry = commons_prepare.AssembledEntryData(
stream_id, stream_entry, [stream_tag])
app_assembled_entry = commons_prepare.AssembledEntryData(
app_id, app_entry, [app_tag])
prepare.EntryRelationshipMapper().fulfill_tag_fields(
[stream_assembled_entry, app_assembled_entry])
self.assertEqual(
'https://console.cloud.google.com/datacatalog/'
'fake_entries/test_stream',
app_tag.fields['stream_entry'].string_value)
def test_fulfill_tag_fields_should_resolve_measure_app_mapping(self):
app_id = 'test_app'
app_entry = self.__make_fake_entry(app_id, 'app')
app_tag = self.__make_fake_tag(string_fields=(('id', app_id), ))
measure_id = 'test_measure'
measure_entry = self.__make_fake_entry(measure_id, 'measure')
string_fields = ('id', measure_id), ('app_id', app_id)
measure_tag = self.__make_fake_tag(string_fields=string_fields)
app_assembled_entry = commons_prepare.AssembledEntryData(
app_id, app_entry, [app_tag])
measure_assembled_entry = commons_prepare.AssembledEntryData(
measure_id, measure_entry, [measure_tag])
prepare.EntryRelationshipMapper().fulfill_tag_fields(
[app_assembled_entry, measure_assembled_entry])
self.assertEqual(
'https://console.cloud.google.com/datacatalog/'
'fake_entries/test_app',
measure_tag.fields['app_entry'].string_value)