fix missing Principal import in Func, RequireExactlyOne import in Var…

…iant, tsType for AzleBlob
demergent-labs · Dec 6, 2023 · fd555e0 · fd555e0
1 parent 1c38ca2
commit fd555e0
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 2 deletions.
diff --git a/property_tests/arbitraries/candid/constructed/variant_arb/definition_arbs.ts b/property_tests/arbitraries/candid/constructed/variant_arb/definition_arbs.ts
@@ -67,7 +67,7 @@ function generateImports(fields: Field[]): Set<string> {
     const fieldImports = fields.flatMap((field) => [
         ...field[1].candidMeta.imports
     ]);
-    return new Set([...fieldImports, 'Variant']);
+    return new Set([...fieldImports, 'RequireExactlyOne', 'Variant']);
 }
 
 function generateVariableAliasDeclarations(

diff --git a/property_tests/arbitraries/candid/reference/func_arb/definition_arb.ts b/property_tests/arbitraries/candid/reference/func_arb/definition_arb.ts
@@ -57,7 +57,8 @@ export function FuncDefinitionArb(
                 const imports = new Set([
                     ...params.flatMap((param) => [...param.candidMeta.imports]),
                     ...returnFunc.candidMeta.imports,
-                    'Func'
+                    'Func',
+                    'Principal'
                 ]);
 
                 return {

diff --git a/src/lib/candid/types/constructed/blob.ts b/src/lib/candid/types/constructed/blob.ts
@@ -6,6 +6,8 @@ export class AzleBlob {
     _azleKind: 'AzleBlob' = 'AzleBlob';
     static _azleKind: 'AzleBlob' = 'AzleBlob';
 
+    static tsType: AzleBlob;
+
     static toBytes(data: any) {
         return encode(this, data);
     }