test_extension.py

from pathlib import Path

import pytest
from pytest import CaptureFixture, MonkeyPatch

from setuptools_rust.extension import RustBin, RustExtension

SETUPTOOLS_RUST_DIR = Path(__file__).parent.parent


@pytest.fixture()
def hello_world_bin() -> RustBin:
    return RustBin(
        "hello-world",
        path=(
            SETUPTOOLS_RUST_DIR / "examples" / "hello-world" / "Cargo.toml"
        ).as_posix(),
    )


@pytest.fixture()
def namespace_package_extension() -> RustExtension:
    return RustExtension(
        "namespace_package.rust",
        path=(
            SETUPTOOLS_RUST_DIR / "examples" / "namespace_package" / "Cargo.toml"
        ).as_posix(),
    )


def test_metadata_contents(hello_world_bin: RustBin) -> None:
    metadata = hello_world_bin.metadata(quiet=False)
    assert "target_directory" in metadata


def test_metadata_cargo_log(
    capfd: CaptureFixture, monkeypatch: MonkeyPatch, hello_world_bin: RustBin
) -> None:
    monkeypatch.setenv("CARGO_LOG", "trace")

    # With quiet unset, no stdout, plenty of logging stderr
    hello_world_bin.metadata(quiet=False)
    captured = capfd.readouterr()
    assert captured.out == ""
    assert "TRACE" in captured.err

    # With quiet set, nothing will be printed
    hello_world_bin.metadata(quiet=True)
    captured = capfd.readouterr()
    assert captured.out == ""
    assert captured.err == ""


def test_get_lib_name_namespace_package(
    namespace_package_extension: RustExtension,
) -> None:
    assert (
        namespace_package_extension.get_lib_name(quiet=True) == "namespace_package_rust"
    )