Implemented tests for missing functionalities in FileClientImpl.createFile()

Signed-off-by: Manish Dait <[email protected]>

Author: manishdait

hiero-enterprise-base/src/test/java/com/openelements/hiero/base/test/FileClientImplTest.java

@@ -3,6 +3,10 @@
 import com.hedera.hashgraph.sdk.FileId;
 import com.openelements.hiero.base.HieroException;
 import com.openelements.hiero.base.implementation.FileClientImpl;
+import com.openelements.hiero.base.protocol.FileCreateResult;
+import com.openelements.hiero.base.protocol.FileCreateRequest;
+import com.openelements.hiero.base.protocol.FileAppendRequest;
+import com.openelements.hiero.base.protocol.FileAppendResult;
 import com.openelements.hiero.base.protocol.FileInfoRequest;
 import com.openelements.hiero.base.protocol.FileInfoResponse;
 import com.openelements.hiero.base.protocol.ProtocolLayerClient;
@@ -11,6 +15,8 @@
 import org.junit.jupiter.api.Test;
 import org.mockito.Mockito;
 
+import java.time.Instant;
+
 import static org.mockito.ArgumentMatchers.any;
 import static org.mockito.Mockito.*;
 
@@ -24,6 +30,80 @@ void setup() {
         fileClientImpl = new FileClientImpl(protocolLayerClient);
     }
 
+    @Test
+    void testCreateFile() throws HieroException {
+        // mock
+        final FileId fileId = FileId.fromString("1.2.3");
+        final FileCreateResult fileCreateResult = Mockito.mock(FileCreateResult.class);
+
+        // given
+        final byte[] content = "Hello Hiero!".getBytes();
+
+        //then
+        when(protocolLayerClient.executeFileCreateTransaction(any(FileCreateRequest.class)))
+                .thenReturn(fileCreateResult);
+        when(fileCreateResult.fileId()).thenReturn(fileId);
+
+        final FileId result = fileClientImpl.createFile(content);
+
+        verify(protocolLayerClient, times(1))
+                .executeFileCreateTransaction(any(FileCreateRequest.class));
+        verify(fileCreateResult, times(1)).fileId();
+        Assertions.assertEquals(fileId, result);
+    }
+
+    @Test
+    void testCreateFileForSizeGreaterThanFileCreateMaxSize() throws HieroException {
+        // mock
+        final FileId fileId = FileId.fromString("1.2.3");
+        final FileCreateResult fileCreateResult = Mockito.mock(FileCreateResult.class);
+
+        // given
+        byte[] content = new byte[FileCreateRequest.FILE_CREATE_MAX_SIZE * 2];
+        // -1 because 1 for executeFileCreateTransaction()
+        int appendCount = Math.floorDiv(content.length, FileCreateRequest.FILE_CREATE_MAX_SIZE) - 1;
+
+        //then
+        when(protocolLayerClient.executeFileCreateTransaction(any(FileCreateRequest.class)))
+                .thenReturn(fileCreateResult);
+        when(fileCreateResult.fileId()).thenReturn(fileId);
+        when(protocolLayerClient.executeFileAppendRequestTransaction(any(FileAppendRequest.class)))
+                .thenReturn(any(FileAppendResult.class));
+
+        final FileId result = fileClientImpl.createFile(content);
+
+        verify(protocolLayerClient, times(1))
+                .executeFileCreateTransaction(any(FileCreateRequest.class));
+        verify(fileCreateResult, times(1)).fileId();
+        verify(protocolLayerClient, times(appendCount))
+                .executeFileAppendRequestTransaction(any(FileAppendRequest.class));
+        Assertions.assertEquals(fileId, result);
+    }
+
+    @Test
+    void testCreateFileThrowsExceptionForSizeGreaterThanMaxFileSize() {
+        // given
+        final byte[] contents = new byte[FileCreateRequest.FILE_MAX_SIZE + 1];
+
+        // then
+        Assertions.assertThrows(HieroException.class, () -> fileClientImpl.createFile(contents));
+    }
+
+    @Test
+    void testCreateFileThrowsExceptionForExpirationTimeBeforeNow() {
+        // given
+        final byte[] contents = "Hello Hiero!".getBytes();
+        final Instant expiration = Instant.now().minusSeconds(1);
+
+        // then
+        Assertions.assertThrows(IllegalArgumentException.class, () -> fileClientImpl.createFile(contents, expiration));
+    }
+
+    @Test
+    void testCreateFileThrowsExceptionForNullContent() {
+        Assertions.assertThrows(NullPointerException.class, () -> fileClientImpl.createFile(null));
+    }
+
     @Test
     void testGetFileSize() throws HieroException {
         // mocks

hiero-enterprise-spring/src/test/java/com/openelements/hiero/spring/test/FileClientTests.java

@@ -7,6 +7,8 @@
 import java.time.Instant;
 import java.time.temporal.ChronoUnit;
 import java.util.stream.IntStream;
+
+import com.openelements.hiero.base.protocol.FileCreateRequest;
 import org.junit.jupiter.api.Assertions;
 import org.junit.jupiter.api.Disabled;
 import org.junit.jupiter.api.Test;
@@ -45,7 +47,7 @@ void testCreateEmptyFile() throws Exception {
     @Test
     void testCreateSmallFile() throws Exception {
         //given
-        final byte[] contents = "Hello, Hedera!".getBytes();
+        final byte[] contents = "Hello, Hiero!".getBytes();
 
         //when
         final FileId fileId = fileClient.createFile(contents);
@@ -57,7 +59,7 @@ void testCreateSmallFile() throws Exception {
     @Test
     void testCreateLargeFile() throws Exception {
         //given
-        final byte[] contents = IntStream.range(0, 500).mapToObj(i -> "Hello, Hedera!")
+        final byte[] contents = IntStream.range(0, 500).mapToObj(i -> "Hello, Hiero!")
                 .reduce((a, b) -> a + b)
                 .get()
                 .getBytes();
@@ -69,6 +71,27 @@ void testCreateLargeFile() throws Exception {
         Assertions.assertNotNull(fileId);
     }
 
+    @Test
+    void testCreateFileThrowExceptionIfExceedMaxFileSize() {
+        // given
+        final byte[] contents = new byte[FileCreateRequest.FILE_MAX_SIZE + 1];
+
+        // then
+        Assertions.assertThrows(HieroException.class, () -> fileClient.createFile(contents));
+    }
+
+    @Test
+    void testCreateFileThrowExceptionIfExpirationTimeBeforeNow() {
+        // given
+        final byte[] contents = "Hello Hiero!".getBytes();
+        final Instant definedExpirationTime = Instant.now().minusSeconds(60);
+
+        // then
+        Assertions.assertThrows(
+                IllegalArgumentException.class, () -> fileClient.createFile(contents, definedExpirationTime)
+        );
+    }
+
     @Test
     void testReadFileByFileId() throws Exception {
         //given