a little modify

diff --git a/src/main/java/io/bytom/api/Balance.java b/src/main/java/io/bytom/api/Balance.java
index f1478f0..18f73ee 100644 (file)
--- a/src/main/java/io/bytom/api/Balance.java
+++ b/src/main/java/io/bytom/api/Balance.java
@@ -88,17 +88,22 @@ public class Balance {
          * @return
          * @throws BytomException
          */
-        public long listByAssetAlias(Client client, String assetAlias) throws BytomException {
+        public Balance listByAssetAlias(Client client, String assetAlias) throws BytomException {
             List<Balance> balanceList = list(client);
             Balance assetBalance = new Balance();
+            assetBalance.assetAlias = assetAlias;
             long amount = 0;
             for (Balance result : balanceList) {
                 if (result.assetAlias.equals(assetAlias)) {
                     amount += result.amount;
+                    assetBalance.assetId = result.assetId;
                 }
             }
-            logger.info("amount:"+amount);
-            return amount;
+            assetBalance.amount = amount;
+
+            logger.info(assetBalance.toJson());
+
+            return assetBalance;
         }
 
         /**

diff --git a/src/test/java/io/bytom/integration/BalanceTest.java b/src/test/java/io/bytom/integration/BalanceTest.java
index 4442a24..8015f4c 100644 (file)
--- a/src/test/java/io/bytom/integration/BalanceTest.java
+++ b/src/test/java/io/bytom/integration/BalanceTest.java
@@ -33,8 +33,8 @@ public class BalanceTest {
 
     @Test
     public void testBalanceByAssetAlias() throws Exception {
-        long amount = new Balance.QueryBuilder().listByAssetAlias(client, "BTM");
-        Assert.assertNotNull(amount);
+        Balance balance = new Balance.QueryBuilder().listByAssetAlias(client, "BTM");
+        Assert.assertNotNull(balance);
     }
 
     @Test