added tests for locktime

cashubtc · Dec 7, 2024 · 02604d6 · 02604d6
1 parent a5ad719
commit 02604d6
Showing 1 changed file with 50 additions and 0 deletions.
diff --git a/test/wallet.test.ts b/test/wallet.test.ts
@@ -15,6 +15,7 @@ import { getDecodedToken } from '../src/utils.js';
 import { Server, WebSocket } from 'mock-socket';
 import { injectWebSocketImpl } from '../src/ws.js';
 import { MintInfo } from '../src/model/MintInfo.js';
+import { BlindingData } from '../src/model/BlindingData.js';
 
 injectWebSocketImpl(WebSocket);
 
@@ -735,3 +736,52 @@ describe('WebSocket Updates', () => {
 		server.close();
 	});
 });
+
+describe('P2PK BlindingData', () => {
+	test('Create BlindingData locked to pk with locktime and single refund key', async () => {
+		const wallet = new CashuWallet(mint);
+		const keys = await wallet.getKeys();
+		const data = BlindingData.createP2PKData(
+			{ pubkey: 'thisisatest', locktime: 212, refundKeys: ['iamarefund'] },
+			21,
+			keys
+		);
+		const decoder = new TextDecoder();
+		const allSecrets = data.map((d) => JSON.parse(decoder.decode(d.secret)));
+		allSecrets.forEach((s) => {
+			expect(s[0] === 'P2PK');
+			expect(s[1].data).toBe('thisisatest');
+			expect(s[1].tags).toContainEqual(['locktime', 212]);
+			expect(s[1].tags).toContainEqual(['refund', ['iamarefund']]);
+		});
+	});
+	test('Create BlindingData locked to pk with locktime and multiple refund keys', async () => {
+		const wallet = new CashuWallet(mint);
+		const keys = await wallet.getKeys();
+		const data = BlindingData.createP2PKData(
+			{ pubkey: 'thisisatest', locktime: 212, refundKeys: ['iamarefund', 'asecondrefund'] },
+			21,
+			keys
+		);
+		const decoder = new TextDecoder();
+		const allSecrets = data.map((d) => JSON.parse(decoder.decode(d.secret)));
+		allSecrets.forEach((s) => {
+			expect(s[0] === 'P2PK');
+			expect(s[1].data).toBe('thisisatest');
+			expect(s[1].tags).toContainEqual(['locktime', 212]);
+			expect(s[1].tags).toContainEqual(['refund', ['iamarefund', 'asecondrefund']]);
+		});
+	});
+	test('Create BlindingData locked to pk without locktime and no refund keys', async () => {
+		const wallet = new CashuWallet(mint);
+		const keys = await wallet.getKeys();
+		const data = BlindingData.createP2PKData({ pubkey: 'thisisatest' }, 21, keys);
+		const decoder = new TextDecoder();
+		const allSecrets = data.map((d) => JSON.parse(decoder.decode(d.secret)));
+		allSecrets.forEach((s) => {
+			expect(s[0] === 'P2PK');
+			expect(s[1].data).toBe('thisisatest');
+			expect(s[1].tags).toEqual([]);
+		});
+	});
+});