mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-15 23:31:29 +00:00
121 lines
4.5 KiB
Python
121 lines
4.5 KiB
Python
from __future__ import annotations
|
|
|
|
import os
|
|
from types import SimpleNamespace
|
|
from typing import Final
|
|
|
|
import pytest
|
|
|
|
from . import native_build
|
|
|
|
|
|
def test_needs_rebuild_when_bridge_is_missing() -> None:
|
|
assert native_build.needs_rebuild(None, 1.0)
|
|
|
|
|
|
def test_needs_rebuild_when_sources_are_newer_than_bridge() -> None:
|
|
assert native_build.needs_rebuild(1.0, 2.0)
|
|
|
|
|
|
def test_fresh_bridge_with_older_sources_needs_no_rebuild() -> None:
|
|
assert not native_build.needs_rebuild(2.0, 1.0)
|
|
|
|
|
|
def test_bridge_without_rust_sources_needs_no_rebuild() -> None:
|
|
assert not native_build.needs_rebuild(2.0, None)
|
|
|
|
|
|
def test_newest_source_mtime_tracks_rust_sources_and_skips_target(tmp_path: Final) -> None:
|
|
source: Final = tmp_path / "litellm-rust" / "crates" / "bridge" / "src"
|
|
source.mkdir(parents=True)
|
|
(source / "lib.rs").write_text("fn main() {}\n")
|
|
os.utime(source / "lib.rs", (1_000, 1_000))
|
|
manifest: Final = tmp_path / "litellm-rust" / "crates" / "bridge" / "Cargo.toml"
|
|
manifest.write_text("[package]\n")
|
|
os.utime(manifest, (2_000, 2_000))
|
|
lockfile: Final = tmp_path / "litellm-rust" / "Cargo.lock"
|
|
lockfile.write_text("")
|
|
os.utime(lockfile, (1_500, 1_500))
|
|
target: Final = tmp_path / "litellm-rust" / "target" / "debug" / "junk.rs"
|
|
target.parent.mkdir(parents=True)
|
|
target.write_text("fn main() {}\n")
|
|
os.utime(target, (9_999, 9_999))
|
|
|
|
assert native_build._newest_source_mtime(tmp_path) == 2_000.0
|
|
|
|
|
|
def test_newest_source_mtime_is_none_without_rust_workspace(tmp_path: Final) -> None:
|
|
assert native_build._newest_source_mtime(tmp_path) is None
|
|
|
|
|
|
def test_ensure_trace_bridge_rebuilds_when_stale(
|
|
tmp_path: Final, monkeypatch: pytest.MonkeyPatch, capsys: pytest.CaptureFixture[str]
|
|
) -> None:
|
|
native: Final = tmp_path / "_native.abi3.so"
|
|
native.write_bytes(b"")
|
|
os.utime(native, (1_000, 1_000))
|
|
source: Final = tmp_path / "litellm-rust" / "crates" / "bridge" / "src" / "lib.rs"
|
|
source.parent.mkdir(parents=True)
|
|
source.write_text("fn main() {}\n")
|
|
os.utime(source, (2_000, 2_000))
|
|
state: Final = SimpleNamespace(rebuilt=False)
|
|
|
|
def fake_rebuild(repo_root: object) -> tuple[bool, str]:
|
|
state.rebuilt = True
|
|
return True, ""
|
|
|
|
monkeypatch.setattr(native_build, "_native_module_path", lambda: native)
|
|
monkeypatch.setattr(native_build, "_rebuild", fake_rebuild)
|
|
monkeypatch.setattr(native_build, "_drop_imported_bridge", lambda: None)
|
|
monkeypatch.setattr(native_build, "get_native_bridge", lambda: SimpleNamespace(_trace=object()))
|
|
|
|
assert native_build.ensure_trace_bridge(tmp_path) is None
|
|
assert state.rebuilt is True
|
|
assert "Rebuilding native Rust bridge" in capsys.readouterr().out
|
|
|
|
|
|
def test_ensure_trace_bridge_reports_failed_rebuild(tmp_path: Final, monkeypatch: pytest.MonkeyPatch) -> None:
|
|
source: Final = tmp_path / "litellm-rust" / "crates" / "bridge" / "src" / "lib.rs"
|
|
source.parent.mkdir(parents=True)
|
|
source.write_text("fn main() {}\n")
|
|
|
|
monkeypatch.setattr(native_build, "_native_module_path", lambda: None)
|
|
monkeypatch.setattr(native_build, "_rebuild", lambda repo_root: (False, "boom"))
|
|
|
|
message: Final = native_build.ensure_trace_bridge(tmp_path)
|
|
|
|
assert message is not None
|
|
assert "rebuild failed" in message
|
|
assert "boom" in message
|
|
|
|
|
|
def test_ensure_trace_bridge_rebuilds_when_trace_feature_is_missing(
|
|
tmp_path: Final, monkeypatch: pytest.MonkeyPatch, capsys: pytest.CaptureFixture[str]
|
|
) -> None:
|
|
native: Final = tmp_path / "_native.abi3.so"
|
|
native.write_bytes(b"")
|
|
os.utime(native, (9_999, 9_999))
|
|
source: Final = tmp_path / "litellm-rust" / "crates" / "bridge" / "src" / "lib.rs"
|
|
source.parent.mkdir(parents=True)
|
|
source.write_text("fn main() {}\n")
|
|
os.utime(source, (1_000, 1_000))
|
|
state: Final = SimpleNamespace(rebuilt=False)
|
|
|
|
def fake_rebuild(repo_root: object) -> tuple[bool, str]:
|
|
state.rebuilt = True
|
|
return True, ""
|
|
|
|
def fake_get_native_bridge() -> SimpleNamespace:
|
|
assert state.rebuilt
|
|
return SimpleNamespace(_trace=object())
|
|
|
|
monkeypatch.setattr(native_build, "_native_module_path", lambda: native)
|
|
monkeypatch.setattr(native_build, "_rebuild", fake_rebuild)
|
|
monkeypatch.setattr(native_build, "_installed_bridge_has_trace", lambda repo_root: False)
|
|
monkeypatch.setattr(native_build, "get_native_bridge", fake_get_native_bridge)
|
|
|
|
message: Final = native_build.ensure_trace_bridge(tmp_path)
|
|
|
|
assert message is None
|
|
assert state.rebuilt is True
|
|
assert "Rebuilding native Rust bridge" in capsys.readouterr().out
|