implement tests

Author: aggre

.eslintrc.json

@@ -27,7 +27,11 @@
 				"functional/functional-parameters": "off",
 				"functional/no-return-void": "off",
 				"@typescript-eslint/prefer-readonly-parameter-types": "off",
-				"@typescript-eslint/no-explicit-any": "off"
+				"@typescript-eslint/no-explicit-any": "off",
+				"functional/no-throw-statements": "off",
+				"functional/no-classes": "off",
+				"functional/no-conditional-statements": "off",
+				"functional/immutable-data": "off"
 			}
 		}
 	]

package.json

@@ -8,8 +8,8 @@
 		"start": "astro dev",
 		"build": "astro build",
 		"preview": "astro preview",
-		"test": "echo 'Skip'",
-		"pretest": "tsc",
+		"test": "vitest",
+		"test:ci": "vitest --run",
 		"prebuild": "rimraf dist",
 		"lint": "yarn lint:eslint && yarn lint:format",
 		"lint:eslint": "eslint . --ext .ts,.js --fix",
@@ -47,7 +47,8 @@
 		"prettier": "3.2.4",
 		"rimraf": "5.0.5",
 		"type-fest": "^4.10.2",
-		"typescript": "5.3.3"
+		"typescript": "5.3.3",
+		"vitest": "^1.2.2"
 	},
 	"repository": "git+https://github.com/dev-protocol/template-repos-ts.git",
 	"bugs": {

utils/tx.test.ts

@@ -0,0 +1,88 @@
+import { expect, describe, it, vi } from 'vitest'
+import { send } from './tx'
+import redis, { createClient } from 'redis'
+import {
+	Contract,
+	JsonRpcProvider,
+	Wallet,
+	type TransactionRequest,
+} from 'ethers'
+import { generateTransactionKey } from './db'
+import { agentAddresses } from '@devprotocol/dev-kit'
+
+enum MockUses {
+	Default = 'default',
+	Error = 'error',
+}
+const redisData = new Map()
+const redisConnectUses: MockUses = MockUses.Default
+
+vi.mock('redis', async () => {
+	const actual: typeof redis = await vi.importActual('redis')
+	const lib = vi.fn(() => ({
+		connect: vi.fn(() => {
+			if (redisConnectUses === MockUses.Default) {
+				return null
+			}
+			if (redisConnectUses === MockUses.Error) {
+				throw new Error('REDIS ERROR')
+			}
+		}),
+		get: vi.fn(async (k) => redisData.get(k)),
+		set: vi.fn(async (k, v) => redisData.set(k, v)),
+		quit: vi.fn(),
+	}))
+
+	return { ...actual, default: actual, createClient: lib }
+})
+vi.mock('JsonRpcProvider', () => {
+	return class {
+		public async getFeeData() {
+			return { maxFeePerGas: 1n, maxPriorityFeePerGas: 1n }
+		}
+	}
+})
+
+describe('send', () => {
+	describe('Normal cases', () => {
+		const redis = createClient()
+		const provider = new JsonRpcProvider(
+			'https://polygon-mumbai-bor-rpc.publicnode.com',
+		)
+		const signer = Wallet.createRandom().connect(provider)
+		vi.spyOn(signer, 'sendTransaction').mockImplementation(
+			async (tx: TransactionRequest) => {
+				return ['sent', tx] as any
+			},
+		)
+
+		it('should return tx', async () => {
+			const contract = new Contract(
+				agentAddresses.polygon.mumbai.weth,
+				['function transfer(address, uint) public'],
+				signer,
+			)
+
+			const { to, data } = await contract.transfer.populateTransaction(
+				signer.address,
+				0n,
+			)
+			redisData.set(
+				generateTransactionKey(to, data, 'id'),
+				new Date().getTime() - 60001,
+			)
+			const res = await send({
+				contract,
+				method: 'transfer',
+				signer,
+				chainId: 80001,
+				rpcUrl: 'https://polygon-mumbai-bor-rpc.publicnode.com/', // Mumbai
+				args: [signer.address, 0n],
+				requestId: 'id',
+				redis,
+			})
+			console.log(res)
+			expect((res as any)[0]).toBe('sent')
+		})
+	})
+})

vitest.config.ts

@@ -0,0 +1,4 @@
+/// <reference types="vitest" />
+import { getViteConfig } from 'astro/config'
+
+export default getViteConfig({})