fix getting balance add integration tests

Author: AlaaElattar

packages/tfchain_client/bin/tfchain_client.dart

@@ -17,9 +17,11 @@ void main() async {
       "secret add bag cluster deposit beach illness letter crouch position rain arctic",
       "sr25519");
   await client.connect();
+  final extrinsic = await client.clientBalances.transfer(
+      address: "5CJrCjZvsudNoJApTGG5PKcZfhAzAyGqgSK8bysoCV2oRBMC", amount: 10);
 
-  // await client.apply(extrinsic);
-  client.kvStrore.list();
+  await client.apply(extrinsic);
+  // await client.kvStrore.list();
 
   // await client.disconnect();
 }

packages/tfchain_client/lib/src/balances.dart

@@ -1,4 +1,5 @@
 import 'package:polkadart/scale_codec.dart';
+import 'package:polkadart_keyring/polkadart_keyring.dart';
 import 'package:tfchain_client/generated/dev/types/frame_system/account_info.dart';
 import 'package:tfchain_client/generated/dev/types/sp_runtime/multiaddress/multi_address.dart';
 import 'package:tfchain_client/generated/dev/types/tfchain_runtime/runtime_call.dart';
@@ -9,9 +10,9 @@ class QueryBalances {
   QueryBalances(this.client);
 
   Future<AccountInfo?> get({required String address}) async {
-    // TODO: should get pair.publicKey.bytes, How to get keypair if i don't have mnemonic
-    final res = await client.api.query.system
-        .account(Address32(address.codeUnits).value0);
+    final keyring = Keyring();
+    final publicKey = keyring.decodeAddress(address);
+    final res = await client.api.query.system.account(publicKey);
     return res;
   }
 }

packages/tfchain_client/lib/src/kvstore.dart

@@ -43,7 +43,25 @@ class KVStore {
       ...client.keypair!.publicKey.bytes
     ]);
 
-    final keys = await client.api.rpc.state.getKeys(key: partialKey);
+    List<Uint8List> keys = [];
+    int count = 2;
+    Uint8List? startKey;
+    bool hasMore = true;
+
+    while (hasMore) {
+      final keysPage = await client.api.rpc.state.getKeysPaged(
+        key: partialKey,
+        count: count,
+        startKey: startKey,
+      );
+
+      if (keysPage.isEmpty) {
+        hasMore = false;
+      } else {
+        keys.addAll(keysPage);
+        startKey = keysPage.last;
+      }
+    }
 
     final keysValues = await client.api.rpc.state.queryStorageAt(keys);
 
@@ -59,6 +77,7 @@ class KVStore {
       keyValueMap[String.fromCharCodes(added)] = String.fromCharCodes(value);
     }
 
+    print(keyValueMap);
     return keyValueMap;
   }
 

packages/tfchain_client/lib/src/tft_price.dart

@@ -4,7 +4,7 @@ class QueryTFTPrice {
   final QueryClient client;
   QueryTFTPrice(this.client);
 
-  Future<int?> get({required String address}) async {
+  Future<int?> get() async {
     final res = await client.api.query.tFTPriceModule.tftPrice();
     return res;
   }

packages/tfchain_client/lib/src/twins.dart

@@ -1,3 +1,4 @@
+import 'package:polkadart_keyring/polkadart_keyring.dart';
 import 'package:tfchain_client/generated/dev/types/pallet_tfgrid/types/twin.dart';
 import 'package:tfchain_client/generated/dev/types/tfchain_runtime/runtime_call.dart';
 import 'package:tfchain_client/tfchain_client.dart';
@@ -11,7 +12,9 @@ class QueryTwins {
     return res;
   }
 
-  Future<int?> getTwinIdByAccountId({required List<int> accountId}) async {
+  Future<int?> getTwinIdByAccountId({required String address}) async {
+    final keyring = Keyring();
+    final accountId = keyring.decodeAddress(address);
     final res =
         await client.api.query.tfgridModule.twinIdByAccountID(accountId);
     return res;

packages/tfchain_client/test/balances_test.dart

@@ -1,14 +1,30 @@
 import 'package:test/test.dart';
+import 'package:tfchain_client/generated/dev/types/frame_system/account_info.dart';
+import 'package:tfchain_client/tfchain_client.dart';
 
 void main() {
   group("Query Balances Test", () {
-    // late QueryClient queryClient;
-    // setUp(() {
-    //   queryClient = QueryClient("wss://tfchain.dev.grid.tf/ws");
-    // });
+    late QueryClient queryClient;
+    setUp(() {
+      queryClient = QueryClient("wss://tfchain.dev.grid.tf/ws");
+    });
 
-    // test('Test Get Balance', () {
-      
-    // });
+    test('Test Get Balance', () async {
+      String address = "5CJrCjZvsudNoJApTGG5PKcZfhAzAyGqgSK8bysoCV2oRBMC";
+      AccountInfo? accountInfo =
+          await queryClient.balances.get(address: address);
+      expect(accountInfo, isNotNull);
+    });
+
+    test('Test Get Balance with Invalid address', () async {
+      String address = "address";
+      try {
+        AccountInfo? accountInfo =
+            await queryClient.balances.get(address: address);
+        expect(accountInfo, isNull);
+      } catch (e) {
+        expect(e, isNotNull);
+      }
+    });
   });
 }

packages/tfchain_client/test/pricing_policies_test.dart

@@ -0,0 +1,17 @@
+import 'package:test/test.dart';
+import 'package:tfchain_client/generated/dev/types/pallet_tfgrid/types/pricing_policy.dart';
+import 'package:tfchain_client/tfchain_client.dart';
+
+void main() {
+  group("Query Pricing Policies", () {
+    late QueryClient queryClient;
+    setUp(() {
+      queryClient = QueryClient("wss://tfchain.dev.grid.tf/ws");
+    });
+
+    test('Test Get Pricing Policy', () async {
+      PricingPolicy? res = await queryClient.policies.get(id: 1);
+      expect(res, isNotNull);
+    });
+  });
+}

packages/tfchain_client/test/tft_bridge_test.dart

@@ -0,0 +1,21 @@
+import 'package:test/test.dart';
+import 'package:tfchain_client/tfchain_client.dart';
+
+void main() {
+  group("Query Bridge Test", () {
+    late QueryClient queryClient;
+    setUp(() {
+      queryClient = QueryClient("wss://tfchain.dev.grid.tf/ws");
+    });
+
+    test('Test Get Withdraw fee', () async {
+      BigInt? fee = await queryClient.bridge.getWithdrawFee();
+      expect(fee, isNotNull);
+    });
+
+    test('Test Get Deposit fee', () async {
+      BigInt? fee = await queryClient.bridge.getDepositFee();
+      expect(fee, isNotNull);
+    });
+  });
+}

packages/tfchain_client/test/tft_price_test.dart

@@ -0,0 +1,16 @@
+import 'package:test/test.dart';
+import 'package:tfchain_client/tfchain_client.dart';
+
+void main() {
+  group("Query Price Test", () {
+    late QueryClient queryClient;
+    setUp(() {
+      queryClient = QueryClient("wss://tfchain.dev.grid.tf/ws");
+    });
+
+    test('Test Get TFT price', () async {
+      final price = await queryClient.price.get();
+      expect(price, isNotNull);
+    });
+  });
+}

packages/tfchain_client/test/twins_test.dart

@@ -0,0 +1,30 @@
+import 'package:test/test.dart';
+import 'package:tfchain_client/tfchain_client.dart';
+
+void main() {
+  group("Query Twins Test", () {
+    late QueryClient queryClient;
+    setUp(() {
+      queryClient = QueryClient("wss://tfchain.dev.grid.tf/ws");
+    });
+
+    test('Test Get Twin with id', () async {
+      int id = 214;
+      final twin = await queryClient.twins.get(id: id);
+      expect(twin, isNotNull);
+    });
+
+    test('Test Get Twin with zero id', () async {
+      int id = 0;
+      final twin = await queryClient.twins.get(id: id);
+      expect(twin, null);
+    });
+
+    test('Test Get Twin Id with account Id', () async {
+      String address = "5CJrCjZvsudNoJApTGG5PKcZfhAzAyGqgSK8bysoCV2oRBMC";
+      final twin =
+          await queryClient.twins.getTwinIdByAccountId(address: address);
+      expect(twin, 7845);
+    });
+  });
+}