diff --git a/package.json b/package.json
index d180541..f4b6779 100644
--- a/package.json
+++ b/package.json
@@ -6,7 +6,8 @@
   "scripts": {
     "dev": "bun --hot index.js",
     "start": "NODE_ENV=production bun index.js",
-    "db:migrate": "psql $DATABASE_URL -f schema.sql"
+    "db:migrate": "psql $DATABASE_URL -f schema.sql",
+    "test": "bun test test/rateLimit.test.js && bun test test/file.test.js && bun test test/telegram.test.js && bun test test/upload.test.js && bun test test/files.test.js && bun test test/health.test.js && bun test test/bot.test.js && bun test test/bootstrap.test.js"
   },
   "dependencies": {
     "@prisma/engines": "^7.8.0",
diff --git a/test/bot.test.js b/test/bot.test.js
index c251974..277991e 100644
--- a/test/bot.test.js
+++ b/test/bot.test.js
@@ -1,4 +1,4 @@
-import { describe, it, expect, mock, spyOn, beforeEach } from "bun:test";
+import { describe, it, expect, mock, spyOn, beforeEach, afterAll } from "bun:test";
 import logger from "../src/utils/logger.js";
 
 // Mock environment
@@ -145,4 +145,8 @@ describe("Telegram Bot Handler", () => {
     expect(mockForwardToStorage).not.toHaveBeenCalled();
     expect(replyMock).toHaveBeenCalledWith(expect.stringContaining("exceeds"));
   });
+
+  afterAll(() => {
+    mock.restore();
+  });
 });
diff --git a/test/files.test.js b/test/files.test.js
index 98a299f..98d4faa 100644
--- a/test/files.test.js
+++ b/test/files.test.js
@@ -1,5 +1,4 @@
-import { describe, it, expect, mock, spyOn, beforeEach } from "bun:test";
-import logger from "../src/utils/logger.js";
+import { describe, it, expect, mock, beforeEach, afterAll } from "bun:test";
 
 // Mock database layer
 const mockSelect = mock(() => ({
@@ -176,4 +175,8 @@ describe("File Route Handlers", () => {
       expect(body.error).toBe("Server error");
     });
   });
+
+  afterAll(() => {
+    mock.restore();
+  });
 });
diff --git a/test/upload.test.js b/test/upload.test.js
index 9affe90..2eafd0f 100644
--- a/test/upload.test.js
+++ b/test/upload.test.js
@@ -1,5 +1,4 @@
-import { describe, it, expect, mock, spyOn, beforeEach, afterEach } from "bun:test";
-import logger from "../src/utils/logger.js";
+import { describe, it, expect, mock, beforeEach, afterAll } from "bun:test";
 
 // Mock db
 const mockInsert = mock(() => ({
@@ -116,4 +115,8 @@ describe("Upload Route Handler", () => {
     expect(body.public_id).toBe("mocked-nanoid-id");
     expect(body.file_name).toBe("test_multi.txt");
   });
+
+  afterAll(() => {
+    mock.restore();
+  });
 });