test: isCostInPoly in the stos

Author: Victor Wiebe

src/procedures/__tests__/LaunchCappedSto.ts

@@ -129,6 +129,26 @@ describe('LaunchCappedSto', () => {
       expect(addProcedureSpy.callCount).toEqual(1);
     });
 
+    test('should add a transaction to the queue to launch a capped sto with cost in poly', async () => {
+      const addProcedureSpy = spy(target, 'addProcedure');
+      const addTransactionSpy = spy(target, 'addTransaction');
+
+      moduleFactoryMock.mock('isCostInPoly', Promise.resolve(true));
+
+      // Real call
+      await target.prepareTransactions();
+
+      // Verifications
+      expect(
+        addTransactionSpy
+          .getCall(0)
+          .calledWith(securityTokenMock.getMockInstance().addModuleWithLabel)
+      ).toEqual(true);
+      expect(addTransactionSpy.callCount).toEqual(1);
+      expect(addProcedureSpy.getCall(0).calledWith(TransferErc20)).toEqual(true);
+      expect(addProcedureSpy.callCount).toEqual(1);
+    });
+
     test('should throw if corresponding capped sto event is not fired', async () => {
       ImportMock.mockFunction(utilsModule, 'findEvents', []);
 

src/procedures/__tests__/LaunchUsdTieredSto.ts

@@ -32,8 +32,6 @@ const params: LaunchUsdTieredStoProcedureArgs = {
     {
       tokensOnSale: new BigNumber(1),
       price: new BigNumber(1),
-      discountedPrice: new BigNumber(1),
-      tokensWithDiscount: new BigNumber(1),
     },
   ],
   nonAccreditedInvestmentLimit: new BigNumber(1),
@@ -151,6 +149,25 @@ describe('LaunchUsdTieredSto', () => {
       expect(addProcedureSpy.callCount).toEqual(1);
     });
 
+    test('should add the transaction to the queue to launch usd tiered sto with cost in poly', async () => {
+      const addProcedureSpy = spy(target, 'addProcedure');
+      const addTransactionSpy = spy(target, 'addTransaction');
+
+      moduleFactoryMock.mock('isCostInPoly', Promise.resolve(true));
+      // Real call
+      await target.prepareTransactions();
+
+      // Verifications
+      expect(
+        addTransactionSpy
+          .getCall(0)
+          .calledWith(securityTokenMock.getMockInstance().addModuleWithLabel)
+      ).toEqual(true);
+      expect(addTransactionSpy.callCount).toEqual(1);
+      expect(addProcedureSpy.getCall(0).calledWith(TransferErc20)).toEqual(true);
+      expect(addProcedureSpy.callCount).toEqual(1);
+    });
+
     test('should throw if corresponding usd tiered sto event is not fired', async () => {
       findEventsStub = ImportMock.mockFunction(utilsModule, 'findEvents', []);