Merge pull request #21421 from BerriAI/fix/mock-prisma-in-backoff-tests

fix(tests): mock prisma.Prisma in backoff retry tests to avoid 'prisma generate'
This commit is contained in:
jquinter 2026-02-17 20:04:34 -03:00 committed by GitHub
commit 153af8b901
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -22,6 +22,20 @@ import httpx
import backoff
@pytest.fixture(autouse=True)
def mock_prisma_binary():
"""Mock prisma.Prisma to avoid requiring 'prisma generate' in CI.
PrismaClient.__init__ does `from prisma import Prisma` inline, which raises
RuntimeError when the Prisma client hasn't been generated yet. Replacing
sys.modules['prisma'] with a MagicMock lets the import succeed so tests can
instantiate a real PrismaClient and override client.db with their own mocks.
"""
mock_module = MagicMock()
with patch.dict(sys.modules, {"prisma": mock_module}):
yield
@pytest.fixture
def mock_prisma_client():
"""Create a mock PrismaClient with necessary attributes"""