Fixed bigquery e2e test (#9555)

This commit is contained in:
Onkar Ravgan 2023-01-02 13:08:47 +05:30 committed by GitHub
parent dc157ee286
commit 420fa1181a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -162,11 +162,13 @@ class BigquerySource(CommonDbSourceService):
for tag in policy_tags: for tag in policy_tags:
yield OMetaTagAndClassification( yield OMetaTagAndClassification(
classification_request=CreateClassificationRequest( classification_request=CreateClassificationRequest(
name=self.service_connection.tagCategoryName, name=self.service_connection.classificationName,
description="", description="",
), ),
tag_request=CreateTagRequest( tag_request=CreateTagRequest(
name=tag.display_name, description="Bigquery Policy Tag" classification=self.service_connection.classificationName,
name=tag.display_name,
description="Bigquery Policy Tag",
), ),
) )
except Exception as exc: except Exception as exc:
@ -193,7 +195,7 @@ class BigquerySource(CommonDbSourceService):
tagFQN=fqn.build( tagFQN=fqn.build(
self.metadata, self.metadata,
entity_type=Tag, entity_type=Tag,
classification_name=self.service_connection.tagCategoryName, classification_name=self.service_connection.classificationName,
tag_name=column["policy_tags"], tag_name=column["policy_tags"],
), ),
labelType="Automated", labelType="Automated",