algorand-python-testing 0.3.0b8__tar.gz → 0.3.0b10__tar.gz

{algorand_python_testing-0.3.0b8 → algorand_python_testing-0.3.0b10}/.github/workflows/ci.yaml

@@ -42,3 +42,6 @@ jobs:
 
       - name: Run tests (examples)
         run: hatch run examples:tests
+
+      - name: Check doctests
+        run: hatch run docs:test

{algorand_python_testing-0.3.0b8 → algorand_python_testing-0.3.0b10}/.github/workflows/gh-pages.yaml

@@ -24,6 +24,9 @@ jobs:
           python-version: "3.12"
           cache: "pip"
 
+      - name: Test doctests
+        run: hatch run docs:test
+
       - name: Build doc
         run: hatch run docs:build
 

{algorand_python_testing-0.3.0b8 → algorand_python_testing-0.3.0b10}/CHANGELOG.md

@@ -1,4 +1,8 @@
 # CHANGELOG
+## v0.3.0-beta.10 (2024-08-22)
+
+## v0.3.0-beta.9 (2024-08-22)
+
 ## v0.3.0-beta.8 (2024-08-22)
 
 ### Feature

{algorand_python_testing-0.3.0b8 → algorand_python_testing-0.3.0b10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: algorand-python-testing
-Version: 0.3.0b8
+Version: 0.3.0b10
 Summary: Algorand Python testing library
 Project-URL: Documentation, https://github.com/algorandfoundation/puya/tree/main/algopy_testing#README.md
 Project-URL: Issues, https://github.com/algorandfoundation/puya/issues

{algorand_python_testing-0.3.0b8 → algorand_python_testing-0.3.0b10}/docs/api.md

@@ -1,5 +1,15 @@
 # API Reference
 
+An overview of the `algorand-python-testing`'s `algopy_testing` module - covering the main classes and functions.
+
+```{hint}
+Spotted a typo in documentation? This project is open source, please submit an issue or a PR on [GitHub](https://github.com/algorand/algorand-python-testing).
+```
+
+```{warning}
+Note, assume `_algopy_testing` to refer to `algopy_testing` namespace in the auto-generated class documentation above. To be patched in near future.
+```
+
 ## Contexts
 
 ```{autodoc2-summary}

{algorand_python_testing-0.3.0b8 → algorand_python_testing-0.3.0b10}/docs/conf.py

@@ -84,3 +84,13 @@ doctest_test_doctest_blocks = "default"
 sphinxmermaid_mermaid_init = {
     "theme": "dark",
 }
+
+# Ignore specific warning types
+# TODO : remove upon refining autodoc2 (or finding alternative to sphinx)
+suppress_warnings = [
+    "myst.xref_missing",
+    "autodoc2.dup_item",
+    "ref.python",  # Ignore Python reference warnings
+    "ref.class",  # Ignore class reference warnings
+    "ref.obj",  # Ignore object reference warnings
+]

algorand_python_testing-0.3.0b10/docs/faq.md

@@ -0,0 +1,61 @@
+# FAQ
+
+## What is a Test Context?
+
+A Test Context is a context manager that provides a simulated Algorand environment for testing Python smart contracts. It allows developers to create and manipulate a virtual Algorand ecosystem for testing purposes. For more details, see the [Test Context section](testing-guide/concepts.md#test-context) in our documentation.
+
+## What is the Algorand Virtual Machine (AVM)?
+
+The Algorand Virtual Machine (AVM) is the runtime environment for Algorand smart contracts. It executes the compiled code of smart contracts on the Algorand blockchain. To learn more about the AVM, visit the [official Algorand documentation](https://developer.algorand.org/docs/get-details/dapps/avm/).
+
+## What are Operational Codes in Algorand?
+
+Operational Codes, or opcodes, are AVM instructions that are executed directly by the AVM. In the context of Algorand Python testing, these opcodes are provided by `algopy` stubs and are either emulated, implemented, or mocked by `algorand-python-testing`. For a comprehensive list of opcodes, refer to the [Algorand Developer Documentation](https://developer.algorand.org/docs/get-details/dapps/avm/teal/opcodes/?from_query=OPcodes#template-modal-overlay).
+
+## What are Value Generators?
+
+Value Generators are helper methods that generate randomized values for testing when the specific value of the tested type is not important. In the context of Algorand Python testing, these are represented by property on the context manager, accessed via `any.*` (or `any.arc4.*`, `any.txn.*`. in the case of ARC 4 types). To understand how to use Value Generators effectively, check out our [Value Generators section](testing-guide/concepts.md#value-generators) in the documentation.
+
+## What are the limitations of the Algorand Python Testing framework?
+
+The Algorand Python Testing framework emulates the Algorand Virtual Machine (AVM) for unit testing Algorand Python smart contracts without interacting with the real Algorand Network. However, it has some limitations due to its scope and purpose:
+
+1. Simplified balance tracking and transaction validation
+2. No consensus mechanism or AVM network operations simulation
+3. Absence of a strict opcode budget system
+4. Certain cryptographic operations are mocked or simplified
+5. No state proof generation or verification
+
+For scenarios where these limitations are crucial, it's recommended to pair this framework with integration testing. If you have a solid reason to justify introducing new emulated behaviour, please open an issue or contribute to the project on [Github](https://github.com/algorandfoundation/algorand-python-testing).
+
+## How does balance tracking work in the testing framework?
+
+The framework uses simplified balance tracking and transaction validation. For scenarios where precise balance or fee verification is important, it's recommended to complement unit tests with integration testing.
+
+## Does the framework simulate the entire AVM network?
+
+No, the framework does not simulate the entire AVM network or consensus mechanism. It focuses on emulating the parts of the AVM relevant to unit testing smart contracts.
+
+## How does the framework handle opcode budgets?
+
+The framework does not implement a strict opcode budget system. For scenarios where validating opcode budget is crucial, it's recommended to use integration testing alongside this framework.
+
+## Are all cryptographic operations fully implemented?
+
+Some cryptographic operations are mocked or simplified in the framework. For a detailed list of which operations are mocked, refer to the _mockable_ types under the [coverage](./coverage.md) section.
+
+## Can I use this framework for security-critical validations?
+
+While this framework is useful for unit testing and local development, it should not be the only tool used for security-critical validations or performance benchmarking. It's designed to approximate AVM behavior for common scenarios. Always complement your testing with additional integration testing options available in `algokit`, where you can test against real localnet or testnet environments.
+
+## Is there an example of how to use this framework alongside integration tests?
+
+Yes, the `algokit-python-template`, accessible via `algokit init`, provides a working example of how to structure `algorand-python-testing` along with regular integration tests against localnet.
+
+```{hint}
+An `algokit-python-template` accessible via `algokit init -t python`, provides a comprehensive and customizable working example of how to structure `algorand-python-testing` along with regular integration tests against localnet.
+```
+
+## Is it compatible with `pytest`?
+
+Yes, it is compatible with `pytest` and _any_ other python testing framework as its agnostic of the testing framework as long as its python. If you spot incompatibility with a certain tool, please open an issue or contribute to the project on [Github](https://github.com/algorandfoundation/algorand-python-testing).

{algorand_python_testing-0.3.0b8 → algorand_python_testing-0.3.0b10}/docs/index.md

@@ -13,18 +13,6 @@ The `algorand-python-testing` package provides:
 -   An offline testing environment that simulates core AVM functionality
 -   A familiar Pythonic experience, compatible with testing frameworks like [pytest](https://docs.pytest.org/en/latest/), [unittest](https://docs.python.org/3/library/unittest.html), and [hypothesis](https://hypothesis.readthedocs.io/en/latest/)
 
-```{testsetup}
-import algopy
-import algopy_testing
-from algopy_testing import algopy_testing_context
-
-# Create the context manager for snippets below
-ctx_manager = algopy_testing_context()
-
-# Enter the context
-context = ctx_manager.__enter__()
-```
-
 ## Quick Start
 
 `algopy` is a prerequisite for `algorand-python-testing`, providing stubs and type annotations for Algorand Python syntax. It enhances code completion and type checking when writing smart contracts. Note that this code isn't directly executable in standard Python interpreters; it's compiled by `puya` into TEAL for Algorand Network deployment.
@@ -188,11 +176,7 @@ hidden: true
 testing-guide/index
 examples
 coverage
-glossary
+faq
 api
 algopy
 ```
-
-```{testcleanup}
-ctx_manager.__exit__(None, None, None)
-```

{algorand_python_testing-0.3.0b8 → algorand_python_testing-0.3.0b10}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "algorand-python-testing"
-version = "0.3.0-beta.8"
+version = "0.3.0-beta.10"
 description = 'Algorand Python testing library'
 readme = "README.md"
 requires-python = ">=3.12"
@@ -150,8 +150,10 @@ post-install-commands = [
 ]
 
 [tool.hatch.envs.docs.scripts]
-build = "sphinx-build -b doctest docs docs/_build -W --keep-going -n -E"
-dev = "sphinx-build -b doctest docs docs/_build && sphinx-autobuild docs docs/_build"
+test = "sphinx-build -b doctest docs docs/_build -W --keep-going -n -E"
+clear = "rm -rf docs/_build"
+build = "hatch run docs:clear && sphinx-build docs docs/_build -W --keep-going -n -E"
+dev = "hatch run docs:test  && sphinx-autobuild docs docs/_build"
 
 # Examples environment
 [tool.hatch.envs.examples]

{algorand_python_testing-0.3.0b8 → algorand_python_testing-0.3.0b10}/src/_algopy_testing/context_helpers/context_storage.py

@@ -78,13 +78,13 @@ class _InternalContext:
 
     def get_asset_data(self, asset_id: int | algopy.UInt64) -> AssetFields:
         try:
-            return self.ledger.asset_data[int(asset_id)]
+            return self.ledger._asset_data[int(asset_id)]
         except KeyError:
             raise ValueError("Unknown asset, check correct testing context is active") from None
 
     def get_account_data(self, account_public_key: str) -> AccountContextData:
         try:
-            return self.ledger.account_data[account_public_key]
+            return self.ledger._account_data[account_public_key]
         except KeyError:
             raise ValueError("Unknown account, check correct testing context is active") from None
 
@@ -104,6 +104,9 @@ def algopy_testing_context(
     """
     from _algopy_testing.context import AlgopyTestContext
 
+    if _var.get(None) is not None:
+        raise RuntimeError("Nested `algopy_testing_context`s are not allowed.")
+
     token = _var.set(
         AlgopyTestContext(
             default_sender=default_sender,

{algorand_python_testing-0.3.0b8 → algorand_python_testing-0.3.0b10}/src/_algopy_testing/context_helpers/ledger_context.py

@@ -23,25 +23,25 @@ class LedgerContext:
     def __init__(self) -> None:
         from _algopy_testing.models.account import AccountContextData, get_empty_account
 
-        self.account_data = defaultdict[str, AccountContextData](get_empty_account)
-        self.app_data: dict[int, ApplicationContextData] = {}
-        self.asset_data: dict[int, AssetFields] = {}
-        self.blocks: dict[int, dict[str, int]] = {}
-        self.global_fields: GlobalFields = get_default_global_fields()
+        self._account_data = defaultdict[str, AccountContextData](get_empty_account)
+        self._app_data: dict[int, ApplicationContextData] = {}
+        self._asset_data: dict[int, AssetFields] = {}
+        self._blocks: dict[int, dict[str, int]] = {}
+        self._global_fields: GlobalFields = get_default_global_fields()
 
         self._asset_id = iter(range(1001, 2**64))
         self._app_id = iter(range(1001, 2**64))
 
-    def get_next_asset_id(self) -> int:
+    def _get_next_asset_id(self) -> int:
         while True:
             asset_id = next(self._asset_id)
-            if asset_id not in self.asset_data:
+            if asset_id not in self._asset_data:
                 return asset_id
 
-    def get_next_app_id(self) -> int:
+    def _get_next_app_id(self) -> int:
         while True:
             app_id = next(self._app_id)
-            if app_id not in self.app_data:
+            if app_id not in self._app_data:
                 return app_id
 
     def get_account(self, address: str) -> algopy.Account:
@@ -68,7 +68,7 @@ class LedgerContext:
             bool: True if the account exists, False otherwise.
         """
         assert_address_is_valid(address)
-        return address in self.account_data
+        return address in self._account_data
 
     def update_account(
         self,
@@ -84,11 +84,11 @@ class LedgerContext:
             **account_fields: The fields to update.
         """
         assert_address_is_valid(address)
-        self.account_data[address].fields.update(account_fields)
+        self._account_data[address].fields.update(account_fields)
 
         if opted_asset_balances is not None:
             for asset_id, balance in opted_asset_balances.items():
-                self.account_data[address].opted_asset_balances[UInt64(asset_id)] = balance
+                self._account_data[address].opted_asset_balances[UInt64(asset_id)] = balance
 
     def get_asset(self, asset_id: algopy.UInt64 | int) -> algopy.Asset:
         """Get an asset by ID.
@@ -105,7 +105,7 @@ class LedgerContext:
         import algopy
 
         asset_id = int(asset_id) if isinstance(asset_id, algopy.UInt64) else asset_id
-        if asset_id not in self.asset_data:
+        if asset_id not in self._asset_data:
             raise ValueError("Asset not found in testing context!")
 
         return algopy.Asset(asset_id)
@@ -122,7 +122,7 @@ class LedgerContext:
         import algopy
 
         asset_id = int(asset_id) if isinstance(asset_id, algopy.UInt64) else asset_id
-        return asset_id in self.asset_data
+        return asset_id in self._asset_data
 
     def update_asset(self, asset_id: int, **asset_fields: typing.Unpack[AssetFields]) -> None:
         """Update asset fields.
@@ -134,9 +134,9 @@ class LedgerContext:
         Raises:
             ValueError: If the asset is not found.
         """
-        if asset_id not in self.asset_data:
+        if asset_id not in self._asset_data:
             raise ValueError("Asset not found in testing context!")
-        self.asset_data[asset_id].update(asset_fields)
+        self._asset_data[asset_id].update(asset_fields)
 
     def get_app(
         self, app_id: algopy.Contract | algopy.Application | algopy.UInt64 | int
@@ -164,7 +164,7 @@ class LedgerContext:
             bool: True if the application exists, False otherwise.
         """
         app_id = _get_app_id(app_id)
-        return app_id in self.app_data
+        return app_id in self._app_data
 
     def update_app(
         self, app_id: int, **application_fields: typing.Unpack[ApplicationFields]
@@ -343,7 +343,7 @@ class LedgerContext:
             seed (algopy.UInt64 | int): The block seed.
             timestamp (algopy.UInt64 | int): The block timestamp.
         """
-        self.blocks[index] = {"seed": int(seed), "timestamp": int(timestamp)}
+        self._blocks[index] = {"seed": int(seed), "timestamp": int(timestamp)}
 
     def get_block_content(self, index: int, key: str) -> int:
         """Get block content.
@@ -358,7 +358,7 @@ class LedgerContext:
         Raises:
             ValueError: If the block content is not found.
         """
-        content = self.blocks.get(index, {}).get(key, None)
+        content = self._blocks.get(index, {}).get(key, None)
         if content is None:
             raise KeyError(
                 f"Block content for index {index} and key {key} not found in testing context!"
@@ -383,7 +383,7 @@ class LedgerContext:
                 f"Invalid field(s) found during patch for `Global`: {', '.join(invalid_keys)}"
             )
 
-        self.global_fields.update(global_fields)
+        self._global_fields.update(global_fields)
 
     def _get_app_data(
         self, app: algopy.UInt64 | algopy.Application | algopy.Contract | int
@@ -401,7 +401,7 @@ class LedgerContext:
         """
         app_id = _get_app_id(app)
         try:
-            return self.app_data[app_id]
+            return self._app_data[app_id]
         except KeyError:
             raise ValueError("Unknown app id, is there an active transaction?") from None
 

{algorand_python_testing-0.3.0b8 → algorand_python_testing-0.3.0b10}/src/_algopy_testing/context_helpers/txn_context.py

@@ -1,6 +1,7 @@
 from __future__ import annotations
 
 import contextlib
+import time
 import typing
 
 import algosdk
@@ -144,6 +145,8 @@ class TransactionContext:
         :param active_txn_overrides: Overrides for active txn
         :return: None
         """
+        if self._active_group is not None:
+            raise RuntimeError("Nested `create_group` calls are not allowed.")
         if gtxns and active_txn_overrides:
             raise ValueError("cannot specified gtxns and active_txn_overrides at the same time")
         if active_txn_index is not None and not gtxns:
@@ -163,18 +166,18 @@ class TransactionContext:
             last_app_call_txn = app_calls[-1]._txns[-1]
             active_txn_index = processed_gtxns.index(last_app_call_txn)
 
-        previous_group = self._active_group
-        active_group = self._active_group = TransactionGroup(
+        new_group = TransactionGroup(
             txns=processed_gtxns,
             active_txn_index=active_txn_index,
             active_txn_overrides=typing.cast(dict[str, typing.Any], active_txn_overrides),
         )
+        self._active_group = new_group
         try:
             yield
         finally:
-            if active_group.txns:
-                self._groups.append(active_group)
-            self._active_group = previous_group
+            if new_group.txns:
+                self._groups.append(new_group)
+            self._active_group = None
 
     @contextlib.contextmanager
     def _get_or_create_group(self) -> Iterator[TransactionGroup]:
@@ -207,6 +210,7 @@ class TransactionGroup:
         active_txn_index: int | None = None,
         active_txn_overrides: dict[str, typing.Any] | None = None,
     ):
+        self._latest_timestamp = int(time.time())
         self._set_txn_group(txns, active_txn_index)
         self._itxn_groups: list[Sequence[InnerTransactionResultType]] = []
         self._constructing_itxn_group: list[InnerTransaction] = []

{algorand_python_testing-0.3.0b8 → algorand_python_testing-0.3.0b10}/src/_algopy_testing/models/account.py

@@ -82,7 +82,7 @@ class Account(BytesBacked):
     def data(self) -> AccountContextData:
         from _algopy_testing.context_helpers import lazy_context
 
-        return lazy_context.ledger.account_data[self.public_key]
+        return lazy_context.get_account_data(self.public_key)
 
     @property
     def balance(self) -> algopy.UInt64:

{algorand_python_testing-0.3.0b8 → algorand_python_testing-0.3.0b10}/src/_algopy_testing/models/application.py

@@ -1,11 +1,10 @@
 from __future__ import annotations
 
-import inspect
 import typing
 
 from _algopy_testing.primitives import UInt64
 from _algopy_testing.protocols import UInt64Backed
-from _algopy_testing.utils import as_int64
+from _algopy_testing.utils import as_int64, resolve_app_index
 
 if typing.TYPE_CHECKING:
     from collections.abc import Sequence
@@ -70,22 +69,15 @@ class Application(UInt64Backed):
     def fields(self) -> ApplicationFields:
         from _algopy_testing.context_helpers import lazy_context
 
-        if self._id == 0:
-            raise ValueError("cannot access properties of an app with an id of 0") from None
-        return lazy_context.get_app_data(self._id).fields
+        return lazy_context.get_app_data(resolve_app_index(self._id)).fields
 
     def __getattr__(self, name: str) -> typing.Any:
-        if name in inspect.get_annotations(ApplicationFields):
-            value = self.fields.get(name)
-            # TODO: 1.0 ensure reasonable default values are present (like account does)
-            if value is None:
-                raise ValueError(
-                    f"The Application value '{name}' has not been defined on the test context. "
-                    f"Make sure to patch the field '{name}' using your `AlgopyTestContext` "
-                    "instance."
-                )
-            return value
-        raise AttributeError(f"'{self.__class__.__name__}' object has no attribute '{name}'")
+        try:
+            return self.fields[name]  # type: ignore[literal-required]
+        except KeyError:
+            raise AttributeError(
+                f"'{self.__class__.__name__}' object has no attribute '{name}'"
+            ) from None
 
     def __eq__(self, other: object) -> bool:
         if isinstance(other, Application):

{algorand_python_testing-0.3.0b8 → algorand_python_testing-0.3.0b10}/src/_algopy_testing/models/asset.py

@@ -3,6 +3,7 @@ from __future__ import annotations
 import typing
 
 from _algopy_testing.protocols import UInt64Backed
+from _algopy_testing.utils import resolve_asset_index
 
 if typing.TYPE_CHECKING:
     import algopy
@@ -65,39 +66,19 @@ class Asset(UInt64Backed):
             "Please mock this method using your python testing framework of choice."
         )
 
-    def __getattr__(self, name: str) -> object:
+    @property
+    def fields(self) -> AssetFields:
         from _algopy_testing.context_helpers import lazy_context
 
-        if int(self.id) not in lazy_context.ledger.asset_data:
-            # check if its not 0 (which means its not
-            # instantiated/opted-in yet, and instantiated directly
-            # without invoking any_asset).
-            if self.id == 0:
-                # Handle dunder methods specially
-                if name.startswith("__") and name.endswith("__"):
-                    return getattr(type(self), name)
-                # For non-dunder attributes, check in __dict__
-                if name in self.__dict__:
-                    return self.__dict__[name]
-                raise AttributeError(
-                    f"'{self.__class__.__name__}' object has no attribute '{name}'"
-                )
-
-            raise ValueError(
-                "`algopy.Asset` is not present in the test context! "
-                "Use `context.add_asset()` or `context.any.asset()` to add the asset to "
-                "your test setup."
-            )
+        return lazy_context.get_asset_data(resolve_asset_index(self.id))
 
-        return_value = lazy_context.get_asset_data(self.id).get(name)
-        if return_value is None:
+    def __getattr__(self, name: str) -> typing.Any:
+        try:
+            return self.fields[name]  # type: ignore[literal-required]
+        except KeyError:
             raise AttributeError(
-                f"The value for '{name}' in the test context is None. "
-                f"Make sure to patch the global field '{name}' using your `AlgopyTestContext` "
-                "instance."
-            )
-
-        return return_value
+                f"'{self.__class__.__name__}' object has no attribute '{name}'"
+            ) from None
 
     def __eq__(self, other: object) -> bool:
         if isinstance(other, Asset):

{algorand_python_testing-0.3.0b8 → algorand_python_testing-0.3.0b10}/src/_algopy_testing/op/global_values.py

@@ -1,6 +1,5 @@
 from __future__ import annotations
 
-import time
 import typing
 from typing import TypedDict, TypeVar
 
@@ -38,7 +37,7 @@ class GlobalFields(TypedDict, total=False):
 class _Global:
     @property
     def _fields(self) -> GlobalFields:
-        return lazy_context.ledger.global_fields
+        return lazy_context.ledger._global_fields
 
     @property
     def current_application_address(self) -> algopy.Account:
@@ -60,8 +59,7 @@ class _Global:
         try:
             return self._fields["latest_timestamp"]
         except KeyError:
-            # TODO: 1.0 Construct this while setting default values rather than here.
-            return UInt64(int(time.time()))
+            return UInt64(lazy_context.active_group._latest_timestamp)
 
     @property
     def group_size(self) -> algopy.UInt64:

{algorand_python_testing-0.3.0b8 → algorand_python_testing-0.3.0b10}/src/_algopy_testing/op/misc.py

@@ -10,7 +10,11 @@ from _algopy_testing.enums import TransactionType
 from _algopy_testing.models import Account, Application, Asset
 from _algopy_testing.primitives.bytes import Bytes
 from _algopy_testing.primitives.uint64 import UInt64
-from _algopy_testing.utils import raise_mocked_function_error
+from _algopy_testing.utils import (
+    raise_mocked_function_error,
+    resolve_app_index,
+    resolve_asset_index,
+)
 
 if typing.TYPE_CHECKING:
     import algopy
@@ -23,10 +27,7 @@ def err() -> None:
 def _get_app(app: algopy.Application | algopy.UInt64 | int) -> Application:
     if isinstance(app, Application):
         return app
-    if app >= 1001:
-        return lazy_context.ledger.get_app(app)
-    txn = lazy_context.active_group.active_txn
-    return txn.apps(app)
+    return lazy_context.ledger.get_app(resolve_app_index(app))
 
 
 def _get_account(acc: algopy.Account | algopy.UInt64 | int) -> Account:
@@ -39,10 +40,7 @@ def _get_account(acc: algopy.Account | algopy.UInt64 | int) -> Account:
 def _get_asset(asset: algopy.Asset | algopy.UInt64 | int) -> Asset:
     if isinstance(asset, Asset):
         return asset
-    if asset >= 1001:
-        return lazy_context.ledger.get_asset(asset)
-    txn = lazy_context.active_group.active_txn
-    return txn.assets(asset)
+    return lazy_context.ledger.get_asset(resolve_asset_index(asset))
 
 
 def _get_bytes(b: algopy.Bytes | bytes) -> bytes:

{algorand_python_testing-0.3.0b8 → algorand_python_testing-0.3.0b10}/src/_algopy_testing/utils.py

@@ -29,6 +29,28 @@ if TYPE_CHECKING:
     from _algopy_testing.op.global_values import GlobalFields
 
 
+def resolve_app_index(app_id_or_index: algopy.UInt64 | int) -> int:
+    from _algopy_testing.context_helpers import lazy_context
+
+    if app_id_or_index >= 1001:
+        app_id = app_id_or_index
+    else:
+        txn = lazy_context.active_group.active_txn
+        app_id = txn.apps(app_id_or_index).id
+    return int(app_id)
+
+
+def resolve_asset_index(asset_id_or_index: algopy.UInt64 | int) -> int:
+    from _algopy_testing.context_helpers import lazy_context
+
+    if asset_id_or_index >= 1001:
+        asset_id = asset_id_or_index
+    else:
+        txn = lazy_context.active_group.active_txn
+        asset_id = txn.assets(asset_id_or_index).id
+    return int(asset_id)
+
+
 def generate_random_int(min_value: int, max_value: int) -> int:
     return secrets.randbelow(max_value - min_value + 1) + min_value
 

{algorand_python_testing-0.3.0b8 → algorand_python_testing-0.3.0b10}/src/_algopy_testing/value_generators/avm.py

@@ -98,7 +98,7 @@ class AVMValueGenerator:
         new_account_address = address or algosdk.account.generate_account()[1]
         new_account = algopy.Account(new_account_address)
         # defaultdict of account_data ensures we get a new initialized account
-        account_data = lazy_context.ledger.account_data[new_account_address]
+        account_data = lazy_context.get_account_data(new_account_address)
         # update so defaults are preserved
         account_data.fields.update(account_fields)
         # can set these since it is a new account
@@ -112,14 +112,14 @@ class AVMValueGenerator:
         r"""Generate and add a new asset with a unique ID."""
         import algopy
 
-        if asset_id and asset_id in lazy_context.ledger.asset_data:
+        if asset_id and asset_id in lazy_context.ledger._asset_data:
             raise ValueError("Asset with such ID already exists in testing context!")
 
         for key in asset_fields:
             if key not in AssetFields.__annotations__:
                 raise AttributeError(f"Invalid field '{key}' for Asset")
 
-        new_asset = algopy.Asset(asset_id or lazy_context.ledger.get_next_asset_id())
+        new_asset = algopy.Asset(asset_id or lazy_context.ledger._get_next_asset_id())
         default_asset_fields = {
             "total": lazy_context.any.uint64(),
             "decimals": lazy_context.any.uint64(1, 6),
@@ -135,7 +135,7 @@ class AVMValueGenerator:
             "reserve": algopy.Account(algosdk.constants.ZERO_ADDRESS),
         }
         merged_fields = dict(ChainMap(asset_fields, default_asset_fields))  # type: ignore[arg-type]
-        lazy_context.ledger.asset_data[int(new_asset.id)] = AssetFields(**merged_fields)  # type: ignore[typeddict-item]
+        lazy_context.ledger._asset_data[int(new_asset.id)] = AssetFields(**merged_fields)  # type: ignore[typeddict-item]
         return new_asset
 
     def application(  # type: ignore[misc]
@@ -147,9 +147,9 @@ class AVMValueGenerator:
     ) -> algopy.Application:
         r"""Generate and add a new application with a unique ID."""
 
-        new_app_id = id if id is not None else lazy_context.ledger.get_next_app_id()
+        new_app_id = id if id is not None else lazy_context.ledger._get_next_app_id()
 
-        if new_app_id in lazy_context.ledger.app_data:
+        if new_app_id in lazy_context.ledger._app_data:
             raise ValueError(
                 f"Application id {new_app_id} has already been configured in test context!"
             )
@@ -184,7 +184,7 @@ class AVMValueGenerator:
                 raise TypeError(f"incorrect type for {field!r}")
             app_fields[field] = value  # type: ignore[literal-required]
 
-        lazy_context.ledger.app_data[new_app_id] = ApplicationContextData(
+        lazy_context.ledger._app_data[new_app_id] = ApplicationContextData(
             fields=app_fields,
             app_id=new_app_id,
             logs=logs or [],