50 lines · 1.5 KB · TypeScript
RawDownload
import { afterEach, describe, expect, it } from "bun:test";import { getSystemRuntimeInfo } from "../services/runtime-info";import type { Config } from "../config/env"; const baseConfig: Config = {  host: "127.0.0.1",  port: 8080,  inference_port: 8000,  temporal_address: "localhost:7233",  data_dir: "/tmp",  db_path: "/tmp/controller.db",  models_dir: "/models",}; const ORIGINAL_ENV = { ...process.env }; afterEach(() => {  for (const key of Object.keys(process.env)) {    if (!(key in ORIGINAL_ENV)) {      delete process.env[key];    }  }  for (const [key, value] of Object.entries(ORIGINAL_ENV)) {    if (typeof value === "string") {      process.env[key] = value;    } else {      delete process.env[key];    }  }}); describe("getSystemRuntimeInfo platform model", () => {  it("reports ROCm platform when forced via smi tool env", async () => {    process.env["VLLM_STUDIO_GPU_SMI_TOOL"] = "amd-smi";    const info = await getSystemRuntimeInfo(baseConfig);    expect(info.platform.kind).toBe("rocm");    expect(info.platform.vendor).toBe("amd");    expect(info.platform.rocm?.smi_tool).toBe("amd-smi");    expect(info.cuda.driver_version).toBeNull();    expect(info.cuda.cuda_version).toBeNull();  });   it("reports CUDA platform when forced via smi tool env", async () => {    process.env["VLLM_STUDIO_GPU_SMI_TOOL"] = "nvidia-smi";    const info = await getSystemRuntimeInfo(baseConfig);    expect(info.platform.kind).toBe("cuda");    expect(info.platform.vendor).toBe("nvidia");    expect(info.platform.rocm).toBeNull();  });});