genlayer-test 0.1.0b1__tar.gz → 0.1.0b2__tar.gz

{genlayer_test-0.1.0b1 → genlayer_test-0.1.0b2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: genlayer-test
-Version: 0.1.0b1
+Version: 0.1.0b2
 Summary: GenLayer Testing Suite
 Author: GenLayer
 License-Expression: MIT

{genlayer_test-0.1.0b1 → genlayer_test-0.1.0b2}/genlayer_test.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: genlayer-test
-Version: 0.1.0b1
+Version: 0.1.0b2
 Summary: GenLayer Testing Suite
 Author: GenLayer
 License-Expression: MIT

{genlayer_test-0.1.0b1 → genlayer_test-0.1.0b2}/genlayer_test.egg-info/SOURCES.txt

@@ -8,11 +8,14 @@ genlayer_test.egg-info/requires.txt
 genlayer_test.egg-info/top_level.txt
 gltest/__init__.py
 gltest/assertions.py
-gltest/cli.py
 gltest/exceptions.py
+gltest/plugin_config.py
+gltest/plugin_hooks.py
+gltest/types.py
 gltest/artifacts/__init__.py
 gltest/artifacts/contract.py
 gltest/glchain/__init__.py
 gltest/glchain/account.py
 gltest/glchain/client.py
-gltest/glchain/contract.py
+gltest/glchain/contract.py
+gltest_cli/main.py

genlayer_test-0.1.0b2/genlayer_test.egg-info/entry_points.txt

@@ -0,0 +1,5 @@
+[console_scripts]
+gltest = gltest_cli.main:main
+
+[pytest11]
+gltest = gltest.plugin_hooks

{genlayer_test-0.1.0b1 → genlayer_test-0.1.0b2}/genlayer_test.egg-info/top_level.txt

@@ -1,4 +1,5 @@
 build
 dist
 gltest
+gltest_cli
 tests

{genlayer_test-0.1.0b1 → genlayer_test-0.1.0b2}/gltest/artifacts/contract.py

@@ -1,7 +1,7 @@
 import ast
-from pathlib import Path
 from typing import Optional
 from dataclasses import dataclass
+from gltest.plugin_config import get_contracts_dir
 
 
 @dataclass
@@ -14,21 +14,18 @@ class ContractDefinition:
     ast_node: ast.ClassDef
 
 
-def find_contract_definition(
-    contract_name: str, relative_contracts_dir: str = "contracts"
-) -> Optional[ContractDefinition]:
+def find_contract_definition(contract_name: str) -> Optional[ContractDefinition]:
     """
     Search in the contracts directory for a contract definition.
     TODO: Make this more robust to handle imports and other files.
     """
-    # Use provided directory or default to 'contracts' in current working directory
-    contracts_dir = Path.cwd() / relative_contracts_dir
+    contracts_dir = get_contracts_dir()
 
     if not contracts_dir.exists():
         raise FileNotFoundError(f"Contracts directory not found at: {contracts_dir}")
 
     # Search through all .gpy files in the contracts directory
-    for file_path in contracts_dir.glob("*.gpy"):
+    for file_path in contracts_dir.rglob("*.gpy"):
         try:
             # Read the file content
             with open(file_path, "r") as f:

{genlayer_test-0.1.0b1 → genlayer_test-0.1.0b2}/gltest/glchain/contract.py

@@ -3,7 +3,7 @@ from gltest.artifacts import find_contract_definition
 from gltest.assertions import tx_execution_failed
 from gltest.exceptions import DeploymentError
 from .client import get_gl_client
-from genlayer_py.types import CalldataEncodable, GenLayerTransaction
+from gltest.types import CalldataEncodable, GenLayerTransaction, TransactionStatus
 from eth_account.signers.local import LocalAccount
 from typing import List, Any, Type, Optional, Dict, Callable
 import types
@@ -18,36 +18,47 @@ class Contract:
 
     address: str
     account: Optional[LocalAccount] = None
+    _schema: Optional[Dict[str, Any]] = None
 
     @classmethod
-    def from_address_and_schema(
-        cls, address: str, schema: Dict[str, Any]
+    def new(
+        cls,
+        address: str,
+        schema: Dict[str, Any],
+        account: Optional[LocalAccount] = None,
     ) -> "Contract":
         """
         Build the methods from the schema.
         """
         if not isinstance(schema, dict) or "methods" not in schema:
             raise ValueError("Invalid schema: must contain 'methods' field")
+        instance = cls(address=address, _schema=schema, account=account)
+        instance._build_methods_from_schema()
+        return instance
 
-        instance = cls(address=address)
-        for method_name, method_info in schema["methods"].items():
+    def _build_methods_from_schema(self):
+        if self._schema is None:
+            raise ValueError("No schema provided")
+        for method_name, method_info in self._schema["methods"].items():
             if not isinstance(method_info, dict) or "readonly" not in method_info:
                 raise ValueError(
-                    f"Invalid method info for {method_name}: must contain 'readonly' field"
+                    f"Invalid method info for '{method_name}': must contain 'readonly' field"
                 )
-            setattr(
-                instance,
-                method_name,
-                types.MethodType(
-                    cls.contract_method_factory(method_name, method_info["readonly"]),
-                    instance,
-                ),
+            method_func = self.contract_method_factory(
+                method_name, method_info["readonly"]
             )
-        return instance
+            bound_method = types.MethodType(method_func, self)
+            setattr(self, method_name, bound_method)
 
     def connect(self, account: LocalAccount) -> "Contract":
-        self.account = account
-        return self
+        """
+        Create a new instance of the contract with the same methods and a different account.
+        """
+        new_contract = self.__class__(
+            address=self.address, account=account, _schema=self._schema
+        )
+        new_contract._build_methods_from_schema()
+        return new_contract
 
     @staticmethod
     def contract_method_factory(method_name: str, read_only: bool) -> Callable:
@@ -78,6 +89,9 @@ class Contract:
             leader_only: bool = False,
             wait_interval: Optional[int] = None,
             wait_retries: Optional[int] = None,
+            wait_transaction_status: TransactionStatus = TransactionStatus.FINALIZED,
+            wait_triggered_transactions: bool = True,
+            wait_triggered_transactions_status: TransactionStatus = TransactionStatus.ACCEPTED,
         ) -> GenLayerTransaction:
             """
             Wrapper to the contract write method.
@@ -98,8 +112,18 @@ class Contract:
             if wait_retries is not None:
                 extra_args["retries"] = wait_retries
             receipt = client.wait_for_transaction_receipt(
-                transaction_hash=tx_hash, status="FINALIZED", **extra_args
+                transaction_hash=tx_hash,
+                status=wait_transaction_status,
+                **extra_args,
             )
+            if wait_triggered_transactions:
+                triggered_transactions = receipt["triggered_transactions"]
+                for triggered_transaction in triggered_transactions:
+                    client.wait_for_transaction_receipt(
+                        transaction_hash=triggered_transaction,
+                        status=wait_triggered_transactions_status,
+                        **extra_args,
+                    )
             return receipt
 
         return read_contract_wrapper if read_only else write_contract_wrapper
@@ -138,6 +162,7 @@ class ContractFactory:
         leader_only: bool = False,
         wait_interval: Optional[int] = None,
         wait_retries: Optional[int] = None,
+        wait_transaction_status: TransactionStatus = TransactionStatus.FINALIZED,
     ) -> Contract:
         """
         Deploy the contract
@@ -157,9 +182,8 @@ class ContractFactory:
             if wait_retries is not None:
                 extra_args["retries"] = wait_retries
             tx_receipt = client.wait_for_transaction_receipt(
-                transaction_hash=tx_hash, status="FINALIZED", **extra_args
+                transaction_hash=tx_hash, status=wait_transaction_status, **extra_args
             )
-
             if (
                 not tx_receipt
                 or "data" not in tx_receipt
@@ -176,8 +200,8 @@ class ContractFactory:
 
             contract_address = tx_receipt["data"]["contract_address"]
             schema = client.get_contract_schema(address=contract_address)
-            return Contract.from_address_and_schema(
-                address=contract_address, schema=schema
+            return Contract.new(
+                address=contract_address, schema=schema, account=account
             )
         except Exception as e:
             raise DeploymentError(

genlayer_test-0.1.0b2/gltest/plugin_config.py

@@ -0,0 +1,12 @@
+from pathlib import Path
+
+_contracts_dir = None
+
+
+def set_contracts_dir(path: Path):
+    global _contracts_dir
+    _contracts_dir = path
+
+
+def get_contracts_dir() -> Path:
+    return Path(_contracts_dir)

genlayer_test-0.1.0b2/gltest/plugin_hooks.py

@@ -0,0 +1,16 @@
+from gltest.plugin_config import set_contracts_dir
+from pathlib import Path
+
+
+def pytest_addoption(parser):
+    parser.addoption(
+        "--contracts-dir",
+        action="store",
+        default="contracts",
+        help="Directory containing contract files",
+    )
+
+
+def pytest_configure(config):
+    contracts_dir = config.getoption("--contracts-dir")
+    set_contracts_dir(Path(contracts_dir))

genlayer_test-0.1.0b2/gltest/types.py

@@ -0,0 +1,7 @@
+# Re-export genlayer-py types
+from genlayer_py.types import (
+    CalldataAddress,
+    GenLayerTransaction,
+    TransactionStatus,
+    CalldataEncodable,
+)

{genlayer_test-0.1.0b1 → genlayer_test-0.1.0b2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "genlayer-test"
-version = "0.1.0b1"
+version = "0.1.0b2"
 description = "GenLayer Testing Suite"
 authors = [
     { name = "GenLayer" }
@@ -30,5 +30,8 @@ classifiers = [
 [tool.setuptools.packages.find]
 where = ["."]
 
+[project.entry-points.pytest11]
+gltest = "gltest.plugin_hooks"
+
 [project.scripts]
-gltest = "gltest.cli:main"
+gltest = "gltest_cli.main:main"

genlayer_test-0.1.0b1/genlayer_test.egg-info/entry_points.txt

@@ -1,2 +0,0 @@
-[console_scripts]
-gltest = gltest.cli:main