Update unit tests

Author: Tati-Moon

jest.config.js

@@ -1,5 +1,10 @@
 // https://jestjs.io/docs/configuration
 
 module.exports = {
-  testMatch: ["**/?(*.)+(spec).js?(x)", "**/?(*.)+(spec).ts?(x)"],
+    testMatch: ["**/?(*.)+(spec).js?(x)", "**/?(*.)+(spec).ts?(x)"],
+    collectCoverageFrom: [
+        'product_service/lambdas/*.js',
+        '!product_service/lambdas/constants.js',
+        '!product_service/lambdas/cors.js',
+    ],
 };

product_service/lambdas/getProductsById.js

@@ -40,8 +40,8 @@ exports.handler = async (event) => {
             dynamoDB.send(new QueryCommand(stockParams))
         ]);
 
-        const product = productResponse.Items[0];
-        const stock = stockResponse.Items[0];
+        const product = productResponse?.Items[0];
+        const stock = stockResponse?.Items[0];
 
         if (product) {
             const returnedBody = JSON.stringify({

product_service/test/validation.spec.js

@@ -0,0 +1,55 @@
+const { validateInput } = require('../lambdas/validation');
+const { MESSAGES } = require('../lambdas/constants');
+const ProductBuilder = require('../utils/productBuilder');
+
+describe('validateInput function', () => {
+    it('should return null if input is valid', () => {
+        const input = new ProductBuilder().build();
+        const result = validateInput(input);
+        expect(result).toBeNull();
+    });
+ 
+    describe.each([{}, null, undefined, NaN, '', '\t', '\n'])(
+        'should return INVALID_INPUT_TITLE if description is "%s"',
+        (title) => {
+            it(`should return ${MESSAGES.INVALID_INPUT_TITLE}`, () => {
+                const input = new ProductBuilder().withTitle(title).build();
+                const result = validateInput(input);
+                expect(result).toEqual(MESSAGES.INVALID_INPUT_TITLE);
+            });
+        }
+    );  
+
+    describe.each([{}, null, undefined, NaN, '', '\t', '\n'])(
+        'should return INVALID_INPUT_DESCRIPTION if description is "%s"',
+        (description) => {
+            it(`should return ${MESSAGES.INVALID_INPUT_DESCRIPTION}`, () => {
+                const input = new ProductBuilder().withDescription(description).build();
+                const result = validateInput(input);
+                expect(result).toEqual(MESSAGES.INVALID_INPUT_DESCRIPTION);
+            });
+        }
+    );    
+
+    describe.each([-2, -1, {}, null, undefined, NaN, '', '\t', '\n'])(
+        'should return INVALID_INPUT_PRICE if price is invalid: "%s"',
+        (price) => {
+            it('should return INVALID_INPUT_PRICE', () => {
+                const input = new ProductBuilder().withPrice(price).build();
+                const result = validateInput(input);
+                expect(result).toEqual(MESSAGES.INVALID_INPUT_PRICE);
+            });
+        }
+    );    
+
+    describe.each([-2, -1, {}, null, undefined, NaN, '', '\t', '\n'])(
+        'should return INVALID_INPUT_COUNT if count is less than zero: "%s"',
+        (count) => {
+            it('should return INVALID_INPUT_COUNT', () => {
+                const input = new ProductBuilder().withCount(count).build();
+                const result = validateInput(input);
+                expect(result).toEqual(MESSAGES.INVALID_INPUT_COUNT);
+            });
+        }
+    );    
+});

product_service/utils/productBuilder.js

@@ -0,0 +1,42 @@
+class ProductBuilder {
+    constructor() {
+        this.product = {
+            id: 'ead768ea-8752-4417-a94d-e8eaf256c500',
+            title: 'some title',
+            description: 'some description',
+            price: 100,
+            count: 50
+        };
+    }
+
+    withId(id) {
+        this.product.id = id;
+        return this;
+    }
+
+    withTitle(title) {
+        this.product.title = title;
+        return this;
+    }
+
+    withDescription(description) {
+        this.product.description = description;
+        return this;
+    }
+
+    withPrice(price) {
+        this.product.price = price;
+        return this;
+    }
+
+    withCount(count) {
+        this.product.count = count;
+        return this;
+    }
+
+    build() {
+        return this.product;
+    }
+}
+
+module.exports = ProductBuilder;

product_service/utils/stockBuilder.js

@@ -0,0 +1,24 @@
+class StockBuilder {
+    constructor() {
+        this.stock = {
+            product_id: 'ead768ea-8752-4417-a94d-e8eaf256c500',
+            count: 50,
+        };
+    }
+
+    withId(product_id) {
+        this.stock.product_id = product_id;
+        return this;
+    }
+
+    withCount(count) {
+        this.stock.count = count;
+        return this;
+    }
+
+    build() {
+        return this.stock;
+    }
+}
+
+module.exports = StockBuilder;