feat: add tests for main class and factory

Author: luchobonatti

test/core/uniDevKitV4.test.ts

@@ -0,0 +1,62 @@
+import { UniDevKitV4 } from "@/core/uniDevKitV4";
+import type { UniDevKitV4Config } from "@/types/core";
+import { beforeEach, describe, expect, it } from "vitest";
+
+describe("UniDevKitV4", () => {
+	let config: UniDevKitV4Config;
+	let sdk: UniDevKitV4;
+
+	beforeEach(() => {
+		config = {
+			chainId: 1,
+			rpcUrl: "https://eth.llamarpc.com",
+			contracts: {
+				poolManager: "0x1234567890123456789012345678901234567890",
+				positionDescriptor: "0x1234567890123456789012345678901234567890",
+				positionManager: "0x1234567890123456789012345678901234567890",
+				quoter: "0x1234567890123456789012345678901234567890",
+				stateView: "0x1234567890123456789012345678901234567890",
+				universalRouter: "0x1234567890123456789012345678901234567890",
+				permit2: "0x1234567890123456789012345678901234567890",
+			},
+		};
+		sdk = new UniDevKitV4(config);
+	});
+
+	it("should initialize with correct config", () => {
+		expect(sdk.getChainId()).toBe(config.chainId);
+		expect(sdk.getContracts()).toEqual(config.contracts);
+	});
+
+	it("should get contract address", () => {
+		expect(sdk.getContractAddress("quoter")).toBe(config.contracts.quoter);
+	});
+
+	it("should throw error for non-existent contract", () => {
+		// @ts-expect-error Testing invalid contract name
+		expect(() => sdk.getContractAddress("invalid")).toThrow();
+	});
+
+	it("should update config", () => {
+		const newConfig: UniDevKitV4Config = {
+			...config,
+			chainId: 5,
+			rpcUrl: "https://goerli.infura.io/v3/your-api-key",
+		};
+		sdk.updateConfig(newConfig);
+		expect(sdk.getChainId()).toBe(newConfig.chainId);
+	});
+
+	it("should create client with custom native currency", () => {
+		const customConfig: UniDevKitV4Config = {
+			...config,
+			nativeCurrency: {
+				name: "Custom Token",
+				symbol: "CTK",
+				decimals: 18,
+			},
+		};
+		const customSdk = new UniDevKitV4(customConfig);
+		expect(customSdk.getClient()).toBeDefined();
+	});
+});

test/core/uniDevKitV4Factory.test.ts

@@ -0,0 +1,142 @@
+import {
+	createInstance,
+	getInstance,
+	listChainIds,
+	removeInstance,
+	resetInstances,
+} from "@/core/uniDevKitV4Factory";
+import type { UniDevKitV4Config } from "@/types/core";
+import { afterEach, beforeEach, describe, expect, it } from "vitest";
+
+describe("UniDevKitV4Factory", () => {
+	let config: UniDevKitV4Config;
+
+	beforeEach(() => {
+		config = {
+			chainId: 1,
+			rpcUrl: "https://eth.llamarpc.com",
+			contracts: {
+				poolManager: "0x1234567890123456789012345678901234567890",
+				positionDescriptor: "0x1234567890123456789012345678901234567890",
+				positionManager: "0x1234567890123456789012345678901234567890",
+				quoter: "0x1234567890123456789012345678901234567890",
+				stateView: "0x1234567890123456789012345678901234567890",
+				universalRouter: "0x1234567890123456789012345678901234567890",
+				permit2: "0x1234567890123456789012345678901234567890",
+			},
+		};
+	});
+
+	afterEach(() => {
+		resetInstances();
+	});
+
+	it("should create and get instance", () => {
+		const instance = createInstance(config);
+		const retrievedInstance = getInstance(config.chainId);
+		expect(retrievedInstance).toBe(instance);
+	});
+
+	it("should reuse existing instance", () => {
+		const instance1 = createInstance(config);
+		const instance2 = createInstance(config);
+		expect(instance1).toBe(instance2);
+	});
+
+	it("should list chain IDs", () => {
+		createInstance(config);
+		const chainIds = listChainIds();
+		expect(chainIds).toContain(config.chainId);
+	});
+
+	it("should remove instance", () => {
+		createInstance(config);
+		removeInstance(config.chainId);
+		expect(() => getInstance(config.chainId)).toThrow();
+	});
+
+	it("should reset all instances", () => {
+		createInstance(config);
+		resetInstances();
+		expect(() => getInstance(config.chainId)).toThrow();
+	});
+
+	it("should throw when getting non-existent instance", () => {
+		expect(() => getInstance(999)).toThrow();
+	});
+
+	it("should get default instance when only one chain is configured", () => {
+		const instance = createInstance(config);
+		const defaultInstance = getInstance();
+		expect(defaultInstance).toBe(instance);
+	});
+
+	it("should throw when getting default instance with multiple chains", () => {
+		createInstance(config);
+		createInstance({ ...config, chainId: 5 });
+		expect(() => getInstance()).toThrow(
+			"Multiple instances found. Please specify a chain ID. Available chains: 1, 5",
+		);
+	});
+
+	describe("Multiple chain configurations", () => {
+		it("should handle multiple chains with different RPCs", () => {
+			const mainnet = createInstance(config);
+			const goerli = createInstance({
+				...config,
+				chainId: 5,
+				rpcUrl: "https://goerli.infura.io/v3/your-api-key",
+			});
+			const sepolia = createInstance({
+				...config,
+				chainId: 11155111,
+				rpcUrl: "https://sepolia.infura.io/v3/your-api-key",
+			});
+
+			expect(getInstance(1)).toBe(mainnet);
+			expect(getInstance(5)).toBe(goerli);
+			expect(getInstance(11155111)).toBe(sepolia);
+		});
+
+		it("should handle multiple chains with different native currencies", () => {
+			const mainnet = createInstance(config);
+			const polygon = createInstance({
+				...config,
+				chainId: 137,
+				rpcUrl: "https://polygon-rpc.com",
+				nativeCurrency: {
+					name: "MATIC",
+					symbol: "MATIC",
+					decimals: 18,
+				},
+			});
+
+			expect(getInstance(1)).toBe(mainnet);
+			expect(getInstance(137)).toBe(polygon);
+		});
+
+		it("should list all configured chain IDs", () => {
+			createInstance(config); // mainnet
+			createInstance({ ...config, chainId: 5 }); // goerli
+			createInstance({ ...config, chainId: 137 }); // polygon
+
+			const chainIds = listChainIds();
+			expect(chainIds).toHaveLength(3);
+			expect(chainIds).toContain(1);
+			expect(chainIds).toContain(5);
+			expect(chainIds).toContain(137);
+		});
+
+		it("should remove specific chain instance", () => {
+			createInstance(config); // mainnet
+			createInstance({ ...config, chainId: 5 }); // goerli
+			createInstance({ ...config, chainId: 137 }); // polygon
+
+			removeInstance(5);
+			const chainIds = listChainIds();
+			expect(chainIds).toHaveLength(2);
+			expect(chainIds).not.toContain(5);
+			expect(() => getInstance(5)).toThrow();
+		});
+	});
+});