add test for agent/lockup

Author: 0xkenta

lib/agent/lockup/calculateCumulativeHoldersRewardAmount.spec.ts

@@ -0,0 +1,31 @@
+import { ethers } from 'ethers'
+import { testProviders } from '../common/const'
+import { calculateCumulativeHoldersRewardAmount } from './calculateCumulativeHoldersRewardAmount'
+import { getLockupContract } from './common'
+
+jest.mock('./common', () => {
+	return {
+		getLockupContract: jest.fn(),
+	}
+})
+
+describe('calculateCumulativeHoldersRewardAmount.ts', () => {
+	const options = {
+		provider: testProviders.ropsten,
+		propertyAddress: '0x38c4bF6cD20d157EE45553b0fAD13B0c6750b439',
+	}
+	it('success', async () => {
+		const expected = '100'
+		;(getLockupContract as jest.Mock).mockReturnValue({
+			calculateCumulativeHoldersRewardAmount: (property: string) => expected,
+		})
+		const result = await calculateCumulativeHoldersRewardAmount(options)
+		expect(result).toEqual(expected)
+	})
+	it('return undefined if no valid network', async () => {
+		const expected = undefined
+		;(getLockupContract as jest.Mock).mockReturnValue(undefined)
+		const result = await calculateCumulativeHoldersRewardAmount(options)
+		expect(result).toEqual(expected)
+	})
+})

lib/agent/lockup/calculateRewardAmount.spec.ts

@@ -0,0 +1,32 @@
+import { calculateRewardAmount } from './calculateRewardAmount'
+import { getLockupContract } from './common'
+import { testProviders } from '../common/const'
+
+jest.mock('./common', () => {
+	return {
+		getLockupContract: jest.fn(),
+	}
+})
+describe('calculateRewardAmount.ts', () => {
+	const options = {
+		provider: testProviders.ropsten,
+		propertyAddress: '0x38c4bF6cD20d157EE45553b0fAD13B0c6750b439',
+	}
+
+	it('success', async () => {
+		const expected = ['abc', '123']
+		;(getLockupContract as jest.Mock).mockReturnValue({
+			calculateRewardAmount: (propertyAddress: string) => expected,
+		})
+		const result = await calculateRewardAmount(options)
+		expect(result).toEqual(expected)
+	})
+	it('return undefined if no valid provider', async () => {
+		const expected = undefined
+		;(getLockupContract as jest.Mock).mockReturnValue({
+			calculateRewardAmount: (propertyAddress: string) => undefined,
+		})
+		const result = await calculateRewardAmount(options)
+		expect(result).toEqual(expected)
+	})
+})

lib/agent/lockup/calculateWithdrawableInterestAmountByPosition.spec.ts

@@ -0,0 +1,34 @@
+import { calculateWithdrawableInterestAmountByPosition } from './calculateWithdrawableInterestAmountByPosition'
+import { getLockupContract } from './common'
+import { testProviders } from '../common/const'
+
+jest.mock('./common', () => {
+	return {
+		getLockupContract: jest.fn(),
+	}
+})
+
+describe('calculateWithdrawableInterestAmountByPosition.ts', () => {
+	const options = {
+		provider: testProviders.ropsten,
+		positionTokenId: '1',
+	}
+
+	it('success', async () => {
+		const expected = '123'
+		;(getLockupContract as jest.Mock).mockReturnValue({
+			calculateWithdrawableInterestAmountByPosition: (
+				positionTokenId: string
+			) => expected,
+		})
+		const result = await calculateWithdrawableInterestAmountByPosition(options)
+		expect(result).toEqual(expected)
+	})
+
+	it('return undefined if no valid network', async () => {
+		const expected = undefined
+		;(getLockupContract as jest.Mock).mockReturnValue(undefined)
+		const result = await calculateWithdrawableInterestAmountByPosition(options)
+		expect(result).toEqual(expected)
+	})
+})

lib/agent/lockup/cap.spec.ts

@@ -0,0 +1,31 @@
+import { getCap } from './cap'
+import { getLockupContract } from './common'
+import { testProviders } from '../common/const'
+
+jest.mock('./common', () => {
+	return {
+		getLockupContract: jest.fn(),
+	}
+})
+
+describe('cap.ts', () => {
+	const options = {
+		provider: testProviders.ropsten,
+	}
+
+	it('success', async () => {
+		const expected = '123456'
+		;(getLockupContract as jest.Mock).mockReturnValue({
+			cap: () => expected,
+		})
+		const result = await getCap(options)
+		expect(result).toEqual(expected)
+	})
+
+	it('return undefined if no valid provider', async () => {
+		const expected = undefined
+		;(getLockupContract as jest.Mock).mockReturnValue(expected)
+		const result = await getCap(options)
+		expect(result).toEqual(expected)
+	})
+})

lib/agent/lockup/depositToPosition.spec.ts

@@ -0,0 +1,33 @@
+import { depositToPosition } from './depositToPosition'
+import { getLockupContract } from './common'
+import { testProviders } from '../common/const'
+
+jest.mock('./common', () => {
+	return {
+        getLockupContract: jest.fn()
+	}
+})
+
+describe('depositToPosition.ts', () => {
+	const options = {
+		provider: testProviders.ropsten,
+		positionTokenId: '1',
+		amount: '100'
+	}
+	it('success', async () => {
+		const expected = true
+
+		;(getLockupContract as jest.Mock).mockReturnValue({
+			depositToPosition: (positionTokenId: string, amount: string) => expected
+		})
+
+		const result = await depositToPosition(options)
+		expect(result).toEqual(expected)
+	})
+    it('return undefined if no valid network', async () => {
+		const expected = undefined
+		;(getLockupContract as jest.Mock).mockReturnValue(undefined)
+        const result = await depositToPosition(options)
+		expect(result).toEqual(expected)
+	})
+})

lib/agent/lockup/withdrawByPosition.spec.ts

@@ -0,0 +1,34 @@
+import { testProviders } from '../common/const'
+import { getLockupContract } from './common'
+import { withdrawByPosition } from './withdrawByPosition'
+
+jest.mock('./common', () => {
+	return {
+		getLockupContract: jest.fn(),
+	}
+})
+
+describe('withdrawByPosition.ts', () => {
+	const options = {
+		provider: testProviders.ropsten,
+		positionTokenId: '1',
+		amount: '100',
+	}
+	it('success', async () => {
+		const expected = true
+		;(getLockupContract as jest.Mock).mockReturnValue({
+			withdrawByPosition: (positionTokenId: string, amount: string) => true,
+		})
+		const result = await withdrawByPosition(options)
+		expect(result).toEqual(expected)
+	})
+
+	it('return undefined if no valid provider', async () => {
+		const expected = undefined
+		;(getLockupContract as jest.Mock).mockReturnValue({
+			withdrawByPosition: () => undefined,
+		})
+		const result = await withdrawByPosition(options)
+		expect(result).toEqual(expected)
+	})
+})