From 3f72a533c6c558ed45323a5812fc162ed8d38588 Mon Sep 17 00:00:00 2001
From: bkellam <bshizzle1234@gmail.com>
Date: Sat, 18 Oct 2025 14:03:05 -0700
Subject: [PATCH] fix tests

---
 packages/backend/vitest.config.ts | 3 +++
 packages/web/src/prisma.ts        | 4 ----
 packages/web/src/withAuthV2.ts    | 4 ++--
 3 files changed, 5 insertions(+), 6 deletions(-)

diff --git a/packages/backend/vitest.config.ts b/packages/backend/vitest.config.ts
index 7c052526..5b2ab0d5 100644
--- a/packages/backend/vitest.config.ts
+++ b/packages/backend/vitest.config.ts
@@ -4,5 +4,8 @@ export default defineConfig({
     test: {
         environment: 'node',
         watch: false,
+        env: {
+            DATA_CACHE_DIR: 'test-data'
+        }
     }
 });
\ No newline at end of file
diff --git a/packages/web/src/prisma.ts b/packages/web/src/prisma.ts
index 6bd9aff8..860d23fa 100644
--- a/packages/web/src/prisma.ts
+++ b/packages/web/src/prisma.ts
@@ -58,7 +58,3 @@ export const userScopedPrismaClientExtension = (userId?: string) => {
             })
         })
 }
-
-export const getPrismaClient = (userId?: string) => {
-    return prisma.$extends(userScopedPrismaClientExtension(userId)) as PrismaClient;
-}
\ No newline at end of file
diff --git a/packages/web/src/withAuthV2.ts b/packages/web/src/withAuthV2.ts
index 1ffd541e..c6cbb8bb 100644
--- a/packages/web/src/withAuthV2.ts
+++ b/packages/web/src/withAuthV2.ts
@@ -1,4 +1,4 @@
-import { getPrismaClient, prisma as __unsafePrisma } from "@/prisma";
+import { prisma as __unsafePrisma, userScopedPrismaClientExtension } from "@/prisma";
 import { hashSecret } from "@sourcebot/crypto";
 import { ApiKey, Org, OrgRole, PrismaClient, User } from "@sourcebot/db";
 import { headers } from "next/headers";
@@ -88,7 +88,7 @@ export const getAuthContext = async (): Promise<OptionalAuthContext | ServiceErr
         },
     }) : null;
 
-    const prisma = getPrismaClient(user?.id);
+    const prisma = __unsafePrisma.$extends(userScopedPrismaClientExtension(user?.id)) as PrismaClient;
 
     return {
         user: user ?? undefined,