MINOR: Update Glossary Term tests (#18698)

* Update Glossary Term tests

* Remove unused code

* Fix test
This commit is contained in:
IceS2 2024-11-20 15:24:53 +01:00 committed by GitHub
parent 0e3e1f59c9
commit 35ce7d7602
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 477 additions and 502 deletions

View File

@ -15,6 +15,13 @@ import uuid
import pytest
from metadata.generated.schema.api.data.createGlossary import CreateGlossaryRequest
from metadata.generated.schema.api.data.createGlossaryTerm import (
CreateGlossaryTermRequest,
)
from metadata.generated.schema.api.teams.createUser import CreateUserRequest
from metadata.generated.schema.entity.data.glossary import Glossary
from metadata.generated.schema.entity.data.glossaryTerm import GlossaryTerm
from metadata.generated.schema.entity.services.connections.database.common.basicAuth import (
BasicAuth,
)
@ -22,6 +29,7 @@ from metadata.generated.schema.entity.services.connections.database.mysqlConnect
MysqlConnection,
)
from metadata.generated.schema.entity.services.databaseService import DatabaseService
from metadata.generated.schema.entity.teams.user import User
from metadata.workflow.metadata import MetadataWorkflow
from ..containers import MySqlContainerConfigs, get_mysql_container
@ -78,3 +86,61 @@ def workflow(metadata, service, mysql_container):
)
workflow_config["ingestionPipelineFQN"] = f"{service_name}.ingestion"
return MetadataWorkflow.create(workflow_config)
@pytest.fixture
def create_glossary(metadata):
glossaries = []
def _create_glossary(create_request: CreateGlossaryRequest):
glossary = metadata.create_or_update(data=create_request)
glossaries.append(glossary)
return glossary
def teardown():
for glossary in glossaries:
metadata.delete(entity=Glossary, entity_id=glossary.id, hard_delete=True)
yield _create_glossary
teardown()
@pytest.fixture
def create_glossary_term(metadata, create_glossary):
glossary_terms = []
def _create_glossary_term(create_request: CreateGlossaryTermRequest):
glossary_term = metadata.create_or_update(data=create_request)
glossary_terms.append(glossary_term)
return glossary_term
def teardown():
glossary_terms.reverse()
for glossary_term in glossary_terms:
metadata.delete(
entity=GlossaryTerm, entity_id=glossary_term.id, hard_delete=True
)
yield _create_glossary_term
teardown()
@pytest.fixture
def create_user(metadata, request):
users = []
def _create_user(create_request: CreateUserRequest):
user = metadata.create_or_update(data=create_request)
users.append(user)
return user
def teardown():
for user in users:
metadata.delete(entity=User, entity_id=user.id, hard_delete=True)
request.addfinalizer(teardown)
return _create_user

File diff suppressed because it is too large Load Diff