Merge pull request #32 from algorandfoundation/fix/acct-data

fix: allow v11 acct param fields to be set in any.account() call
algorandfoundation · Jan 15, 2025 · d90b5f6 · d90b5f6
2 parents 7086c0c + d60f869
commit d90b5f6
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/src/value-generators/avm.ts b/src/value-generators/avm.ts
@@ -10,6 +10,8 @@ import { asBigInt, asUint64Cls, getRandomBigInt, getRandomBytes } from '../util'
 type AccountContextData = Partial<AccountData['account']> & {
   address?: Account
   incentiveEligible?: boolean
+  lastProposed?: uint64
+  lastHeartbeat?: uint64
   optedAssetBalances?: Map<internal.primitives.StubUint64Compat, internal.primitives.StubUint64Compat>
   optedApplications?: Application[]
 }
@@ -57,8 +59,10 @@ export class AvmValueGenerator {
     }
 
     const data = new AccountData()
-    const { address, optedAssetBalances, optedApplications, incentiveEligible, ...accountData } = input ?? {}
+    const { address, optedAssetBalances, optedApplications, incentiveEligible, lastProposed, lastHeartbeat, ...accountData } = input ?? {}
     data.incentiveEligible = incentiveEligible ?? false
+    data.lastProposed = lastProposed
+    data.lastHeartbeat = lastHeartbeat
     data.account = {
       ...data.account,
       ...accountData,

diff --git a/tests/state-op-codes.spec.ts b/tests/state-op-codes.spec.ts
@@ -136,8 +136,10 @@ describe('State op codes', async () => {
         address: dummyAccount,
         balance: Uint64(INITIAL_BALANCE_MICRO_ALGOS + 100000),
       })
+      const incentiveEligible = op.AcctParams.acctIncentiveEligible(mockAccount)
       const lastProposed = op.AcctParams.acctLastProposed(mockAccount)
       const lastHeartbeat = op.AcctParams.acctLastHeartbeat(mockAccount)
+      expect(incentiveEligible).toEqual([false, true])
       expect(lastProposed).toEqual([Global.round, true])
       expect(lastHeartbeat).toEqual([Global.round, true])
     })
@@ -146,10 +148,14 @@ describe('State op codes', async () => {
       const mockAccount = ctx.any.account({
         address: dummyAccount,
         balance: Uint64(INITIAL_BALANCE_MICRO_ALGOS + 100000),
+        incentiveEligible: true,
+        lastProposed: 100,
+        lastHeartbeat: 200,
       })
-      ctx.ledger.patchAccountData(mockAccount, { lastProposed: 100, lastHeartbeat: 200 })
+      const incentiveEligible = op.AcctParams.acctIncentiveEligible(mockAccount)
       const lastProposed = op.AcctParams.acctLastProposed(mockAccount)
       const lastHeartbeat = op.AcctParams.acctLastHeartbeat(mockAccount)
+      expect(incentiveEligible).toEqual([true, true])
       expect(lastProposed).toEqual([100, true])
       expect(lastHeartbeat).toEqual([200, true])
     })