|
14 | 14 |
|
15 | 15 | import com.hedera.hashgraph.sdk.FileId; |
16 | 16 | import java.time.Instant; |
| 17 | +import java.util.List; |
| 18 | + |
17 | 19 | import org.hiero.base.HieroException; |
18 | 20 | import org.hiero.base.implementation.FileClientImpl; |
19 | 21 | import org.hiero.base.protocol.ProtocolLayerClient; |
|
30 | 32 | import org.junit.jupiter.api.Assertions; |
31 | 33 | import org.junit.jupiter.api.BeforeEach; |
32 | 34 | import org.junit.jupiter.api.Test; |
| 35 | +import org.mockito.ArgumentCaptor; |
33 | 36 | import org.mockito.Mockito; |
34 | 37 |
|
35 | 38 | public class FileClientImplTest { |
36 | 39 | ProtocolLayerClient protocolLayerClient; |
37 | 40 | FileClientImpl fileClientImpl; |
38 | 41 |
|
| 42 | + |
| 43 | + ArgumentCaptor<FileAppendRequest> fileAppendCaptor = ArgumentCaptor.forClass(FileAppendRequest.class); |
| 44 | + |
39 | 45 | @BeforeEach |
40 | 46 | void setup() { |
41 | 47 | protocolLayerClient = Mockito.mock(ProtocolLayerClient.class); |
@@ -251,8 +257,12 @@ void testAppendFile() throws HieroException { |
251 | 257 | fileClientImpl.appendFile(fileId, content); |
252 | 258 |
|
253 | 259 | verify(protocolLayerClient, times(1)) |
254 | | - .executeFileAppendRequestTransaction(any(FileAppendRequest.class)); |
| 260 | + .executeFileAppendRequestTransaction(fileAppendCaptor.capture()); |
255 | 261 | verify(protocolLayerClient, times(1)).executeFileInfoQuery(any(FileInfoRequest.class)); |
| 262 | + |
| 263 | + final FileAppendRequest captureRequest = fileAppendCaptor.getValue(); |
| 264 | + Assertions.assertEquals(fileId, captureRequest.fileId()); |
| 265 | + Assertions.assertArrayEquals(content, captureRequest.contents()); |
256 | 266 | } |
257 | 267 |
|
258 | 268 | @Test |
@@ -315,8 +325,43 @@ void testAppendFileWithLargeContent() throws HieroException { |
315 | 325 |
|
316 | 326 | // call for each 2048 chunk |
317 | 327 | verify(protocolLayerClient, times(2)) |
318 | | - .executeFileAppendRequestTransaction(any(FileAppendRequest.class)); |
| 328 | + .executeFileAppendRequestTransaction(fileAppendCaptor.capture()); |
319 | 329 | verify(protocolLayerClient, times(1)).executeFileInfoQuery(any(FileInfoRequest.class)); |
| 330 | + |
| 331 | + final List<FileAppendRequest> capturedRequests = fileAppendCaptor.getAllValues(); |
| 332 | + Assertions.assertEquals(fileId, capturedRequests.get(0).fileId()); |
| 333 | + Assertions.assertEquals(FileCreateRequest.FILE_CREATE_MAX_SIZE, capturedRequests.get(0).contents().length); |
| 334 | + |
| 335 | + Assertions.assertEquals(fileId, capturedRequests.get(1).fileId()); |
| 336 | + Assertions.assertEquals(FileCreateRequest.FILE_CREATE_MAX_SIZE, capturedRequests.get(1).contents().length); |
| 337 | + } |
| 338 | + |
| 339 | + @Test |
| 340 | + void testAppendFileWithLargeContentUnevenSize() throws HieroException { |
| 341 | + // mocks |
| 342 | + final FileInfoResponse fileInfoResponse = Mockito.mock(FileInfoResponse.class); |
| 343 | + |
| 344 | + // given |
| 345 | + final FileId fileId = FileId.fromString("1.2.3"); |
| 346 | + final byte[] content = new byte[FileCreateRequest.FILE_CREATE_MAX_SIZE + 2]; |
| 347 | + |
| 348 | + when(protocolLayerClient.executeFileInfoQuery(any(FileInfoRequest.class))) |
| 349 | + .thenReturn(fileInfoResponse); |
| 350 | + when(fileInfoResponse.size()).thenReturn(1); |
| 351 | + |
| 352 | + fileClientImpl.appendFile(fileId, content); |
| 353 | + |
| 354 | + // call for each chunk 2048 and 2 |
| 355 | + verify(protocolLayerClient, times(2)) |
| 356 | + .executeFileAppendRequestTransaction(fileAppendCaptor.capture()); |
| 357 | + verify(protocolLayerClient, times(1)).executeFileInfoQuery(any(FileInfoRequest.class)); |
| 358 | + |
| 359 | + final List<FileAppendRequest> capturedRequests = fileAppendCaptor.getAllValues(); |
| 360 | + Assertions.assertEquals(fileId, capturedRequests.get(0).fileId()); |
| 361 | + Assertions.assertEquals(FileCreateRequest.FILE_CREATE_MAX_SIZE, capturedRequests.get(0).contents().length); |
| 362 | + |
| 363 | + Assertions.assertEquals(fileId, capturedRequests.get(1).fileId()); |
| 364 | + Assertions.assertEquals(2, capturedRequests.get(1).contents().length); |
320 | 365 | } |
321 | 366 |
|
322 | 367 | @Test |
|
0 commit comments