Initial public ModelForge release

backend/tests/test_hardware_collectors.py

@@ -0,0 +1,56 @@
+from pathlib import Path
+
+from hardware_fakes import FakeNvml
+
+from modelforge_api.domain.enums import Availability
+from modelforge_api.hardware.collectors import (
+    NodeIdentityProvider,
+    NvidiaNvmlCollector,
+    SystemHostCollector,
+)
+
+
+def test_nvml_unavailable_is_graceful_and_does_not_shutdown_uninitialized() -> None:
+    api = FakeNvml(init_error=True)
+    result = NvidiaNvmlCollector(api).collect()
+    assert result.availability is Availability.UNAVAILABLE
+    assert result.inventory == []
+    assert api.shutdown_calls == 0
+
+
+def test_zero_one_and_multiple_gpus_and_cleanup() -> None:
+    zero_api = FakeNvml(count=0)
+    assert NvidiaNvmlCollector(zero_api).collect().inventory == []
+    assert zero_api.shutdown_calls == 1
+    one = NvidiaNvmlCollector(FakeNvml(count=1)).collect()
+    assert one.inventory[0].device_uuid == "GPU-0"
+    assert one.inventory[0].total_vram_bytes.value == 16 * 1024**3
+    multiple = NvidiaNvmlCollector(FakeNvml(count=2)).collect()
+    assert [item.device_uuid for item in multiple.inventory] == ["GPU-0", "GPU-1"]
+
+
+def test_optional_metric_unsupported_and_individual_device_failure_are_isolated() -> None:
+    result = NvidiaNvmlCollector(FakeNvml(count=1, unsupported_power=True)).collect()
+    assert result.telemetry[0].power_draw_w.availability is Availability.UNSUPPORTED
+    degraded = NvidiaNvmlCollector(FakeNvml(count=2, device_error=1)).collect()
+    assert degraded.availability is Availability.TEMPORARILY_FAILED
+    assert len(degraded.inventory) == 1
+
+
+def test_host_inventory_and_persisted_node_identity(tmp_path: Path) -> None:
+    identity_file = tmp_path / "node-id"
+    provider = NodeIdentityProvider(identity_file, explicit_identity="stable-node")
+    collector = SystemHostCollector(provider, {"artifacts": tmp_path})
+    result = collector.collect()
+    assert result.identity_key == "stable-node"
+    assert result.total_ram_bytes.value and result.total_ram_bytes.value > 0
+    assert result.logical_cpu_count.value and result.logical_cpu_count.value > 0
+    assert result.storage[0].free_bytes.value is not None
+
+
+def test_generated_node_identity_is_stable(monkeypatch, tmp_path: Path) -> None:
+    monkeypatch.setattr(NodeIdentityProvider, "_system_machine_id", staticmethod(lambda: None))
+    provider = NodeIdentityProvider(tmp_path / "node-id")
+    first = provider.resolve()
+    second = NodeIdentityProvider(tmp_path / "node-id").resolve()
+    assert first == second