simple pr fixes

demergent-labs · Jan 17, 2024 · 4ffda3f · 4ffda3f
1 parent 43d286d
commit 4ffda3f
Show file tree

Hide file tree

Showing 6 changed files with 19 additions and 20 deletions.
diff --git a/property_tests/arbitraries/canister_arb.ts b/property_tests/arbitraries/canister_arb.ts
@@ -72,13 +72,12 @@ export function CanisterArb<
             ...(config.updateMethods ?? [])
         ];
 
-        const deployArgs = config.initMethod?.params.map(
-            ({ value: { value } }) => {
-                return value.runtimeCandidTypeObject
-                    .getIdl([])
-                    .valueToString(value.agentArgumentValue);
-            }
-        );
+        const deployArgs = config.initMethod?.params.map((param) => {
+            const value = param.value.value;
+            return value.runtimeCandidTypeObject
+                .getIdl([])
+                .valueToString(value.agentArgumentValue);
+        });
 
         const sourceCode = generateSourceCode(
             config.globalDeclarations ?? [],

diff --git a/property_tests/arbitraries/canister_methods/init_method_arb.ts b/property_tests/arbitraries/canister_methods/init_method_arb.ts
@@ -45,7 +45,7 @@ export function InitMethodArb<
         >;
         callbackLocation?: CallbackLocation;
     }
-) {
+): fc.Arbitrary<InitMethod<ParamAgentArgumentValue, ParamAgentResponseValue>> {
     return fc
         .tuple(
             UniqueIdentifierArb('canisterMethod'),

diff --git a/property_tests/arbitraries/canister_methods/post_upgrade_arb.ts b/property_tests/arbitraries/canister_methods/post_upgrade_arb.ts
@@ -45,16 +45,15 @@ export function PostUpgradeMethodArb<
         >;
         callbackLocation?: CallbackLocation;
     }
-) {
+): fc.Arbitrary<
+    PostUpgradeMethod<ParamAgentArgumentValue, ParamAgentResponseValue>
+> {
     return fc
         .tuple(
             UniqueIdentifierArb('canisterMethod'),
             paramTypeArrayArb,
             VoidArb(),
-            fc.constantFrom(
-                'INLINE',
-                'STANDALONE'
-            ) as fc.Arbitrary<CallbackLocation>,
+            fc.constantFrom<CallbackLocation>('INLINE', 'STANDALONE'),
             UniqueIdentifierArb('typeDeclaration')
             // TODO: This unique id would be better named globalScope or something
             // But needs to match the same scope as typeDeclarations so I'm using
@@ -67,9 +66,10 @@ export function PostUpgradeMethodArb<
                 returnType,
                 defaultCallbackLocation,
                 callbackName
-            ]) => {
-                // TODO: Add a return type to this map callback of type PostUpgradeMethod<Something, Something>
-
+            ]): PostUpgradeMethod<
+                ParamAgentArgumentValue,
+                ParamAgentResponseValue
+            > => {
                 const callbackLocation =
                     constraints.callbackLocation ?? defaultCallbackLocation;
 
@@ -81,7 +81,7 @@ export function PostUpgradeMethodArb<
                 const namedParams = paramTypes.map(
                     <T>(param: T, index: number): Named<T> => ({
                         name: `param${index}`,
-                        el: param
+                        value: param
                     })
                 );
 

diff --git a/property_tests/tests/post_upgrade_method/test/generate_init_tests.ts b/property_tests/tests/post_upgrade_method/test/generate_init_tests.ts
@@ -9,7 +9,7 @@ export function generateTests(
 ): Test[][] {
     const expectedResult = [
         true,
-        ...params.map((param) => param.el.value.agentResponseValue)
+        ...params.map((param) => param.value.value.agentResponseValue)
     ];
 
     return [

diff --git a/property_tests/tests/post_upgrade_method/test/generate_post_upgrade_tests.ts b/property_tests/tests/post_upgrade_method/test/generate_post_upgrade_tests.ts
@@ -9,7 +9,7 @@ export function generateTests(
 ): Test[][] {
     const expectedResult = [
         true,
-        ...params.map((param) => param.el.value.agentResponseValue)
+        ...params.map((param) => param.value.value.agentResponseValue)
     ];
 
     return [

diff --git a/property_tests/tests/post_upgrade_method/test/test.ts b/property_tests/tests/post_upgrade_method/test/test.ts
@@ -1,7 +1,7 @@
 import fc from 'fast-check';
 
 import { runPropTests } from 'azle/property_tests';
-import { CandidValueAndMetaArbWithoutFuncs as CandidValueAndMetaArb } from 'azle/property_tests/arbitraries/candid/candid_value_and_meta_arb';
+import { CandidValueAndMetaArb } from 'azle/property_tests/arbitraries/candid/candid_value_and_meta_arb';
 import { CandidReturnTypeArb } from 'azle/property_tests/arbitraries/candid/candid_return_type_arb';
 import {
     CanisterArb,