open-metadata--openmetadata/ingestion/tests/integration/mongodb/conftest.py

import os
import uuid
from datetime import datetime

import pytest

from _openmetadata_testutils.helpers.docker import try_bind
from metadata.generated.schema.api.services.createDatabaseService import (
    CreateDatabaseServiceRequest,
)
from metadata.generated.schema.entity.services.connections.database.mongoDBConnection import (
    MongoDBConnection,
)
from metadata.generated.schema.entity.services.databaseService import (
    DatabaseConnection,
    DatabaseServiceType,
)


@pytest.fixture(scope="module")
def mongodbContainer(tmp_path_factory):  # noqa: N802
    """
    Start a Mongodb container
    """
    from testcontainers.mongodb import MongoDbContainer

    container = MongoDbContainer()

    with try_bind(container, 27017, None) if not os.getenv("CI") else container as container:
        db = container.get_connection_client().test
        db.user_profiles.insert_one(
            {
                "user_id": "123e4567-e89b-12d3-a456-426614174000",
                "first_name": "John",
                "last_name": "Doe",
                "email": "john.doe@example.com",
                "signup_date": datetime.now(),
                "is_active": True,
            }
        )

        yield container


@pytest.fixture(scope="module")
def create_service_request(mongodbContainer):  # noqa: N803
    return CreateDatabaseServiceRequest(
        name=f"docker_test_mongodb_{uuid.uuid4().hex[:8]}",
        serviceType=DatabaseServiceType.MongoDB,
        connection=DatabaseConnection(
            config=MongoDBConnection(
                username="test",
                password="test",
                hostPort=f"localhost:{mongodbContainer.get_exposed_port(27017)}",
                databaseName="test",
            )
        ),
    )