From a01e4410ecb985e5d6307b0ee247304356f54382 Mon Sep 17 00:00:00 2001
From: Mark Roberts <mark.roberts@vercel.com>
Date: Mon, 14 Jul 2025 23:37:01 +0200
Subject: [PATCH] chore: get tests working again

---
 package.json      | 3 ++-
 tests/e2e.test.ts | 2 +-
 tests/mcp.test.ts | 2 +-
 3 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/package.json b/package.json
index 852d45d..fd5c566 100644
--- a/package.json
+++ b/package.json
@@ -42,7 +42,8 @@
   "scripts": {
     "build": "tsup",
     "dev": "tsup --watch",
-    "test": "vitest",
+    "test": "vitest run",
+    "test:watch": "vitest watch",
     "version-packages": "changeset version",
     "changeset": "changeset",
     "release": "pnpm build && changeset publish"
diff --git a/tests/e2e.test.ts b/tests/e2e.test.ts
index 5335615..a64c263 100644
--- a/tests/e2e.test.ts
+++ b/tests/e2e.test.ts
@@ -10,7 +10,7 @@ import type { AddressInfo } from "node:net";
 import { StreamableHTTPClientTransport } from "@modelcontextprotocol/sdk/client/streamableHttp.js";
 import { Client } from "@modelcontextprotocol/sdk/client/index.js";
 import { createMcpHandler } from "../src/index";
-import { withMcpAuth } from "../src/next/auth-wrapper";
+import { withMcpAuth } from "../src/auth/auth-wrapper";
 
 describe("e2e", () => {
   let server: Server;
diff --git a/tests/mcp.test.ts b/tests/mcp.test.ts
index 3f930b2..2e7fb7c 100644
--- a/tests/mcp.test.ts
+++ b/tests/mcp.test.ts
@@ -1,5 +1,5 @@
 import { describe, it, expect } from 'vitest';
-import { calculateEndpoints } from '../src/next/mcp-api-handler';
+import { calculateEndpoints } from '../src/handler/mcp-api-handler';
 
 describe('calculateEndpoints', () => {
   it('derives all endpoints from basePath', () => {