Skip to content
This repository has been archived by the owner on Jul 6, 2022. It is now read-only.

Commit

Permalink
Merge f3d3722 into 4fdf427
Browse files Browse the repository at this point in the history
  • Loading branch information
@onigiri-x
onigiri-x committed Jun 7, 2019
2 parents 4fdf427 + f3d3722 commit fbe61cb
Show file tree
Hide file tree
Showing 14 changed files with 717 additions and 744 deletions.
73 changes: 0 additions & 73 deletions src/contract_wrappers/modules/sto/__tests__/capped_sto_factory_wrapper.test.ts
View file

This file was deleted.

98 changes: 80 additions & 18 deletions src/contract_wrappers/modules/sto/__tests__/capped_sto_wrapper.test.ts
View file
Original file line number Diff line number Diff line change
Expand Up @@ -12,8 +12,8 @@ import { getMockedPolyResponse, MockedCallMethod, MockedSendMethod } from '../..
import CappedSTOWrapper from '../capped_sto_wrapper';
import ContractFactory from '../../../../factories/contractFactory';
import STOWrapper from '../sto_wrapper';
import { dateToBigNumber } from '../../../../utils/convert';
import { FundRaiseType } from '../../../../types';
import { dateToBigNumber, valueToWei, weiToValue } from '../../../../utils/convert';
import { FULL_DECIMALS, FundRaiseType } from '../../../../types';

describe('CappedSTOWrapper', () => {
// Capped STO Wrapper is used as contract target here as STOWrapper is abstract
Expand Down Expand Up @@ -63,7 +63,7 @@ describe('CappedSTOWrapper', () => {
// Real call
const result = await target.rate();
// Result expectation
expect(result).toBe(expectedResult);
expect(result).toEqual(weiToValue(expectedResult, FULL_DECIMALS));
// Verifications
verify(mockedContract.rate).once();
verify(mockedMethod.callAsync()).once();
Expand All @@ -74,6 +74,20 @@ describe('CappedSTOWrapper', () => {
test('should get the cap', async () => {
// Address expected
const expectedResult = new BigNumber(1);

// Security Token, its address, and decimals mocked
const expectedSecurityTokenAddress = '0x3333333333333333333333333333333333333333';
const mockedGetSecurityTokenAddressMethod = mock(MockedCallMethod);
when(mockedGetSecurityTokenAddressMethod.callAsync()).thenResolve(expectedSecurityTokenAddress);
when(mockedContract.securityToken).thenReturn(instance(mockedGetSecurityTokenAddressMethod));
when(mockedContractFactory.getSecurityTokenContract(expectedSecurityTokenAddress)).thenResolve(
instance(mockedSecurityTokenContract),
);
const expectedDecimalsResult = new BigNumber(18);
const mockedDecimalsMethod = mock(MockedCallMethod);
when(mockedSecurityTokenContract.decimals).thenReturn(instance(mockedDecimalsMethod));
when(mockedDecimalsMethod.callAsync()).thenResolve(expectedDecimalsResult);

// Mocked method
const mockedMethod = mock(MockedCallMethod);
// Stub the method
Expand All @@ -84,10 +98,16 @@ describe('CappedSTOWrapper', () => {
// Real call
const result = await target.cap();
// Result expectation
expect(result).toBe(expectedResult);
expect(result).toEqual(weiToValue(expectedResult, expectedDecimalsResult));
// Verifications
verify(mockedContract.cap).once();
verify(mockedMethod.callAsync()).once();

verify(mockedSecurityTokenContract.decimals).once();
verify(mockedDecimalsMethod.callAsync()).once();
verify(mockedGetSecurityTokenAddressMethod.callAsync()).once();
verify(mockedContract.securityToken).once();
verify(mockedContractFactory.getSecurityTokenContract(expectedSecurityTokenAddress)).once();
});
});

Expand Down Expand Up @@ -236,6 +256,20 @@ describe('CappedSTOWrapper', () => {
test('should get the tokens sold', async () => {
// Address expected
const expectedResult = new BigNumber(1);

// Security Token, its address, and decimals mocked
const expectedSecurityTokenAddress = '0x3333333333333333333333333333333333333333';
const mockedGetSecurityTokenAddressMethod = mock(MockedCallMethod);
when(mockedGetSecurityTokenAddressMethod.callAsync()).thenResolve(expectedSecurityTokenAddress);
when(mockedContract.securityToken).thenReturn(instance(mockedGetSecurityTokenAddressMethod));
when(mockedContractFactory.getSecurityTokenContract(expectedSecurityTokenAddress)).thenResolve(
instance(mockedSecurityTokenContract),
);
const expectedDecimalsResult = new BigNumber(18);
const mockedDecimalsMethod = mock(MockedCallMethod);
when(mockedSecurityTokenContract.decimals).thenReturn(instance(mockedDecimalsMethod));
when(mockedDecimalsMethod.callAsync()).thenResolve(expectedDecimalsResult);

// Mocked method
const mockedMethod = mock(MockedCallMethod);
// Stub the method
Expand All @@ -246,27 +280,50 @@ describe('CappedSTOWrapper', () => {
// Real call
const result = await target.getTokensSold();
// Result expectation
expect(result).toBe(expectedResult);
expect(result).toEqual(weiToValue(expectedResult, expectedDecimalsResult));
// Verifications
verify(mockedContract.getTokensSold).once();
verify(mockedMethod.callAsync()).once();
verify(mockedGetSecurityTokenAddressMethod.callAsync()).once();
verify(mockedContract.securityToken).once();
verify(mockedContractFactory.getSecurityTokenContract(expectedSecurityTokenAddress)).once();
verify(mockedSecurityTokenContract.decimals).once();
verify(mockedDecimalsMethod.callAsync()).once();
});
});

describe('GetSTODetails', () => {
test('should call getSTODetails', async () => {
const expectedStartTime = new Date(2025, 1);
const expectedEndTime = new Date(2026, 1);
const cap = new BigNumber(1);
const rate = new BigNumber(1);
const fundsRaised = new BigNumber(1);
const totalTokensSold = new BigNumber(1);
const expectedResult = [
dateToBigNumber(expectedStartTime),
dateToBigNumber(expectedEndTime),
cap,
rate,
fundsRaised,
new BigNumber(1),
new BigNumber(1),
new BigNumber(1),
new BigNumber(1),
new BigNumber(1),
totalTokensSold,
true,
];

// Security Token, its address, and decimals mocked
const expectedSecurityTokenAddress = '0x3333333333333333333333333333333333333333';
const mockedGetSecurityTokenAddressMethod = mock(MockedCallMethod);
when(mockedGetSecurityTokenAddressMethod.callAsync()).thenResolve(expectedSecurityTokenAddress);
when(mockedContract.securityToken).thenReturn(instance(mockedGetSecurityTokenAddressMethod));
when(mockedContractFactory.getSecurityTokenContract(expectedSecurityTokenAddress)).thenResolve(
instance(mockedSecurityTokenContract),
);
const expectedDecimalsResult = new BigNumber(18);
const mockedDecimalsMethod = mock(MockedCallMethod);
when(mockedSecurityTokenContract.decimals).thenReturn(instance(mockedDecimalsMethod));
when(mockedDecimalsMethod.callAsync()).thenResolve(expectedDecimalsResult);

// Mocked method
const mockedMethod = mock(MockedCallMethod);
// Stub the method
Expand All @@ -279,16 +336,21 @@ describe('CappedSTOWrapper', () => {
// Result expectation
expect(result.startTime).toEqual(expectedStartTime);
expect(result.endTime).toEqual(expectedEndTime);
expect(result.cap).toBe(expectedResult[2]);
expect(result.rate).toBe(expectedResult[3]);
expect(result.fundsRaised).toBe(expectedResult[4]);
expect(result.cap).toEqual(weiToValue(cap, expectedDecimalsResult));
expect(result.rate).toEqual(weiToValue(rate, FULL_DECIMALS));
expect(result.fundsRaised).toEqual(weiToValue(fundsRaised, FULL_DECIMALS));
expect(result.investorCount).toEqual(1);
expect(result.totalTokensSold).toBe(expectedResult[6]);
expect(result.totalTokensSold).toEqual(weiToValue(totalTokensSold, expectedDecimalsResult));
expect(result.isRaisedInPoly).toBe(expectedResult[7]);

// Verifications
verify(mockedContract.getSTODetails).once();
verify(mockedMethod.callAsync()).once();
verify(mockedGetSecurityTokenAddressMethod.callAsync()).once();
verify(mockedContract.securityToken).once();
verify(mockedContractFactory.getSecurityTokenContract(expectedSecurityTokenAddress)).once();
verify(mockedSecurityTokenContract.decimals).once();
verify(mockedDecimalsMethod.callAsync()).once();
});
});

Expand All @@ -305,7 +367,7 @@ describe('CappedSTOWrapper', () => {

// Owner Address expected
const expectedOwnerResult = '0x5555555555555555555555555555555555555555';
const expectedETHBalanceResult = new BigNumber(100);
const expectedETHBalanceResult = valueToWei(new BigNumber(100), FULL_DECIMALS);

// Mock web3 wrapper owner
when(mockedWrapper.getAvailableAddressesAsync()).thenResolve([expectedOwnerResult]);
Expand Down Expand Up @@ -354,7 +416,7 @@ describe('CappedSTOWrapper', () => {
};
const txPayableData = {
...{},
value: mockedParams.value,
value: valueToWei(mockedParams.value, FULL_DECIMALS),
};

const expectedResult = getMockedPolyResponse();
Expand Down Expand Up @@ -440,7 +502,7 @@ describe('CappedSTOWrapper', () => {
when(mockedEndTimeMethod.callAsync()).thenResolve(expectedEndTimeResult);

// Security Token Address expected
const expectedBalanceOfResult = new BigNumber(100);
const expectedBalanceOfResult = valueToWei(new BigNumber(100), FULL_DECIMALS);
// Setup get Security Token Address
const mockedBalanceOfAddressMethod = mock(MockedCallMethod);
when(mockedPolyTokenContract.balanceOf).thenReturn(instance(mockedBalanceOfAddressMethod));
Expand All @@ -464,7 +526,7 @@ describe('CappedSTOWrapper', () => {
// Stub the request
when(
mockedMethod.sendTransactionAsync(
objectContaining(mockedParams.investedPOLY),
objectContaining(valueToWei(mockedParams.investedPOLY, FULL_DECIMALS)),
mockedParams.txData,
mockedParams.safetyFactor,
),
Expand All @@ -479,7 +541,7 @@ describe('CappedSTOWrapper', () => {
verify(mockedContract.buyTokensWithPoly).once();
verify(
mockedMethod.sendTransactionAsync(
objectContaining(mockedParams.investedPOLY),
objectContaining(valueToWei(mockedParams.investedPOLY, FULL_DECIMALS)),
mockedParams.txData,
mockedParams.safetyFactor,
),
Expand Down

0 comments on commit fbe61cb

Please sign in to comment.