tests edited

Author: rabianadeem-121

__tests__/errors.test.js

@@ -1,21 +1,26 @@
-// error.test.js
-const { handler } = require('../index'); // go up one level
-
+// __tests__/error.test.js
+const { handler } = require('../index');
 
 describe('Error handling tests', () => {
-
   test('GET /users/:id returns 404 if user does not exist', async () => {
     const event = {
       httpMethod: 'GET',
-      path: '/users/9999'
+      path: '/users/9999',
     };
 
+    // Set RDS credentials
+    process.env.DB_HOST = process.env.DB_HOST;
+    process.env.DB_USER = process.env.DB_USER;
+    process.env.DB_PASSWORD = process.env.DB_PASSWORD;
+    process.env.DB_NAME = process.env.DB_NAME;
+    process.env.DB_PORT = process.env.DB_PORT || 5432;
+
     const response = await handler(event);
 
     expect(response.statusCode).toBe(404);
 
     const body = JSON.parse(response.body);
     expect(body.message).toBe('User not found');
   });
-
 });
+

__tests__/example.test.js

@@ -1,26 +1,27 @@
-// lambda.test.js
-const { handler } = require('../index'); // go up one level
-
+// __tests__/example.test.js
+const { handler } = require('../index');
 
 describe('Lambda API tests', () => {
-
   test('GET /users should return status 200 and a list of users', async () => {
     const event = {
       httpMethod: 'GET',
       path: '/users',
     };
 
+    // Set RDS credentials from environment variables
+    process.env.DB_HOST = process.env.DB_HOST;
+    process.env.DB_USER = process.env.DB_USER;
+    process.env.DB_PASSWORD = process.env.DB_PASSWORD;
+    process.env.DB_NAME = process.env.DB_NAME;
+    process.env.DB_PORT = process.env.DB_PORT || 5432;
+
     const response = await handler(event);
 
-    // Check HTTP status
     expect(response.statusCode).toBe(200);
 
-    // Check response body is an array (JSON parsed)
     const body = JSON.parse(response.body);
     expect(Array.isArray(body)).toBe(true);
-
-    // Optional: check there is at least one user
-    expect(body.length).toBeGreaterThan(0);
+    expect(body.length).toBeGreaterThanOrEqual(0);
   });
-
 });
+

__tests__/users.test.js

@@ -1,16 +1,21 @@
-// users.test.js
-const { handler } = require('../index'); // go up one level
-
+// __tests__/users.test.js
+const { handler } = require('../index');
 
 describe('POST /users endpoint', () => {
-
   test('should create a new user and return status 201', async () => {
     const event = {
       httpMethod: 'POST',
       path: '/users',
-      body: JSON.stringify({ name: 'John Doe', email: 'john@example.com' })
+      body: JSON.stringify({ name: 'John Doe', email: 'john@example.com' }),
     };
 
+    // Set RDS credentials
+    process.env.DB_HOST = process.env.DB_HOST;
+    process.env.DB_USER = process.env.DB_USER;
+    process.env.DB_PASSWORD = process.env.DB_PASSWORD;
+    process.env.DB_NAME = process.env.DB_NAME;
+    process.env.DB_PORT = process.env.DB_PORT || 5432;
+
     const response = await handler(event);
 
     expect(response.statusCode).toBe(201);
@@ -20,5 +25,4 @@ describe('POST /users endpoint', () => {
     expect(body.email).toBe('john@example.com');
     expect(body.id).toBeDefined();
   });
-
 });