Merge pull request #123 from Lemeri123/main

Added tests for AccountClientImpl.getAccountBalance functionality

Author: hendrikebbers

hiero-enterprise-base/pom.xml

@@ -42,6 +42,11 @@
       <artifactId>junit-jupiter</artifactId>
       <scope>test</scope>
     </dependency>
+    <dependency>
+  <groupId>org.mockito</groupId>
+  <artifactId>mockito-core</artifactId>
+  <scope>test</scope>
+</dependency>
     <dependency>
       <groupId>io.grpc</groupId>
       <artifactId>grpc-okhttp</artifactId>

hiero-enterprise-base/src/main/java/com/openelements/hiero/base/implementation/AccountClientImpl.java

@@ -1,5 +1,4 @@
 package com.openelements.hiero.base.implementation;
-
 import com.hedera.hashgraph.sdk.AccountId;
 import com.hedera.hashgraph.sdk.Hbar;
 import com.openelements.hiero.base.data.Account;

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

@@ -0,0 +1,95 @@
+package com.openelements.hiero.base.test;
+
+import com.openelements.hiero.base.implementation.AccountClientImpl;
+import com.hedera.hashgraph.sdk.AccountId;
+import com.hedera.hashgraph.sdk.Hbar;
+import com.openelements.hiero.base.HieroException;
+import com.openelements.hiero.base.protocol.AccountBalanceRequest;
+import com.openelements.hiero.base.protocol.AccountBalanceResponse;
+import com.openelements.hiero.base.protocol.ProtocolLayerClient;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+import org.mockito.ArgumentMatchers;
+
+import static org.junit.jupiter.api.Assertions.*;
+import static org.mockito.Mockito.*;
+
+public class AccountClientImplTest {
+
+    private ProtocolLayerClient mockProtocolLayerClient;
+    private AccountClientImpl accountClientImpl;
+
+    @BeforeEach
+    public void setUp() {
+        mockProtocolLayerClient = mock(ProtocolLayerClient.class);
+        accountClientImpl = new AccountClientImpl(mockProtocolLayerClient);
+    }
+
+    @Test
+    public void testGetAccountBalance_ValidPositiveBalance() throws HieroException {
+        AccountId accountId = AccountId.fromString("0.0.12345");
+        Hbar expectedBalance = new Hbar(10);
+
+        // Mock the response
+        AccountBalanceResponse mockResponse = mock(AccountBalanceResponse.class);
+        when(mockResponse.hbars()).thenReturn(expectedBalance);
+
+        when(mockProtocolLayerClient.executeAccountBalanceQuery(
+                ArgumentMatchers.any(AccountBalanceRequest.class)
+        )).thenReturn(mockResponse);
+
+        Hbar balance = accountClientImpl.getAccountBalance(accountId);
+
+        assertEquals(expectedBalance, balance);
+    }
+
+    @Test
+    public void testGetAccountBalance_ZeroBalance() throws HieroException {
+        AccountId accountId = AccountId.fromString("0.0.67890");
+        Hbar expectedBalance = new Hbar(0);
+
+        AccountBalanceResponse mockResponse = mock(AccountBalanceResponse.class);
+        when(mockResponse.hbars()).thenReturn(expectedBalance);
+
+        when(mockProtocolLayerClient.executeAccountBalanceQuery(
+                ArgumentMatchers.any(AccountBalanceRequest.class)
+        )).thenReturn(mockResponse);
+
+        Hbar balance = accountClientImpl.getAccountBalance(accountId);
+
+        assertEquals(expectedBalance, balance);
+    }
+
+    @Test
+    public void testGetAccountBalance_InvalidAccount_ThrowsException() throws HieroException {
+        AccountId invalidAccountId = AccountId.fromString("0.0.9999999");
+
+        when(mockProtocolLayerClient.executeAccountBalanceQuery(
+                ArgumentMatchers.any(AccountBalanceRequest.class)
+        )).thenThrow(new HieroException("Invalid account"));
+
+        assertThrows(HieroException.class, () -> {
+            accountClientImpl.getAccountBalance(invalidAccountId);
+        });
+    }
+
+    @Test
+    public void testGetAccountBalance_NullThrowsException() {
+        assertThrows(NullPointerException.class, () -> {
+            accountClientImpl.getAccountBalance((AccountId) null);
+        });
+    }
+
+    @Test
+    public void testGetAccountBalance_ProtocolLayerClientFails() throws HieroException {
+        AccountId accountId = AccountId.fromString("0.0.12345");
+
+        when(mockProtocolLayerClient.executeAccountBalanceQuery(
+                ArgumentMatchers.any(AccountBalanceRequest.class)
+        )).thenThrow(new RuntimeException("Protocol Layer Failure"));
+
+        assertThrows(RuntimeException.class, () -> {
+            accountClientImpl.getAccountBalance(accountId);
+        });
+    }
+}

pom.xml

@@ -159,6 +159,11 @@
         <artifactId>junit-jupiter</artifactId>
         <version>${junit.version}</version>
       </dependency>
+    <dependency>
+  <groupId>org.mockito</groupId>
+  <artifactId>mockito-core</artifactId>
+  <version> 5.11.0</version>
+</dependency>
       <dependency>
         <groupId>org.slf4j</groupId>
         <artifactId>slf4j-api</artifactId>