def test_transform_function(self):
# type: () -> None
config = ConfigFactory.from_dict({})
transformer = BigqueryUsageTransformer()
transformer.init(config)
key = TableColumnUsageTuple(
database=TestBigQueryUsageTransform.DATABASE,
cluster=TestBigQueryUsageTransform.CLUSTER,
schema=TestBigQueryUsageTransform.DATASET,
table=TestBigQueryUsageTransform.TABLE,
column=TestBigQueryUsageTransform.COLUMN,
email=TestBigQueryUsageTransform.EMAIL)
t1 = (key, TestBigQueryUsageTransform.READ_COUNT)
xformed = transformer.transform(t1)
self.assertIsInstance(xformed, TableColumnUsage)
self.assertEqual(len(xformed.col_readers), 1)
col_reader = xformed.col_readers[0]
self.assertEqual(col_reader.cluster,
TestBigQueryUsageTransform.CLUSTER)
self.assertEqual(col_reader.database,
TestBigQueryUsageTransform.DATABASE)
self.assertEqual(col_reader.schema, TestBigQueryUsageTransform.DATASET)
self.assertEqual(col_reader.table, TestBigQueryUsageTransform.TABLE)
self.assertEqual(col_reader.column, TestBigQueryUsageTransform.COLUMN)
self.assertEqual(col_reader.user_email,
TestBigQueryUsageTransform.EMAIL)
self.assertEqual(col_reader.read_count,
TestBigQueryUsageTransform.READ_COUNT)
def test_transform_function(self) -> None:
config = ConfigFactory.from_dict({})
transformer = BigqueryUsageTransformer()
transformer.init(config)
key = TableColumnUsageTuple(database=TestBigQueryUsageTransform.DATABASE,
cluster=TestBigQueryUsageTransform.CLUSTER,
schema=TestBigQueryUsageTransform.DATASET,
table=TestBigQueryUsageTransform.TABLE,
column=TestBigQueryUsageTransform.COLUMN,
email=TestBigQueryUsageTransform.EMAIL)
t1 = (key, TestBigQueryUsageTransform.READ_COUNT)
xformed = transformer.transform(t1)
assert xformed is not None
self.assertIsInstance(xformed, TableColumnUsage)
col_readers = list(xformed.col_readers)
self.assertEqual(len(col_readers), 1)
col_reader = col_readers[0]
self.assertEqual(col_reader.start_label, 'Table')
self.assertEqual(col_reader.start_key, TestBigQueryUsageTransform.TABLE_KEY)
self.assertEqual(col_reader.user_email, TestBigQueryUsageTransform.EMAIL)
self.assertEqual(col_reader.read_count, TestBigQueryUsageTransform.READ_COUNT)