other ci issue resolved

Author: shrirajpawar4

src/core/data-set/get-data-set-pieces.ts

@@ -7,7 +7,7 @@
  */
 
 import { getActivePieces, getScheduledRemovals } from '@filoz/synapse-core/pdp-verifier'
-import { getSizeFromPieceCID } from '@filoz/synapse-core/piece'
+import * as pieceTools from '@filoz/synapse-core/piece'
 import { getDataSet as getProviderDataSet } from '@filoz/synapse-core/sp'
 import { getAllPieceMetadata } from '@filoz/synapse-core/warm-storage'
 import { type DataSetPieceData, METADATA_KEYS, type Synapse } from '@filoz/synapse-sdk'
@@ -17,6 +17,22 @@ import { type DataSetPiecesResult, type GetDataSetPiecesOptions, type PieceInfo,
 
 const ACTIVE_PIECES_BATCH_SIZE = 100n
 
+type PieceSizeTools = {
+  getSizeFromPieceCID?: (pieceCid: unknown) => number
+  from?: (pieceCid: unknown) => { size: number }
+}
+
+function getPieceSize(pieceCid: unknown): number {
+  const tools = pieceTools as PieceSizeTools
+  if (tools.getSizeFromPieceCID != null) {
+    return tools.getSizeFromPieceCID(pieceCid)
+  }
+  if (tools.from != null) {
+    return tools.from(pieceCid).size
+  }
+  throw new Error('No piece size helper is available from @filoz/synapse-core/piece')
+}
+
 /**
  * Get all pieces for a dataset.
  *
@@ -100,7 +116,7 @@ export async function getDataSetPieces(
         }
 
         try {
-          pieceInfo.size = getSizeFromPieceCID(piece.cid)
+          pieceInfo.size = getPieceSize(piece.cid)
         } catch (error) {
           logger?.warn(
             { pieceId: piece.id.toString(), pieceCid: piece.cid.toString(), error },
@@ -124,7 +140,7 @@ export async function getDataSetPieces(
           status: PieceStatus.OFFCHAIN_ORPHANED,
         }
         try {
-          pieceInfo.size = getSizeFromPieceCID(providerPiece.pieceCid)
+          pieceInfo.size = getPieceSize(providerPiece.pieceCid)
         } catch {
           // size calculation is best-effort
         }

src/test/unit/core-data-set.test.ts

@@ -20,6 +20,7 @@ const {
   mockGetScheduledRemovals,
   mockGetProviderDataSet,
   mockGetSizeFromPieceCID,
+  mockPieceFromCID,
   state,
 } = vi.hoisted(() => {
   const state = {
@@ -67,6 +68,7 @@ const {
     if (cidString === 'bafkpiece2') return 4194304
     throw new Error(`Invalid piece CID: ${cidString}`)
   })
+  const mockPieceFromCID = vi.fn((cid: { toString: () => string } | string) => ({ size: mockGetSizeFromPieceCID(cid) }))
 
   const mockSynapse = {
     client: { account: { address: '0xtest-address' as const } },
@@ -84,6 +86,7 @@ const {
     mockGetScheduledRemovals,
     mockGetProviderDataSet,
     mockGetSizeFromPieceCID,
+    mockPieceFromCID,
     state,
   }
 })
@@ -111,6 +114,7 @@ vi.mock('@filoz/synapse-core/sp', () => ({
 // Mock piece size calculation
 vi.mock('@filoz/synapse-core/piece', () => ({
   getSizeFromPieceCID: mockGetSizeFromPieceCID,
+  from: mockPieceFromCID,
   MAX_UPLOAD_SIZE: 32 * 1024 * 1024 * 1024, // 32 GiB
 }))
 
@@ -215,6 +219,9 @@ describe('getDataSetPieces', () => {
       if (cidString === 'bafkpiece2') return 4194304
       throw new Error(`Invalid piece CID: ${cidString}`)
     })
+    mockPieceFromCID.mockImplementation((cid: { toString: () => string } | string) => ({
+      size: mockGetSizeFromPieceCID(cid),
+    }))
   })
 
   it('returns empty array when dataset has no pieces', async () => {

src/test/unit/data-set.test.ts

@@ -199,6 +199,10 @@ vi.mock('@filoz/synapse-core/sp', () => ({
 // Mock piece size calculation
 vi.mock('@filoz/synapse-core/piece', () => ({
   MAX_UPLOAD_SIZE: 1048576,
+  getSizeFromPieceCID: vi.fn(() => {
+    // Return a realistic piece size (1 MiB = 1048576 bytes)
+    return 1048576
+  }),
   from: vi.fn(() => ({
     // Return a realistic piece size (1 MiB = 1048576 bytes)
     size: 1048576,