Added final tests

src/shipments/shipments.service.ts

@@ -10,7 +10,6 @@ export class ShipmentsService {
     private readonly shipmentRepository: Repository<ShipmentEntity>,
   ) {}
 
-  // @todo add pagination
   getAll() {
     return this.shipmentRepository.find({
       relations: ['liveTracking', 'containers'],

tests/shipments/shipments.service.spec.ts

@@ -1,18 +1,59 @@
 import { Test, TestingModule } from '@nestjs/testing';
 import { ShipmentsService } from '@/shipments/shipments.service';
+import { getRepositoryToken } from '@nestjs/typeorm';
+import { ShipmentEntity } from '@/entities/shipment.entity';
+import { Repository } from 'typeorm';
+import { mockShipment } from '@tests/data/tracking.mock';
 
 describe('ShipmentsService', () => {
+  const shipmentEntityToken = getRepositoryToken(ShipmentEntity);
   let service: ShipmentsService;
+  let repository: Repository<ShipmentEntity>;
 
   beforeEach(async () => {
     const module: TestingModule = await Test.createTestingModule({
       providers: [ShipmentsService],
-    }).compile();
+    })
+      .useMocker((token) => {
+        switch (token) {
+          case shipmentEntityToken:
+            return {
+              find: jest.fn().mockResolvedValue([]),
+              save: jest
+                .fn()
+                .mockImplementation((shipmentEntity: ShipmentEntity) =>
+                  Promise.resolve(shipmentEntity),
+                ),
+            };
+        }
+      })
+      .compile();
 
     service = module.get<ShipmentsService>(ShipmentsService);
+    repository = module.get<Repository<ShipmentEntity>>(shipmentEntityToken);
   });
 
   it('should be defined', () => {
     expect(service).toBeDefined();
   });
+
+  it('should have a getAll method', () => {
+    expect(service.getAll).toBeDefined();
+  });
+
+  it('should fetch data with relations liveTracking and containers', async () => {
+    await expect(service.getAll()).resolves.toEqual([]);
+    expect(repository.find).toHaveBeenCalledWith({
+      relations: ['liveTracking', 'containers'],
+    });
+  });
+
+  it('should have an update method', () => {
+    expect(service.update).toBeDefined();
+  });
+
+  it('should fetch data with relations liveTracking and containers', async () => {
+    await expect(service.update(mockShipment)).resolves.toEqual(mockShipment);
+    expect(repository.save).toHaveBeenCalledWith(mockShipment);
+  });
 });