Refactor how group transactions are implemented in testing

algorandfoundation · Sep 25, 2024 · 24e2179 · 24e2179
1 parent ab8b347
commit 24e2179
Show file tree

Hide file tree

Showing 7 changed files with 321 additions and 241 deletions.
diff --git a/examples/hello-world-abi/contract.spec.ts b/examples/hello-world-abi/contract.spec.ts
@@ -1,7 +1,8 @@
-import type { gtxn } from '@algorandfoundation/algo-ts'
+import { TransactionType } from '@algorandfoundation/algo-ts'
+import { TestExecutionContext } from '@algorandfoundation/algo-ts-testing'
 import { afterEach, describe, expect, it } from 'vitest'
+import { invariant } from '../../src/util'
 import HelloWorldContract from './contract.algo'
-import { TestExecutionContext } from '@algorandfoundation/algo-ts-testing'
 
 describe('HelloWorldContract', () => {
   const ctx = new TestExecutionContext()
@@ -11,19 +12,17 @@ describe('HelloWorldContract', () => {
   it('logs the returned value when sayBananas is called', async () => {
     const contract = ctx.contract.create(HelloWorldContract)
     const result = contract.sayBananas()
-
-    const application = (ctx.txn.lastActive as gtxn.ApplicationTxn).appId
+    invariant(ctx.txn.lastActive.type === TransactionType.ApplicationCall, 'Last txn must be app')
 
     expect(result).toBe('Bananas')
-    expect(ctx.exportLogs(application.id, 's')).toStrictEqual([result])
+    expect(ctx.exportLogs(ctx.txn.lastActive.appId.id, 's')).toStrictEqual([result])
   })
   it('logs the returned value when sayHello is called', async () => {
     const contract = ctx.contract.create(HelloWorldContract)
     const result = contract.sayHello('John', 'Doe')
-
-    const application = (ctx.txn.lastActive as gtxn.ApplicationTxn).appId
+    invariant(ctx.txn.lastActive.type === TransactionType.ApplicationCall, 'Last txn must be app')
 
     expect(result).toBe('Hello John Doe')
-    expect(ctx.exportLogs(application.id, 's')).toStrictEqual([result])
+    expect(ctx.exportLogs(ctx.txn.lastActive.appId.id, 's')).toStrictEqual([result])
   })
 })