nodejs: fix some block tests

iotaledger · Dec 8, 2023 · 12a79a4 · 12a79a4
1 parent 6a4f41b
commit 12a79a4
Show file tree

Hide file tree

Showing 11 changed files with 596 additions and 105 deletions.
diff --git a/bindings/core/src/method/utils.rs b/bindings/core/src/method/utils.rs
@@ -57,6 +57,7 @@ pub enum UtilsMethod {
     #[serde(rename_all = "camelCase")]
     BlockId {
         block: BlockDto,
+        #[serde(rename = "params")]
         protocol_parameters: ProtocolParameters,
     },
     /// Returns the transaction ID (Blake2b256 hash of the provided transaction payload)

diff --git a/bindings/nodejs/lib/client/client.ts b/bindings/nodejs/lib/client/client.ts
@@ -340,8 +340,13 @@ export class Client {
     /**
      * Get the token supply.
      */
-    async getTokenSupply(): Promise<u64> {
-        return (await this.getProtocolParameters()).tokenSupply;
+    async getTokenSupply(): Promise<BigInt> {
+        const tokenSupply = (await this.getProtocolParameters()).tokenSupply;
+        if (tokenSupply !== undefined) {
+            return BigInt(tokenSupply);
+        } else {
+            throw new Error("ProtocolParameters.tokenSupply is undefined");
+        }
     }
 
     /**

diff --git a/bindings/nodejs/lib/types/block/core/block.ts b/bindings/nodejs/lib/types/block/core/block.ts
@@ -10,6 +10,7 @@ import { BlockBody } from './block-body';
 import { BasicBlockBody } from './basic';
 import { ValidationBlockBody } from './validation';
 import { BlockBodyDiscriminator } from '.';
+import { Utils, ProtocolParameters, BlockId } from '../../..';
 
 /**
  * The block header.
@@ -118,6 +119,13 @@ class Block {
     asValidation(): ValidationBlockBody {
         return this.body.asValidation();
     }
+
+    /**
+     * Gets the block id.
+     */
+    id(params: ProtocolParameters): BlockId {
+        return Utils.blockId(this, params);
+    }
 }
 
 function parseBlock(data: any): Block {