mirror of
https://github.com/datahub-project/datahub.git
synced 2025-08-24 00:57:59 +00:00
fix(ci): reverting unnecessary test changes (#4013)
This commit is contained in:
parent
c36662f837
commit
c1bcc9924e
@ -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
|
||||
|
Loading…
x
Reference in New Issue
Block a user