import json from datahub.cli.cli_utils import get_aspects_for_entity def test_no_client_id(graph_client): client_id_urn = "urn:li:telemetry:clientId" aspect = [ "clientId" ] # this is checking for the removal of the invalid aspect RemoveClientIdAspectStep.java res_data = json.dumps( get_aspects_for_entity( session=graph_client._session, gms_host=graph_client.config.server, entity_urn=client_id_urn, aspects=aspect, typed=False, ) ) assert res_data == "{}" def test_no_telemetry_client_id(graph_client): client_id_urn = "urn:li:telemetry:clientId" aspect = ["telemetryClientId"] # telemetry expected to be disabled for tests res_data = json.dumps( get_aspects_for_entity( session=graph_client._session, gms_host=graph_client.config.server, entity_urn=client_id_urn, aspects=aspect, typed=False, ) ) assert res_data == "{}"