Added tests for AccountClientImpl.getAccountBalance functionality

Signed-off-by: Lemeri123 <[email protected]>

Author: Lemeri123

hiero-enterprise-base/pom.xml

@@ -42,6 +42,12 @@
       <artifactId>junit-jupiter</artifactId>
       <scope>test</scope>
     </dependency>
+    <dependency>
+  <groupId>org.mockito</groupId>
+  <artifactId>mockito-core</artifactId>
+  <version> 5.11.0</version>
+  <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,6 +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;
 import com.openelements.hiero.base.AccountClient;
@@ -44,7 +42,7 @@ public void deleteAccount(@NonNull Account account, @NonNull Account toAccount)
 
     @NonNull
     @Override
-    public Hbar getAccountBalance(@NonNull AccountId account) throws HieroException {
+    public Hbar getAccountBalance(com.hedera.hashgraph.sdk.AccountId account) throws HieroException {
         final AccountBalanceRequest request = AccountBalanceRequest.of(account);
         final AccountBalanceResponse response = client.executeAccountBalanceQuery(request);
         return response.hbars();

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

@@ -0,0 +1,98 @@
+package com.openelements.hiero.base.test;
+import com.openelements.hiero.base.implementation.AccountClientImpl;
+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 {
+        // Fully qualify AccountId to ensure no ambiguity
+        com.hedera.hashgraph.sdk.AccountId accountId = com.hedera.hashgraph.sdk.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);
+
+        // Explicit cast ensures we're calling the correct method
+        Hbar balance = accountClientImpl.getAccountBalance(accountId);
+
+        assertEquals(expectedBalance, balance);
+    }
+
+    @Test
+    public void testGetAccountBalance_ZeroBalance() throws HieroException {
+        com.hedera.hashgraph.sdk.AccountId accountId = com.hedera.hashgraph.sdk.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 {
+        // Use a valid but non-existing account ID to simulate invalid behavior
+        com.hedera.hashgraph.sdk.AccountId invalidAccountId = com.hedera.hashgraph.sdk.AccountId.fromString("0.0.9999999"); // Simulate invalid or non-existing account
+    
+        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((com.hedera.hashgraph.sdk.AccountId) null);
+        });
+    }
+    
+
+    @Test
+    public void testGetAccountBalance_ProtocolLayerClientFails() throws HieroException {
+        com.hedera.hashgraph.sdk.AccountId accountId = com.hedera.hashgraph.sdk.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,10 @@
         <artifactId>junit-jupiter</artifactId>
         <version>${junit.version}</version>
       </dependency>
+       <dependency>
+  <groupId>org.mockito</groupId>
+  <artifactId>mockito-core</artifactId>
+</dependency>
       <dependency>
         <groupId>org.slf4j</groupId>
         <artifactId>slf4j-api</artifactId>