improve balance and add address + tests

packages/auto-consensus/__test__/balances.test.ts

@@ -1,21 +1,79 @@
-import { activate, disconnect } from '@autonomys/auto-utils'
-import { totalIssuance } from '../src/balances'
+import type { NetworkInput } from '@autonomys/auto-utils'
+import { activate, activateWallet, disconnect, networks } from '@autonomys/auto-utils'
+import { address } from '../src/address'
+import { totalIssuance, transfer } from '../src/balances'
 
 describe('Verify balances functions', () => {
+  const isLocalhost = process.env.LOCALHOST === 'true'
+
+  // Define the test network and its details
+  const TEST_NETWORK: NetworkInput = !isLocalhost
+    ? { networkId: networks[0].id }
+    : { networkId: 'autonomys-localhost' }
+  const TEST_INVALID_NETWORK = { networkId: 'invalid-network' }
+
+  const TEST_MNEMONIC = 'test test test test test test test test test test test junk'
+  const TEST_ADDRESS = '5Fj5aLd4crCYn7zM5hLZL8m6e9aNzWssiTgA3TrprLjxy6Mc'
+  const ALICE_URI = '//BOB'
+  const ALICE_ADDRESS = '5DAw2FpYk2y3JHrsia14KEx7tpezNymdFKkunicZ5ygPGXYF'
+  const BOB_URI = '//BOB'
+  const BOB_ADDRESS = '5DAw2FpYk2y3JHrsia14KEx7tpezNymdFKkunicZ5ygPGXYF'
+
   beforeAll(async () => {
-    await activate()
+    await activate(TEST_NETWORK)
   })
 
   afterAll(async () => {
     await disconnect()
   })
 
-  test('Check totalIssuance return a number greater than zero', async () => {
-    // totalIssuance is an async function that returns a hex number as a string
-    const rawIssuance = await totalIssuance()
-    // Convert the hex number to a BigInt
-    const issuance = BigInt(rawIssuance.toString())
-    // Check if the issuance is greater than zero
-    expect(issuance).toBeGreaterThan(BigInt(0))
+  describe('Test totalIssuance()', () => {
+    test('Check totalIssuance return a number greater than zero', async () => {
+      // totalIssuance is an async function that returns a hex number as a string
+      const rawIssuance = await totalIssuance()
+      // Convert the hex number to a BigInt
+      const issuance = BigInt(rawIssuance.toString())
+      // Check if the issuance is greater than zero
+      expect(issuance).toBeGreaterThan(BigInt(0))
+    })
+  })
+
+  describe('Test balance()', () => {
+    test('Check balance of Test wallet is 0', async () => {
+      const { api, accounts } = await activateWallet({
+        ...TEST_NETWORK,
+        mnemonic: TEST_MNEMONIC,
+      })
+
+      const balance = await api.query.balances.account(address(accounts[0].address))
+      expect(balance.free.toBigInt()).toEqual(BigInt(0))
+    })
+  })
+
+  describe('Test transfer()', () => {
+    if (isLocalhost) {
+      test('Check transfer 1 ATC between Alice and Bob', async () => {
+        const { api, accounts } = await activateWallet({
+          ...TEST_NETWORK,
+          uri: ALICE_URI,
+        })
+        const sender = accounts[0]
+
+        const hash = await transfer(api, sender, BOB_ADDRESS, 1)
+      })
+    }
+
+    test('Check transfer 1 ATC between Test wallet and Alice should fail (no balance)', async () => {
+      const { api, accounts } = await activateWallet({
+        ...TEST_NETWORK,
+        mnemonic: TEST_MNEMONIC,
+      })
+      const sender = accounts[0]
+
+      expect(() => transfer(api, sender, ALICE_ADDRESS, 1)).toThrow(
+        'Unreachable code should not be executed (evaluating',
+        // To-Do: Confirm this is the expected error message
+      )
+    })
   })
 })

packages/auto-consensus/src/address.ts

@@ -0,0 +1,5 @@
+import { encodeAddress } from '@polkadot/keyring'
+
+export const address = (address: string | Uint8Array): string => {
+  return encodeAddress(address, 2254)
+}

packages/auto-consensus/src/balances.ts

@@ -1,4 +1,6 @@
 import { activate } from '@autonomys/auto-utils'
+import { ApiPromise } from '@polkadot/api'
+import type { KeyringPair } from '@polkadot/keyring/types'
 
 export const totalIssuance = async (networkId?: string) => {
   // Get the api instance for the network
@@ -9,3 +11,15 @@ export const totalIssuance = async (networkId?: string) => {
 
   return totalIssuance
 }
+
+export const transfer = async (
+  api: ApiPromise,
+  sender: KeyringPair,
+  receiver: string,
+  amount: BigInt | number | string,
+) => {
+  // Transfer the tokens
+  const transfer = await api.tx.balances.transferKeepAlive(receiver, amount).signAndSend(sender)
+
+  return transfer
+}

packages/auto-consensus/src/index.ts

@@ -1,2 +1,3 @@
+export * from './address'
 export * from './balances'
 export * from './info'

packages/auto-utils/__test__/wallet.test.ts

@@ -14,7 +14,7 @@ describe('Verify wallet functions', () => {
 
   const TEST_INVALID_NETWORK = { networkId: 'invalid-network' }
 
-  const TEST_MNEMONIC = 'test test test test test test test test test test test junk' //Alice
+  const TEST_MNEMONIC = 'test test test test test test test test test test test junk'
   const TEST_ADDRESS = '5Fj5aLd4crCYn7zM5hLZL8m6e9aNzWssiTgA3TrprLjxy6Mc'
   const ALICE_URI = '//BOB'
   const ALICE_ADDRESS = '5DAw2FpYk2y3JHrsia14KEx7tpezNymdFKkunicZ5ygPGXYF'