UNPKG

filecoin-pin

Version:

IPFS Pinning Service API implementation that pins to Filecoin's PDP service

243 lines 10.9 kB
import { describe, it, expect, beforeEach, afterEach } from 'vitest'; import { rm, readFile } from 'node:fs/promises'; import { existsSync } from 'node:fs'; import { CARWritingBlockstore } from '../../car-blockstore.js'; import { CID } from 'multiformats/cid'; import { sha256 } from 'multiformats/hashes/sha2'; import * as raw from 'multiformats/codecs/raw'; describe('CARWritingBlockstore', () => { let blockstore; let testCID; let testBlock; const testOutputPath = './test-output.car'; beforeEach(async () => { // Create a test block and CID testBlock = new TextEncoder().encode('Hello, IPFS!'); const hash = await sha256.digest(testBlock); testCID = CID.create(1, raw.code, hash); blockstore = new CARWritingBlockstore({ rootCID: testCID, outputPath: testOutputPath }); }); afterEach(async () => { // Clean up test files await blockstore.cleanup(); if (existsSync(testOutputPath)) { await rm(testOutputPath); } }); describe('Initialization', () => { it('should initialize with root CID and output path', () => { expect(blockstore).toBeDefined(); const stats = blockstore.getStats(); expect(stats.blocksWritten).toBe(0); expect(stats.totalSize).toBe(0); expect(stats.finalized).toBe(false); }); it('should emit initialized event', async () => { let eventData; blockstore.on('initialized', (data) => { eventData = data; }); await blockstore.initialize(); expect(eventData).toBeDefined(); expect(eventData.rootCID).toEqual(testCID); expect(eventData.outputPath).toBe(testOutputPath); }); }); describe('Block Operations', () => { it('should put a block and update statistics', async () => { const returnedCID = await blockstore.put(testCID, testBlock); expect(returnedCID).toEqual(testCID); const stats = blockstore.getStats(); expect(stats.blocksWritten).toBe(1); expect(stats.totalSize).toBe(testBlock.length); }); it('should emit block:stored event when putting blocks', async () => { let eventData; blockstore.on('block:stored', (data) => { eventData = data; }); await blockstore.put(testCID, testBlock); expect(eventData).toBeDefined(); expect(eventData.cid).toEqual(testCID); expect(eventData.size).toBe(testBlock.length); }); it('should get a block that was previously put', async () => { await blockstore.put(testCID, testBlock); const retrievedBlock = await blockstore.get(testCID); expect(retrievedBlock).toEqual(testBlock); }); it('should throw error when getting non-existent block', async () => { const nonExistentHash = await sha256.digest(new TextEncoder().encode('nonexistent')); const nonExistentCID = CID.create(1, raw.code, nonExistentHash); await expect(blockstore.get(nonExistentCID)).rejects.toThrow('Block not found'); }); it('should emit block:missing event for missing blocks', async () => { let eventData; blockstore.on('block:missing', (data) => { eventData = data; }); const nonExistentHash = await sha256.digest(new TextEncoder().encode('nonexistent')); const nonExistentCID = CID.create(1, raw.code, nonExistentHash); try { await blockstore.get(nonExistentCID); } catch (error) { // Expected to throw } expect(eventData).toBeDefined(); expect(eventData.cid).toEqual(nonExistentCID); const stats = blockstore.getStats(); expect(stats.missingBlocks.has(nonExistentCID.toString())).toBe(true); }); it('should check if block exists', async () => { expect(await blockstore.has(testCID)).toBe(false); await blockstore.put(testCID, testBlock); expect(await blockstore.has(testCID)).toBe(true); }); it('should throw error when trying to delete blocks', async () => { await blockstore.put(testCID, testBlock); expect(await blockstore.has(testCID)).toBe(true); await expect(blockstore.delete(testCID)).rejects.toThrow('Delete operation not supported on CAR writing blockstore'); // Block should still exist after failed delete expect(await blockstore.has(testCID)).toBe(true); }); }); describe('Batch Operations', () => { it('should put many blocks', async () => { const blocks = []; for (let i = 0; i < 3; i++) { const data = new TextEncoder().encode(`Block ${i}`); const hash = await sha256.digest(data); const cid = CID.create(1, raw.code, hash); blocks.push({ cid, block: data }); } const results = []; for await (const cid of blockstore.putMany(blocks)) { results.push(cid); } expect(results).toHaveLength(3); const stats = blockstore.getStats(); expect(stats.blocksWritten).toBe(3); }); it('should throw error when trying to delete many blocks', async () => { // Put some blocks first const cids = []; for (let i = 0; i < 3; i++) { const data = new TextEncoder().encode(`Block ${i}`); const hash = await sha256.digest(data); const cid = CID.create(1, raw.code, hash); cids.push(cid); await blockstore.put(cid, data); } // deleteMany should throw an error await expect(async () => { // eslint-disable-next-line @typescript-eslint/no-unused-vars for await (const _ of blockstore.deleteMany(cids)) { // Should not reach here } }).rejects.toThrow('DeleteMany operation not supported on CAR writing blockstore'); }); it('should get many blocks', async () => { // Put some blocks first const blocks = []; const cids = []; for (let i = 0; i < 3; i++) { const data = new TextEncoder().encode(`Block ${i}`); const hash = await sha256.digest(data); const cid = CID.create(1, raw.code, hash); blocks.push({ cid, block: data }); cids.push(cid); await blockstore.put(cid, data); } const results = []; for await (const pair of blockstore.getMany(cids)) { results.push(pair); } expect(results).toHaveLength(3); expect(results[0]?.block).toEqual(blocks[0]?.block); }); it('should get all blocks', async () => { // Put some blocks first for (let i = 0; i < 3; i++) { const data = new TextEncoder().encode(`Block ${i}`); const hash = await sha256.digest(data); const cid = CID.create(1, raw.code, hash); await blockstore.put(cid, data); } const results = []; for await (const pair of blockstore.getAll()) { results.push(pair); } expect(results).toHaveLength(3); }); }); describe('CAR File Operations', () => { it('should create a CAR file', async () => { await blockstore.put(testCID, testBlock); await blockstore.finalize(); expect(existsSync(testOutputPath)).toBe(true); // Verify file is not empty const fileContent = await readFile(testOutputPath); expect(fileContent.length).toBeGreaterThan(0); }); it('should finalize and return statistics', async () => { await blockstore.put(testCID, testBlock); const finalStats = await blockstore.finalize(); expect(finalStats.finalized).toBe(true); expect(finalStats.blocksWritten).toBe(1); expect(finalStats.totalSize).toBe(testBlock.length); expect(finalStats.missingBlocks.size).toBe(0); }); it('should emit finalized event', async () => { let eventData; blockstore.on('finalized', (data) => { eventData = data; }); await blockstore.put(testCID, testBlock); await blockstore.finalize(); expect(eventData).toBeDefined(); expect(eventData.finalized).toBe(true); }); it('should prevent putting blocks after finalization', async () => { // Write at least one block before finalizing await blockstore.put(testCID, testBlock); await blockstore.finalize(); // Now try to put another block - this should fail const anotherBlock = new TextEncoder().encode('Another block'); const anotherHash = await sha256.digest(anotherBlock); const anotherCID = CID.create(1, raw.code, anotherHash); await expect(blockstore.put(anotherCID, anotherBlock)) .rejects.toThrow('Cannot put blocks in finalized CAR blockstore'); }); it('should handle multiple finalize calls gracefully', async () => { await blockstore.put(testCID, testBlock); const stats1 = await blockstore.finalize(); const stats2 = await blockstore.finalize(); expect(stats1).toEqual(stats2); }); }); describe('Error Handling', () => { it('should handle cleanup gracefully', async () => { await blockstore.put(testCID, testBlock); let cleanupEmitted = false; blockstore.on('cleanup', () => { cleanupEmitted = true; }); await blockstore.cleanup(); expect(cleanupEmitted).toBe(true); }); it('should throw error when finalizing without any blocks', async () => { // Create a fresh blockstore that hasn't written any blocks const emptyBlockstore = new CARWritingBlockstore({ rootCID: testCID, outputPath: './test-empty.car' }); await expect(emptyBlockstore.finalize()) .rejects.toThrow('Cannot finalize CAR blockstore without any blocks written'); }); }); }); //# sourceMappingURL=car-blockstore.test.js.map