def test_tags_not_added_to_other_objects(self):
transformer = TableTagTransformer()
config = ConfigFactory.from_dict({
TableTagTransformer.TAGS: 'new_tag',
})
transformer.init(conf=config)
class NotATable():
tags = 'existing_tag'
result = transformer.transform(NotATable())
self.assertEqual(result.tags, 'existing_tag')
def test_add_tag_to_existing_tags(self) -> None:
transformer = TableTagTransformer()
config = ConfigFactory.from_dict({
TableTagTransformer.TAGS: 'baz',
})
transformer.init(conf=config)
result = transformer.transform(TableMetadata(
database='test_db',
cluster='test_cluster',
schema='test_schema',
name='test_table',
description='',
tags='foo,bar',
))
self.assertEqual(result.tags, ['foo', 'bar', 'baz'])
def test_multiple_tags_comma_delimited(self) -> None:
transformer = TableTagTransformer()
config = ConfigFactory.from_dict({
TableTagTransformer.TAGS: 'foo,bar',
})
transformer.init(conf=config)
result = transformer.transform(TableMetadata(
database='test_db',
cluster='test_cluster',
schema='test_schema',
name='test_table',
description='',
))
self.assertEqual(result.tags, ['foo', 'bar'])