#32-unit-tests-contract-service

src/services/contract.service.spec.ts

@@ -0,0 +1,36 @@
+import { ConfigService } from './config.service';
+import { Test, TestingModule } from '@nestjs/testing';
+import { ContractService } from './contract.service';
+
+describe('ContractService', () => {
+    let contractService: ContractService;
+    let configService: ConfigService;
+    let moduleRef:TestingModule;
+    beforeAll(async () => {
+        //mock conifg service
+        const configServiceMock = {
+            get: (key: string) => {
+                return "random string"
+            },
+            getCreds: () => {
+                return {
+                    key: 'something',
+                    secret: 'secret',
+                    passphrase: 'passphrase'
+                }
+            }
+        };
+        moduleRef = await Test.createTestingModule({
+            providers: [ConfigService],
+        }).overrideProvider(ConfigService)
+            .useValue(configServiceMock).compile();
+    });
+
+    it('should console.log that rpc provider or private key is incorrect', async () => {
+        const consoleLogSpy = jest.spyOn(console, 'log');
+        configService = moduleRef.get<ConfigService>(ConfigService);
+        contractService = new ContractService(configService);
+        expect(consoleLogSpy).toHaveBeenCalledWith('Please provide valid private key and rpc provider in config.json file');
+    });
+
+});

src/services/polymarket.service.ts

@@ -15,20 +15,28 @@ type FeeResponse = { side: 'taker' | 'maker', fee: number };
 @Injectable()
 export class PolymarketService {
   clobClient: ClobClient;
+  signer: ethers.Wallet;
 
   constructor(private configService: ConfigService) {
     const provider = new ethers.providers.JsonRpcProvider(
       this.configService.get("rpcProvider")
     );
-    const signer = new ethers.Wallet(
-      this.configService.get("privateKey"),
-      provider
-    );
+    try {
+      this.signer = new ethers.Wallet(
+        this.configService.get("privateKey"),
+        provider
+      );
+    } catch (error) {
+      console.log(
+        "Please provide valid private key and rpc provider in config.json file"
+      );
+    }
+
     creds = this.configService.getCreds();
     this.clobClient = new ClobClient(
       CLOB_API_ENDPOINT,
       Chain.POLYGON,
-      signer,
+      this.signer,
       creds
     );
   }