add contracts and balances tests

Author: AlaaElattar

packages/tfchain_client/lib/src/balances.dart

@@ -1,3 +1,5 @@
+import 'dart:math';
+
 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';
@@ -20,19 +22,17 @@ class Balances extends QueryBalances {
 
   final Client client;
 
-  Future<AccountInfo?> transfer(
-      {required String address, required int amount}) async {
+  Future<void> transfer({required String address, required int amount}) async {
     if (amount.isNaN || amount <= 0) {
       throw Exception("Amount must be a positive numeric value");
     }
     final keyring = Keyring();
     final publicKey = keyring.decodeAddress(address);
     MultiAddress multiAddress = Address32(publicKey);
 
-    final extrinsic = client.api.tx.balances
-        .transfer(dest: multiAddress, value: BigInt.from(amount));
-
-    return await this.get(address: client.address);
+    final extrinsic = client.api.tx.balances.transfer(
+        dest: multiAddress, value: BigInt.from(amount * pow(10, 7).toInt()));
+    await client.apply(extrinsic);
   }
 
   Future<AccountInfo?> getMyBalance() async {

packages/tfchain_client/lib/src/contracts.dart

@@ -1,4 +1,3 @@
-import 'package:tfchain_client/generated/dev/types/pallet_smart_contract/grid_contract/name_contract_name.dart';
 import 'package:tfchain_client/generated/dev/types/pallet_smart_contract/types/contract.dart';
 import 'package:tfchain_client/generated/dev/types/pallet_smart_contract/types/contract_lock.dart';
 import 'package:tfchain_client/generated/dev/types/pallet_smart_contract/types/service_contract.dart';
@@ -31,7 +30,7 @@ class QueryContracts {
 
   Future<BigInt?> getContractIdByName({required String name}) async {
     final res = await client.api.query.smartContractModule
-        .contractIDByNameRegistration(name as NameContractName);
+        .contractIDByNameRegistration(name.codeUnits);
     return res;
   }
 
@@ -85,6 +84,7 @@ class Contracts extends QueryContracts {
   Contracts(Client this.client) : super(client);
   final Client client;
 
+  // TODO: should i return bigint or convert it to int ??
   Future<BigInt?> createNode(
       {required int nodeId,
       required List<int> deploymentHash,
@@ -125,13 +125,13 @@ class Contracts extends QueryContracts {
   Future<BigInt?> createRent(
       {required int nodeId, required int solutionProviderId}) async {
     final extrinsic = client.api.tx.smartContractModule.createRentContract(
-        nodeId: nodeId, solutionProviderId: solutionProviderId);
+        nodeId: nodeId, solutionProviderId: BigInt.from(solutionProviderId));
     await client.apply(extrinsic);
 
     return await getContractIdByActiveRentForNode(nodeId: nodeId);
   }
 
-  Future<Contract?> cancel({required BigInt contractId}) async {
+  Future<void> cancel({required BigInt contractId}) async {
     final contract = await get(contractId: contractId);
     if (contract == null) {
       throw Exception("Contract not found");
@@ -140,14 +140,12 @@ class Contracts extends QueryContracts {
     final extrinsic = client.api.tx.smartContractModule
         .cancelContract(contractId: contractId);
     await client.apply(extrinsic);
-    // IF not found it means its cancelled successfully
-    return await get(contractId: contractId);
   }
 
   Future<BigInt?> setDedicatedNodeExtraFee(
       {required int nodeId, required int extraFee}) async {
     final extrinsic = client.api.tx.smartContractModule
-        .setDedicatedNodeExtraFee(nodeId: nodeId, extraFee: extraFee);
+        .setDedicatedNodeExtraFee(nodeId: nodeId, extraFee: BigInt.from(extraFee));
     await client.apply(extrinsic);
 
     return await getDedicatedNodeExtraFee(nodeId: nodeId);

packages/tfchain_client/lib/tfchain_client.dart

@@ -15,11 +15,11 @@ import 'package:polkadart/polkadart.dart'
 import 'package:polkadart/scale_codec.dart';
 import 'package:polkadart_keyring/polkadart_keyring.dart';
 import 'package:tfchain_client/generated/dev/types/tfchain_runtime/runtime_call.dart';
-import 'package:tfchain_client/src/Error_mapping.dart';
 import 'package:tfchain_client/src/balances.dart' as balance;
 import 'package:tfchain_client/src/contracts.dart';
 import 'package:tfchain_client/generated/dev/dev.dart' as polkadot;
 import 'package:tfchain_client/src/dao.dart' as Dao;
+import 'package:tfchain_client/src/error_mapper.dart';
 import 'package:tfchain_client/src/farms.dart';
 import 'package:tfchain_client/src/kvstore.dart';
 import 'package:tfchain_client/src/nodes.dart';

packages/tfchain_client/test/balances_test.dart

@@ -22,9 +22,48 @@ void main() {
         AccountInfo? accountInfo =
             await queryClient.balances.get(address: address);
         expect(accountInfo, isNull);
-      } catch (e) {
-        expect(e, isNotNull);
+      } catch (error) {
+        expect(error, isNotNull);
       }
     });
   });
+
+  group("Test Balances", () {
+    late Client client;
+    setUp(() async {
+      client = Client(
+          "wss://tfchain.dev.grid.tf/ws",
+          "secret add bag cluster deposit beach illness letter crouch position rain arctic",
+          "sr25519");
+      await client.connect();
+    });
+
+    test('Test Transfer TFTs with invalid amount', () async {
+      try {
+        await client.balances.transfer(
+            address:
+                "oven strong mention shoulder night ghost correct exercise surge lady jungle hundred",
+            amount: 0);
+      } catch (error) {
+        expect(error, isNotNull);
+      }
+    });
+
+    // TODO:
+    test('Test Transfer TFTs', () async {
+      try {
+        await client.balances.transfer(
+            address: "5EcSXeEH35LriE2aWxX6v4yZSMq47vdJ1GgHEXDdhJxg9XjG",
+            amount: 10);
+      } catch (error) {
+        print(error);
+        expect(error, isNull);
+      }
+    });
+
+    test('Test get my balance', () async {
+      AccountInfo? info = await client.balances.getMyBalance();
+      expect(info, isNotNull);
+    });
+  });
 }

packages/tfchain_client/test/contracts_test.dart

@@ -6,8 +6,9 @@ import 'package:tfchain_client/tfchain_client.dart';
 void main() {
   group("Query Contracts Test", () {
     late QueryClient queryClient;
-    setUp(() {
+    setUp(() async {
       queryClient = QueryClient("wss://tfchain.dev.grid.tf/ws");
+      await queryClient.connect();
     });
 
     test('Test Get Deleted Contract by Id', () async {
@@ -83,11 +84,67 @@ void main() {
       }
     });
 
-    test('Test Get ADedicated Node extra fee', () async {
+    test('Test Get a Dedicated Node extra fee', () async {
       BigInt? fee =
           await queryClient.contracts.getDedicatedNodeExtraFee(nodeId: 206);
       print(fee);
       expect(fee, BigInt.from(0));
     });
   });
+
+  group("Test Contracts", () {
+    late Client client;
+    setUp(() async {
+      client = Client(
+          "wss://tfchain.dev.grid.tf/ws",
+          "secret add bag cluster deposit beach illness letter crouch position rain arctic",
+          "sr25519");
+      await client.connect();
+    });
+
+    test('Test Create Node Contract with empty data', () async {
+      try {
+        BigInt? contractId = await client.contracts.createNode(
+            nodeId: 11, deploymentData: [], deploymentHash: [], publicIps: 0);
+      } catch (error) {
+        expect(error, isNotNull);
+      }
+    });
+
+    test('Test Update Node Contract with wrong data', () async {
+      try {
+        Contract? contract = await client.contracts
+            .updateNode(contractId: 11, deploymentData: [], deploymentHash: []);
+      } catch (error) {
+        expect(error, isNotNull);
+      }
+    });
+
+    test('Test Create Name Contract then cancel it', () async {
+      BigInt? contractId =
+          await client.contracts.createName(name: "contractname");
+      expect(contractId, isNotNull);
+      print(contractId);
+      try {
+        await client.contracts.cancel(contractId: contractId!);
+      } catch (error) {
+        expect(error, isNull);
+      }
+    });
+
+    test('Test Create Rent Contract with no solution provider', () async {
+      try {
+        BigInt? contractId = await client.contracts
+            .createRent(nodeId: 72, solutionProviderId: 0);
+      } catch (error) {
+        expect(error.toString(), contains('NoSuchSolutionProvider'));
+      }
+    });
+
+    test('Test set dedicated node extra fee', () async {
+      BigInt? extraFee = await client.contracts
+          .setDedicatedNodeExtraFee(nodeId: 140, extraFee: 2);
+      expect(extraFee, BigInt.from(2));
+    });
+  });
 }