Fix ConstantManager.set to accept array of serializable (#343)

massalabs · May 24, 2024 · ccdc140 · ccdc140
1 parent ecca530
commit ccdc140
Show file tree

Hide file tree

Showing 4 changed files with 19 additions and 5 deletions.
diff --git a/assembly/helpers/__tests__/constantKeyManager.spec.ts b/assembly/helpers/__tests__/constantKeyManager.spec.ts
@@ -51,4 +51,18 @@ describe('ConstantManager - unit tests', () => {
     expect(cst.tryValue().isErr()).toBe(true);
     expect(cst.tryValue().error).toBe('Key not found');
   });
+
+  test('get/set - array of serializable', () => {
+    const addr1 = new Address(
+      'AU12UBnqTHDQALpocVBnkPNy7y5CndUJQTLutaVDDFgMJcq5kQiKq',
+    );
+    const addr2 = new Address(
+      'AU12UBnqTHDQALpocVBnkPNy7y5CndUJQTLutaVDDFgMJcq5kQiKr',
+    );
+
+    const addresses = new ConstantManager<Array<Address>, u8, Address>();
+    addresses.set([addr1, addr2]);
+
+    expect(addresses.mustValue()).toStrictEqual([addr1, addr2]);
+  });
 });
diff --git a/assembly/helpers/constantKeyManager.ts b/assembly/helpers/constantKeyManager.ts
@@ -49,6 +49,6 @@ export class ConstantManager<TValue, TKey = u8, TArray = void> {
    * @param value - The value to store. Must be an A
    */
   public set(value: TValue): void {
-    Storage.set(this.key, new Args().add(value).serialize());
+    Storage.set(this.key, new Args().add<TValue, TArray>(value).serialize());
   }
 }
diff --git a/package-lock.json b/package-lock.json
diff --git a/package.json b/package.json
@@ -38,7 +38,7 @@
   "devDependencies": {
     "@as-pect/cli": "^8.1.0",
     "@massalabs/as-transformer": "^0.3.2",
-    "@massalabs/as-types": "^2.0.1-dev.20240521132240",
+    "@massalabs/as-types": "^2.0.1-dev.20240524080933",
     "@massalabs/eslint-config": "^0.0.8",
     "@massalabs/prettier-config-as": "^0.0.2",
     "as-bignum": "^0.2.40",