Added test for TokenClient

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

Author: manishdait

hiero-enterprise-spring/src/main/java/com/openelements/hiero/spring/implementation/HieroAutoConfiguration.java

@@ -4,6 +4,7 @@
 import com.openelements.hiero.base.FileClient;
 import com.openelements.hiero.base.HieroContext;
 import com.openelements.hiero.base.NftClient;
+import com.openelements.hiero.base.TokenClient;
 import com.openelements.hiero.base.SmartContractClient;
 import com.openelements.hiero.base.config.HieroConfig;
 import com.openelements.hiero.base.implementation.AccountClientImpl;
@@ -12,6 +13,7 @@
 import com.openelements.hiero.base.implementation.HieroNetwork;
 import com.openelements.hiero.base.implementation.NetworkRepositoryImpl;
 import com.openelements.hiero.base.implementation.NftClientImpl;
+import com.openelements.hiero.base.implementation.TokenClientImpl;
 import com.openelements.hiero.base.implementation.NftRepositoryImpl;
 import com.openelements.hiero.base.implementation.ProtocolLayerClientImpl;
 import com.openelements.hiero.base.implementation.SmartContractClientImpl;
@@ -82,6 +84,11 @@ NftClient nftClient(final ProtocolLayerClient protocolLayerClient, HieroContext
         return new NftClientImpl(protocolLayerClient, hieroContext.getOperatorAccount());
     }
 
+    @Bean
+    TokenClient tokenClient(final ProtocolLayerClient protocolLayerClient, HieroContext hieroContext) {
+        return new TokenClientImpl(protocolLayerClient, hieroContext.getOperatorAccount());
+    }
+
     @Bean
     @ConditionalOnProperty(prefix = "spring.hiero", name = "mirrorNodeSupported",
             havingValue = "true", matchIfMissing = true)

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

@@ -0,0 +1,81 @@
+package com.openelements.hiero.spring.test;
+
+import com.hedera.hashgraph.sdk.TokenId;
+import com.openelements.hiero.base.data.Account;
+import com.openelements.hiero.base.AccountClient;
+import com.openelements.hiero.base.HieroException;
+import com.openelements.hiero.base.TokenClient;
+import org.junit.jupiter.api.Assertions;
+import org.junit.jupiter.api.Test;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.test.context.SpringBootTest;
+
+@SpringBootTest(classes = TestConfig.class)
+public class TokenClientTest {
+    @Autowired
+    private TokenClient tokenClient;
+
+    @Autowired
+    private AccountClient accountClient;
+
+    @Test
+    void createToken() throws HieroException {
+        final String name = "TOKEN";
+        final String symbol = "FT";
+
+        final TokenId tokenId = tokenClient.createToken(name, symbol);
+
+        Assertions.assertNotNull(tokenId);
+    }
+
+    @Test
+    void associateToken() throws HieroException {
+        final String name = "TOKEN";
+        final String symbol = "FT";
+        final TokenId tokenId = tokenClient.createToken(name, symbol);
+
+        final Account account = accountClient.createAccount(1);
+
+        Assertions.assertDoesNotThrow(() -> tokenClient.associateToken(tokenId, account));
+    }
+
+    @Test
+    void mintToken() throws HieroException {
+        final String name = "TOKEN";
+        final String symbol = "FT";
+        final Long amount = 1L;
+
+        final TokenId tokenId = tokenClient.createToken(name, symbol);
+        final long totalSupply = tokenClient.mintToken(tokenId, amount);
+
+        Assertions.assertEquals(amount, totalSupply);
+    }
+
+    @Test
+    void burnToken() throws HieroException {
+        final String name = "TOKEN";
+        final String symbol = "FT";
+        final long amount = 1L;
+
+        final TokenId tokenId = tokenClient.createToken(name, symbol);
+        tokenClient.mintToken(tokenId, amount);
+
+        final long supplyTotal = tokenClient.burnToken(tokenId, 1L);
+        Assertions.assertEquals(0, supplyTotal);
+    }
+
+    @Test
+    void transferToken() throws HieroException {
+        final Account toAccount = accountClient.createAccount(1);
+        final String name = "TOKEN";
+        final String symbol = "FT";
+
+        final TokenId tokenId = tokenClient.createToken(name, symbol);
+        tokenClient.associateToken(tokenId, toAccount);
+
+        long totalSupply = tokenClient.mintToken(tokenId, 1L);
+
+        Assertions.assertDoesNotThrow(() -> tokenClient.transferToken(tokenId,  toAccount.accountId(), totalSupply));
+    }
+
+}