ckb-sdk-java/ckb-mercury-sdk/src/test/java/mercury/BalanceTest.java at master · nervosnetwork/ckb-sdk-java · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
package mercury;

import constant.AddressWithKeyHolder;
import constant.ApiFactory;
import constant.UdtHolder;
import org.junit.jupiter.api.Assertions;
import org.junit.jupiter.api.Test;
import org.nervos.ckb.utils.Numeric;
import org.nervos.mercury.model.GetBalancePayloadBuilder;
import org.nervos.mercury.model.common.AssetInfo;
import org.nervos.mercury.model.req.item.ItemFactory;
import org.nervos.mercury.model.resp.GetBalanceResponse;

import java.io.IOException;

import static org.junit.jupiter.api.Assertions.assertNotNull;

public class BalanceTest {
  @Test
  void getCkbBalance() throws IOException {
    GetBalancePayloadBuilder builder = new GetBalancePayloadBuilder();
    builder.item(ItemFactory.newIdentityItemByCkb(AddressWithKeyHolder.testPubKey4()));
    builder.addAssetInfo(AssetInfo.newCkbAsset());

    GetBalanceResponse balance = ApiFactory.getApi().getBalance(builder.build());
    Assertions.assertEquals(2, balance.balances.size());
  }

  @Test
  void getSudtBalance() throws IOException {
    GetBalancePayloadBuilder builder = new GetBalancePayloadBuilder();
    builder.item(ItemFactory.newIdentityItemByCkb(AddressWithKeyHolder.testPubKey4()));
    builder.addAssetInfo(AssetInfo.newUdtAsset(UdtHolder.UDT_HASH));

    GetBalanceResponse balance = ApiFactory.getApi().getBalance(builder.build());
    Assertions.assertEquals(2, balance.balances.size());
    assertNotNull(balance, "Balance is not empty");
  }

  @Test
  void getAllBalance() throws IOException {
    GetBalancePayloadBuilder builder = new GetBalancePayloadBuilder();
    builder.item(ItemFactory.newIdentityItemByCkb(AddressWithKeyHolder.testPubKey4()));

    GetBalanceResponse balance = ApiFactory.getApi().getBalance(builder.build());
    assertNotNull(balance, "Balance is not empty");
    Assertions.assertNotEquals(0, balance.balances.size());
  }

  @Test
  void getBalanceByAddress() throws IOException {
    GetBalancePayloadBuilder builder = new GetBalancePayloadBuilder();
    builder.item(ItemFactory.newAddressItem(AddressWithKeyHolder.testAddress4()));
    builder.addAssetInfo(AssetInfo.newCkbAsset());

    GetBalanceResponse balance = ApiFactory.getApi().getBalance(builder.build());
    Assertions.assertNotEquals(0, balance.balances.size());
  }

  @Test
  void getBalanceByIdentity() throws IOException {
    GetBalancePayloadBuilder builder = new GetBalancePayloadBuilder();
    builder.item(ItemFactory.newIdentityItemByCkb(AddressWithKeyHolder.testPubKey4()));
    builder.addAssetInfo(AssetInfo.newUdtAsset(UdtHolder.UDT_HASH));

    GetBalanceResponse balance = ApiFactory.getApi().getBalance(builder.build());
    Assertions.assertNotEquals(0, balance.balances.size());
  }

  @Test
  void getBalanceByRecordByScriptByChequeCellSender() throws IOException {
    GetBalancePayloadBuilder builder = new GetBalancePayloadBuilder();
    builder.item(
        ItemFactory.newOutPointItem(
            Numeric.hexStringToByteArray(
                "0x52b1cf0ad857d53e1a3552944c1acf268f6a6aea8e8fc85fe8febcb8127d56f0"),
            0));
    builder.addAssetInfo(AssetInfo.newCkbAsset());

    GetBalanceResponse balance = ApiFactory.getApi().getBalance(builder.build());
    Assertions.assertNotEquals(0, balance.balances.size());
  }

  @Test
  void getBalanceByRecordByScriptChequeCellReceiver() throws IOException {
    GetBalancePayloadBuilder builder = new GetBalancePayloadBuilder();
    builder.item(
        ItemFactory.newOutPointItem(
            Numeric.hexStringToByteArray(
                "0x52b1cf0ad857d53e1a3552944c1acf268f6a6aea8e8fc85fe8febcb8127d56f0"),
            0));
    builder.addAssetInfo(AssetInfo.newUdtAsset(UdtHolder.UDT_HASH));

    GetBalanceResponse balance = ApiFactory.getApi().getBalance(builder.build());
    Assertions.assertNotEquals(0, balance.balances.size());
  }
}