yujunjun/OpenMetadata
1
0
Fork 0
mirror of https://github.com/open-metadata/OpenMetadata.git synced 2025-08-16 04:57:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add table query test (#9000)

Browse Source 
* Add table query test

* Add table query test
This commit is contained in:
Pere Miquel Brull 2022-11-25 06:56:12 +01:00 committed by GitHub
parent afca46d45b
commit eccbb087d3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 41 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
ingestion/Dockerfile
View File

@ -25,6 +25,7 @@ RUN apt-get update \
tdsodbc \
unixodbc \
unixodbc-dev \
vim \
wget --no-install-recommends \
# Accept MSSQL ODBC License
&& ACCEPT_EULA=Y apt-get install -y msodbcsql18 \

41
ingestion/tests/integration/ometa/test_ometa_table_api.py
View File

@ -35,6 +35,7 @@ from metadata.generated.schema.entity.data.table import (
ColumnProfile,
DataType,
JoinedWith,
SqlQuery,
Table,
TableData,
TableJoins,
@ -325,7 +326,7 @@ class OMetaTableTest(TestCase):
profile = CreateTableProfileRequest(
tableProfile=table_profile, columnProfile=column_profile
)
res_profile = self.metadata.ingest_profile_data(res, profile)
self.metadata.ingest_profile_data(res, profile)
res_from_table = self.metadata.get_by_name(
entity=Table, fqn=self.entity.fullyQualifiedName, fields=["profile"]
)
@ -408,6 +409,44 @@ class OMetaTableTest(TestCase):
entity=Table, entity_id=str(direct_join_table_res.id.__root__)
)
def test_table_queries(self):
"""
Test add and update table query data
"""
self.metadata.create_or_update(data=self.create)
res = self.metadata.get_by_name(
entity=Table, fqn=self.entity.fullyQualifiedName
)
query_no_user = SqlQuery(query="select * from awesome")
self.metadata.ingest_table_queries_data(
table=res, table_queries=[query_no_user]
)
table_with_query: Table = self.metadata.get_by_name(
entity=Table, fqn=self.entity.fullyQualifiedName, fields=["tableQueries"]
)
assert len(table_with_query.tableQueries) == 1
assert table_with_query.tableQueries[0].query == query_no_user.query
assert table_with_query.tableQueries[0].users is None
# Validate that we can properly add user information
query_with_user = SqlQuery(query="select * from awesome", users=[self.owner])
self.metadata.ingest_table_queries_data(
table=res, table_queries=[query_with_user]
)
table_with_query: Table = self.metadata.get_by_name(
entity=Table, fqn=self.entity.fullyQualifiedName, fields=["tableQueries"]
)
assert len(table_with_query.tableQueries) == 1
assert table_with_query.tableQueries[0].query == query_with_user.query
assert table_with_query.tableQueries[0].users == [self.owner]
def test_list_versions(self):
"""
test list table entity versions