mirror of
https://github.com/BerriAI/litellm.git
synced 2026-08-28 05:25:59 +00:00
e824510765
prisma resolves its prisma-client-py generator through a plain /bin/sh PATH lookup, never through the interpreter that ran prisma generate, so the gate's generate step landed the client in whatever venv the caller had on PATH: the owned env never received one, every gate run regenerated, the caller's venv was mutated instead, and any invocation without a venv on PATH (the rewritten publisher workflow) failed outright The generate now runs with the target interpreter's bin directory pinned to the front of the child PATH. The prisma schema joins the environment fingerprint so clientless counts recorded before this commit can never be compared against clientful ones, a cold provision announces itself on stderr instead of sitting silent for two minutes, and the CI gate step reuses the job's prisma binary cache
64 lines
2.2 KiB
Python
64 lines
2.2 KiB
Python
import importlib.util
|
|
import os
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
_MODULE_PATH = (
|
|
Path(__file__).resolve().parents[2] / "scripts" / "prisma_generate_if_needed.py"
|
|
)
|
|
_spec = importlib.util.spec_from_file_location("prisma_generate_if_needed", _MODULE_PATH)
|
|
mod = importlib.util.module_from_spec(_spec)
|
|
_spec.loader.exec_module(mod)
|
|
|
|
|
|
def test_stamp_changes_with_schema_and_with_prisma_version():
|
|
stamp = mod.stamp_value(b"model A {}", "0.11.0")
|
|
assert mod.stamp_value(b"model A {}", "0.11.0") == stamp
|
|
assert mod.stamp_value(b"model B {}", "0.11.0") != stamp
|
|
assert mod.stamp_value(b"model A {}", "0.12.0") != stamp
|
|
|
|
|
|
def test_skip_requires_a_matching_stamp(tmp_path):
|
|
stamp = tmp_path / "stamp"
|
|
expected = mod.stamp_value(b"schema", "0.11.0")
|
|
assert mod.should_skip(stamp, expected, client_generated=True) is False
|
|
stamp.write_text(expected)
|
|
assert mod.should_skip(stamp, expected, client_generated=True) is True
|
|
assert (
|
|
mod.should_skip(stamp, mod.stamp_value(b"other", "0.11.0"), client_generated=True)
|
|
is False
|
|
)
|
|
|
|
|
|
def test_skip_requires_a_generated_client_even_with_a_matching_stamp(tmp_path):
|
|
stamp = tmp_path / "stamp"
|
|
expected = mod.stamp_value(b"schema", "0.11.0")
|
|
stamp.write_text(expected)
|
|
assert mod.should_skip(stamp, expected, client_generated=False) is False
|
|
|
|
|
|
def test_env_puts_this_interpreters_bin_dir_first_on_path():
|
|
env = mod.env_with_own_bin_first({"PATH": "/usr/bin", "HOME": "/home"})
|
|
bin_dir = str(Path(sys.executable).parent)
|
|
assert env["PATH"].split(os.pathsep) == [bin_dir, "/usr/bin"]
|
|
assert env["HOME"] == "/home"
|
|
|
|
|
|
def test_env_without_an_inherited_path_is_just_the_bin_dir():
|
|
env = mod.env_with_own_bin_first({})
|
|
assert env["PATH"] == str(Path(sys.executable).parent)
|
|
|
|
|
|
def test_generate_runs_prisma_with_its_own_bin_dir_leading_the_childs_path():
|
|
seen = {}
|
|
|
|
def recorder(cmd, cwd, env):
|
|
seen["cmd"] = cmd
|
|
seen["cwd"] = cwd
|
|
seen["env"] = env
|
|
return 0
|
|
|
|
assert mod.run_generate(run=recorder) == 0
|
|
assert seen["cmd"][:4] == [sys.executable, "-m", "prisma", "generate"]
|
|
assert seen["cwd"] == mod.REPO_ROOT
|
|
assert seen["env"]["PATH"].split(os.pathsep)[0] == str(Path(sys.executable).parent)
|