Function pytest_configure

tests/conftest.py:100–200

(config)

def pytest_configure(config):
    # patch the configurations to use test storage by default, we modify the types (classes) fields
    # the dataclass implementation will use those patched values when creating instances (the values present
    # in the declaration are not frozen allowing patching). this is needed by common storage tests
    from tests.utils import set_environment_test_storage_root, compute_test_storage_root

    test_storage_root = compute_test_storage_root()
    Path(test_storage_root).mkdir(exist_ok=True)
    set_environment_test_storage_root(test_storage_root)

    from dlt.common.configuration.specs import runtime_configuration
    from dlt.common.storages import configuration as storage_configuration

    runtime_configuration.RuntimeConfiguration.config_files_storage_path = os.path.join(
        test_storage_root, "config/"
    )
    # always use CI track endpoint when running tests
    runtime_configuration.RuntimeConfiguration.dlthub_telemetry_endpoint = (
        "https://telemetry-tracker.services4758.workers.dev"
    )

    del runtime_configuration.RuntimeConfiguration.__init__
    runtime_configuration.RuntimeConfiguration = dataclasses.dataclass(  # type: ignore[misc]
        runtime_configuration.RuntimeConfiguration, init=True, repr=False
    )

    storage_configuration.LoadStorageConfiguration.load_volume_path = os.path.join(
        test_storage_root, "load"
    )
    del storage_configuration.LoadStorageConfiguration.__init__
    storage_configuration.LoadStorageConfiguration = dataclasses.dataclass(  # type: ignore[misc]
        storage_configuration.LoadStorageConfiguration, init=True, repr=False
    )

    storage_configuration.NormalizeStorageConfiguration.normalize_volume_path = os.path.join(
        test_storage_root, "normalize"
    )
    # delete __init__, otherwise it will not be recreated by dataclass
    del storage_configuration.NormalizeStorageConfiguration.__init__
    storage_configuration.NormalizeStorageConfiguration = dataclasses.dataclass(  # type: ignore[misc]
        storage_configuration.NormalizeStorageConfiguration, init=True, repr=False
    )

    storage_configuration.SchemaStorageConfiguration.schema_volume_path = os.path.join(
        test_storage_root, "schemas"
    )
    del storage_configuration.SchemaStorageConfiguration.__init__
    storage_configuration.SchemaStorageConfiguration = dataclasses.dataclass(  # type: ignore[misc]
        storage_configuration.SchemaStorageConfiguration, init=True, repr=False
    )

    assert runtime_configuration.RuntimeConfiguration.config_files_storage_path == os.path.join(
        test_storage_root, "config/"
    )
    assert runtime_configuration.RuntimeConfiguration().config_files_storage_path == os.path.join(
        test_storage_root, "config/"
    )