diff --git a/smoke-test/tests/domains/domains_test.py b/smoke-test/tests/domains/domains_test.py index b1eb68ac30..11b41cb424 100644 --- a/smoke-test/tests/domains/domains_test.py +++ b/smoke-test/tests/domains/domains_test.py @@ -5,18 +5,6 @@ from tests.utils import GMS_ENDPOINT from tests.utils import ingest_file_via_rest from tests.utils import delete_urns_from_file -from typing import List - -import datahub.emitter.mce_builder as builder -from datahub.emitter.mcp import MetadataChangeProposalWrapper -from datahub.emitter.rest_emitter import DatahubRestEmitter -from datahub.metadata.com.linkedin.pegasus2avro.dataset import ( - DatasetLineageTypeClass, - UpstreamClass, - UpstreamLineage, -) -from datahub.metadata.schema_classes import ChangeTypeClass - @pytest.fixture(scope="module", autouse=False) def ingest_cleanup_data(request): print("ingesting domains test data") @@ -33,39 +21,6 @@ def test_healthchecks(wait_for_healthchecks): @pytest.mark.dependency(depends=["test_healthchecks"]) def test_create_list_get_domain(frontend_session): - # Construct upstream tables. - upstream_tables: List[UpstreamClass] = [] - upstream_table_1 = UpstreamClass( - dataset=builder.make_dataset_urn("bigquery", "upstream_table_1", "PROD"), - type=DatasetLineageTypeClass.TRANSFORMED, - ) - upstream_tables.append(upstream_table_1) - upstream_table_2 = UpstreamClass( - dataset=builder.make_dataset_urn("bigquery", "upstream_table_2", "PROD"), - type=DatasetLineageTypeClass.TRANSFORMED, - ) - upstream_tables.append(upstream_table_2) - - # Construct a lineage object. - upstream_lineage = UpstreamLineage(upstreams=upstream_tables) - - # Construct a MetadataChangeProposalWrapper object. - lineage_mcp = MetadataChangeProposalWrapper( - entityType="dataset", - changeType=ChangeTypeClass.UPSERT, - entityUrn=builder.make_dataset_urn("bigquery", "downstream"), - aspectName="upstreamLineage", - aspect=upstream_lineage, - systemMetadata= - ) - - # Create an emitter to the GMS REST API. - emitter = DatahubRestEmitter("http://localhost:8080") - - # Emit metadata! - emitter.emit_mcp(lineage_mcp) - - # Get count of existing secrets list_domains_json = { "query": """query listDomains($input: ListDomainsInput!) {\n