genlayer-test 0.7.0__py3-none-any.whl → 0.9.0__py3-none-any.whl

{genlayer_test-0.7.0.dist-info → genlayer_test-0.9.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: genlayer-test
-Version: 0.7.0
+Version: 0.9.0
 Summary: GenLayer Testing Suite
 Author: GenLayer
 License-Expression: MIT
@@ -15,7 +15,7 @@ Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: pytest
 Requires-Dist: setuptools>=77.0
-Requires-Dist: genlayer-py==0.8.1
+Requires-Dist: genlayer-py==0.9.0
 Requires-Dist: colorama>=0.4.6
 Requires-Dist: pyyaml
 Requires-Dist: python-dotenv
@@ -140,6 +140,7 @@ networks:
   custom_network:  # Custom network configuration
     id: 1234
     url: "http://custom.network:8545"
+    chain_type: "localnet"  # Required for custom networks: localnet, studionet, or testnet_asimov
     accounts:
       - "${CUSTOM_ACCOUNT_1}"
       - "${CUSTOM_ACCOUNT_2}"
@@ -163,10 +164,11 @@ Key configuration sections:
    - Network configurations can include:
      - `url`: The RPC endpoint for the network (optional for pre-configured networks)
      - `id`: Chain ID (optional for pre-configured networks)
+     - `chain_type`: Chain type - one of: `localnet`, `studionet`, or `testnet_asimov` (required for custom networks)
      - `accounts`: List of account private keys (using environment variables)
      - `from`: Specify which account to use as the default for transactions (optional, defaults to first account)
      - `leader_only`: Leader only mode
-   - For custom networks (non-pre-configured), `id`, `url`, and `accounts` are required fields
+   - For custom networks (non-pre-configured), `id`, `url`, `chain_type`, and `accounts` are required fields
 
 **Note on Environment Variables**: When using environment variables in your configuration (e.g., `${ACCOUNT_PRIVATE_KEY_1}`), ensure they are properly set in your `environment` file. If an environment variable is not found, the system will raise a clear error message indicating which variable is missing.
 
@@ -182,6 +184,13 @@ testnet_asimov:
   from: "${ADMIN_KEY}"       # Use ADMIN_KEY as default instead of DEPLOYER_KEY
 ```
 
+**Chain vs Network**: 
+- **Network**: Defines the connection details (URL, accounts, etc.) for a specific environment
+- **Chain**: Defines the genlayer chain type and its associated behaviors (localnet, studionet, or testnet_asimov)
+- Pre-configured networks automatically have the correct chain type set
+- Custom networks must specify the chain type explicitly
+- The `--chain-type` CLI flag can override the chain type for any network, allowing you to test different chain behaviors with the same network configuration
+
 2. **Paths**: Define important directory paths
    - `contracts`: Location of your contract files
    - `artifacts`: Location of your artifacts files (analysis results will be stored here)
@@ -299,6 +308,24 @@ When this flag is enabled, all contracts deployed and all write transactions wil
 
 **Note:** Leader-only mode is only available for studio-based networks (localhost, 127.0.0.1, *.genlayer.com, *.genlayerlabs.com). When enabled on other networks, it will have no effect and a warning will be logged.
 
+12. Override the chain type
+```bash
+$ gltest --chain-type localnet
+$ gltest --chain-type studionet
+$ gltest --chain-type testnet_asimov
+```
+The `--chain-type` flag allows you to override the chain type configured for the network. This is useful when:
+- Testing different chain behaviors without changing network configuration
+- Switching between chain types for testing purposes
+- Using a custom network URL with a specific chain type
+
+Available chain types:
+- `localnet`: Local development chain
+- `studionet`: Studio-based chain
+- `testnet_asimov`: Testnet Asimov chain
+
+The chain type determines various behaviors including RPC endpoints, consensus mechanisms, and available features. When specified, this flag overrides the chain type configured in your network settings.
+
 ## 🚀 Key Features
 
 - **Pytest Integration** – Extends pytest to support intelligent contract testing, making it familiar and easy to adopt.
@@ -308,6 +335,7 @@ When this flag is enabled, all contracts deployed and all write transactions wil
 - **State Injection & Consensus Simulation** – Modify contract states dynamically and simulate consensus scenarios for advanced testing.
 - **Prompt Testing & Statistical Analysis** – Evaluate and statistically test prompts for AI-driven contract execution.
 - **Scalability to Security & Audit Tools** – Designed to extend into security testing and smart contract auditing.
+- **Custom Transaction Context** – Set custom validators with specific LLM providers and models, and configure GenVM datetime for deterministic testing scenarios.
 
 ## 📚 Examples
 
@@ -387,6 +415,7 @@ def test_deployment():
         args=["initial_value"],  # Constructor arguments
         account=get_default_account(),  # Account to deploy from
         consensus_max_rotations=3,  # Optional: max consensus rotations
+        transaction_context=None,  # Optional: custom transaction context
     )
     
     # Contract is now deployed and ready to use
@@ -419,7 +448,9 @@ def test_read_methods():
     contract = factory.deploy()
 
     # Call a read-only method
-    result = contract.get_storage(args=[]).call()
+    result = contract.get_storage(args=[]).call(
+        transaction_context=None,  # Optional: custom transaction context
+    )
     
     # Assert the result matches the initial value
     assert result == "initial_value"
@@ -446,6 +477,7 @@ def test_write_methods():
         consensus_max_rotations=3,  # Optional: max consensus rotations
         wait_interval=1,  # Optional: seconds between status checks
         wait_retries=10,  # Optional: max number of retries
+        transaction_context=None,  # Optional: custom transaction context
     )
     
     # Verify the transaction was successful
@@ -609,6 +641,7 @@ def test_analyze_method():
         config=None,                # Optional: provider-specific config
         plugin=None,                # Optional: plugin name
         plugin_config=None,         # Optional: plugin configuration
+        genvm_datetime="2024-01-15T10:30:00Z",  # Optional: GenVM datetime in ISO format
     )
     
     # Access analysis results
@@ -757,6 +790,214 @@ def test_with_mocked_llm(setup_validators):
 - The `setup_validators` fixture handles the mock setup when provided with a mock_response
 - Mocking is particularly useful for CI/CD pipelines where deterministic results are required
 
+### Custom Transaction Context
+
+The GenLayer Testing Suite allows you to customize the transaction execution environment by providing a `transaction_context` parameter with custom validators and GenVM datetime settings.
+
+#### Using Transaction Context
+
+Set custom validators and GenVM datetime for deterministic testing:
+
+```python
+from gltest import get_contract_factory, get_validator_factory
+
+def test_with_custom_transaction_context():
+    factory = get_contract_factory("MyContract")
+    validator_factory = get_validator_factory()
+    
+    # Create custom validators
+    validators = validator_factory.batch_create_validators(
+        count=3,
+        stake=10,
+        provider="openai",
+        model="gpt-4o",
+        config={"temperature": 0.7, "max_tokens": 1000},
+        plugin="openai-compatible",
+        plugin_config={"api_key_env_var": "OPENAI_API_KEY"}
+    )
+    
+    # Create transaction context with custom validators and datetime
+    transaction_context = {
+        "validators": [v.to_dict() for v in validators],
+        "genvm_datetime": "2024-03-15T14:30:00Z"  # ISO format datetime
+    }
+    
+    # Deploy with custom context
+    contract = factory.deploy(
+        args=["initial_value"],
+        transaction_context=transaction_context
+    )
+    
+    # Call methods with custom context
+    result = contract.read_method().call(
+        transaction_context=transaction_context
+    )
+    
+    # Write operations with custom context
+    tx_receipt = contract.write_method(args=["value"]).transact(
+        transaction_context=transaction_context
+    )
+```
+
+#### Mock Validators with Transaction Context
+
+Combine mock validators with custom datetime for fully deterministic tests:
+
+```python
+def test_with_mocked_context():
+    factory = get_contract_factory("LLMContract")
+    validator_factory = get_validator_factory()
+    
+    # Define mock LLM responses
+    mock_response = {
+        "nondet_exec_prompt": {
+            "analyze this": "positive sentiment"
+        },
+        "eq_principle_prompt_comparative": {
+            "values match": True
+        }
+    }
+    
+    # Create mock validators
+    mock_validators = validator_factory.batch_create_mock_validators(
+        count=5,
+        mock_llm_response=mock_response
+    )
+    
+    # Set up deterministic context
+    transaction_context = {
+        "validators": [v.to_dict() for v in mock_validators],
+        "genvm_datetime": "2024-01-01T00:00:00Z"  # Fixed datetime for reproducibility
+    }
+    
+    # Deploy and test with deterministic context
+    contract = factory.deploy(transaction_context=transaction_context)
+    
+    # All operations will use the same mocked validators and datetime
+    result = contract.analyze_text(args=["analyze this"]).transact(
+        transaction_context=transaction_context
+    )
+    # Result will consistently return "positive sentiment"
+```
+
+### Custom Validators
+
+The GenLayer Testing Suite includes a `get_validator_factory()` function that allows you to create custom validators with specific configurations for testing different LLM providers and consensus scenarios.
+
+#### Creating Custom Validators
+
+```python
+from gltest import get_validator_factory
+
+def test_with_custom_validators():
+    factory = get_validator_factory()
+    
+    # Create validators with different LLM providers
+    openai_validator = factory.create_validator(
+        stake=10,
+        provider="openai",
+        model="gpt-4o",
+        config={"temperature": 0.8, "max_tokens": 2000},
+        plugin="openai-compatible",
+        plugin_config={"api_key_env_var": "OPENAI_API_KEY"}
+    )
+    
+    ollama_validator = factory.create_validator(
+        stake=8,
+        provider="ollama",
+        model="mistral",
+        config={"temperature": 0.5},
+        plugin="ollama",
+        plugin_config={"api_url": "http://localhost:11434"}
+    )
+    
+    # Use validators in your tests
+    validators = [openai_validator, ollama_validator]
+    # Configure your test environment with these validators
+```
+
+#### Batch Creation
+
+Create multiple validators with the same configuration:
+
+```python
+def test_batch_validators():
+    factory = get_validator_factory()
+    
+    # Create 5 validators with identical configuration
+    validators = factory.batch_create_validators(
+        count=5,
+        stake=8,
+        provider="openai",
+        model="gpt-4o",
+        config={"temperature": 0.7, "max_tokens": 1000},
+        plugin="openai-compatible",
+        plugin_config={"api_key_env_var": "OPENAI_API_KEY"}
+    )
+```
+
+#### Mock Validators
+
+For deterministic testing, create mock validators that return predefined responses:
+
+```python
+def test_with_mock_validators():
+    factory = get_validator_factory()
+    
+    # Define mock responses
+    mock_response = {
+        "nondet_exec_prompt": {
+            "What is 2+2?": "4",
+            "Explain quantum physics": "It's complicated"
+        },
+        "eq_principle_prompt_comparative": {
+            "values must match": True
+        },
+        "eq_principle_prompt_non_comparative": {
+            "Is this valid?": True
+        }
+    }
+    
+    # Create a single mock validator
+    mock_validator = factory.create_mock_validator(mock_response)
+    
+    # Create multiple mock validators
+    mock_validators = factory.batch_create_mock_validators(
+        count=5,
+        mock_llm_response=mock_response
+    )
+```
+
+#### Validator Methods
+
+Each validator object provides useful methods:
+- `to_dict()`: Convert validator to dictionary format for API calls
+- `clone()`: Create an identical copy of the validator
+- `batch_clone(count)`: Create multiple identical copies
+
+Example:
+```python
+def test_validator_cloning():
+    factory = get_validator_factory()
+    
+    # Create a base validator
+    base_validator = factory.create_validator(
+        stake=10,
+        provider="openai",
+        model="gpt-4o",
+        config={"temperature": 0.7},
+        plugin="openai-compatible",
+        plugin_config={"api_key_env_var": "OPENAI_API_KEY"}
+    )
+    
+    # Clone it to create identical validators
+    cloned = base_validator.clone()
+    multiple_clones = base_validator.batch_clone(3)
+    
+    # Convert to dictionary for API usage
+    validator_dict = base_validator.to_dict()
+```
+
 ## 📝 Best Practices
 
 1. **Test Organization**

genlayer_test-0.9.0.dist-info/RECORD

@@ -0,0 +1,38 @@
+genlayer_test-0.9.0.dist-info/licenses/LICENSE,sha256=che_H4vE0QUx3HvWrAa1_jDEVInift0U6VO15-QqEls,1064
+gltest/__init__.py,sha256=49112x2CLdYwvCbBZ1laJmMk0NQ7S3u5YUbxPefqhrk,454
+gltest/accounts.py,sha256=HUmWguJMolggQaZNRPw-LGlRlQCjLLdUanKRowMv6pI,812
+gltest/assertions.py,sha256=0dEk0VxcHK4I7GZPHxJmz-2jaA60V499gOSR74rZbfM,1748
+gltest/clients.py,sha256=1dX6wmG3QCevQRLbSaFlHymZSb-sJ5aYwet1IoX2nbA,1554
+gltest/exceptions.py,sha256=deJPmrTe5gF33qkkKF2IVJY7lc_knI7Ql3N7jZ8aLZs,510
+gltest/fixtures.py,sha256=EJXmqcC3LD03v07mepacFl58lAdhbLj6bP5rtALYISk,2507
+gltest/logging.py,sha256=jAkHsuMm-AEx1Xu1srU6W-0YzTwXJB48mCK-OVzAiN4,342
+gltest/types.py,sha256=H32fHrU5aFMaPHXgEWcHAmLWOZ9pBFVp8PK_ncpVOgM,940
+gltest/utils.py,sha256=-gHhjrS7i_GhDG3sKOq2qsTtYBt4HHgXHEXh-3RB_rI,573
+gltest/artifacts/__init__.py,sha256=qTt3TE19gVNWnQLUlt5aDe4nNvJ2YJ1jzDkMmYIsCG0,194
+gltest/artifacts/contract.py,sha256=KChpmfjZod_0dVB8y-dvWz6IVm7QlIJsgG2ArtvVDaU,6457
+gltest/contracts/__init__.py,sha256=A9bvEtYOoqoHS8TLlFBfmNOnfwdsJPEf-AZuikagCHM,166
+gltest/contracts/contract.py,sha256=jLF_ojSM6IIbdGO2_DhsO79r2wZ2Z8eBAJRrZk2qVTI,7748
+gltest/contracts/contract_factory.py,sha256=PpWh4mKf1hDMv_yms5lwFV_EoXxXiuNfdXbwD74hbAU,8929
+gltest/contracts/contract_functions.py,sha256=W6Dpw1z-n9EeJxlNtIbeVnzpv4BPABhtgmgjnS8-P0w,2573
+gltest/contracts/method_stats.py,sha256=zWWjvf7K5VC4yrHpDIR717VF7LYp1RaZ1Hr_RZvWxJA,5150
+gltest/contracts/stats_collector.py,sha256=iwsnoYo5aZbI4SVMH7dR-5CQgkglExXfsvaUDpwcdss,9286
+gltest/contracts/utils.py,sha256=TTXgcXn9BuRIlKJrjwmU7R3l1IgXsXk2luM-r3lfbbg,296
+gltest/helpers/__init__.py,sha256=I7HiTu_H7_hP65zY6Wl02r-5eAMr2eZvqBVmusuQLX4,180
+gltest/helpers/fixture_snapshot.py,sha256=bMgvlEVQBGIQzj7NOyosXWlphI1H2C1o75Zo0C-kGfQ,1931
+gltest/helpers/take_snapshot.py,sha256=-QkaBvFG4ZsNKv_nCSEsy5Ze1pICOHxVhReSeQmZUlY,1276
+gltest/validators/__init__.py,sha256=AXboXORF5a8MVtG7jWMT1fJcwGXNzcX6txXQstwX2EU,152
+gltest/validators/validator_factory.py,sha256=fpb-YyAKuWo4-pXBjrZ_TApYLsm6HHa6kGpbFByRucs,3886
+gltest_cli/logging.py,sha256=WXVhfq9vT6FtV_jxDqGEGia1ZWSIUKAfmWRnZd_gWQk,1266
+gltest_cli/main.py,sha256=Ti2-0Ev1x5_cM0D1UKqdgaDt80CDHEQGtdRne2qLm4M,53
+gltest_cli/config/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+gltest_cli/config/constants.py,sha256=llQD_yiJDwpohMBWufBKcmY8BXLkBeLe5eTj0QUOxSU,603
+gltest_cli/config/general.py,sha256=ezpoGsT8grO9zQH6RugV14b1GzeFt-htYToHQBJhNvY,186
+gltest_cli/config/plugin.py,sha256=o-tfm0KmbMe4iVLWGWUws5Wg2pwOEnfnASQv4ovXEXQ,7109
+gltest_cli/config/pytest_context.py,sha256=Ze8JSkrwMTCE8jIhpzU_71CEXg92SiEPvSgNTp-gbS4,243
+gltest_cli/config/types.py,sha256=oKRRFvCQESt3oca6T79RATHl9BtLNrVCqGc9tB5hovY,10083
+gltest_cli/config/user.py,sha256=jtqhEkp2pEh67Pk4xcshXNcApeCBRjLesZgpqJQuCYg,13625
+genlayer_test-0.9.0.dist-info/METADATA,sha256=09bCBWnQ0MR5FYhKFqZWgx7-nThb5RgR8xfFhl0nxMs,40707
+genlayer_test-0.9.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+genlayer_test-0.9.0.dist-info/entry_points.txt,sha256=RWPcSArBpz_G4BYioh5L8Q8hyClRbSgzLimjcWMp-BQ,94
+genlayer_test-0.9.0.dist-info/top_level.txt,sha256=GSdrnQbiLcZssmtCpbDgBTygsc8Bt_TPeYjwm0FmpdA,18
+genlayer_test-0.9.0.dist-info/RECORD,,

{genlayer_test-0.7.0.dist-info → genlayer_test-0.9.0.dist-info}/top_level.txt

@@ -1,3 +1,2 @@
 gltest
 gltest_cli
-tests

gltest/__init__.py

@@ -8,6 +8,8 @@ from gltest.clients import (
     get_gl_client,
 )
 from gltest.contracts import get_contract_factory
+from gltest.validators import get_validator_factory
+
 
 __all__ = [
     "create_account",
@@ -16,4 +18,5 @@ __all__ = [
     "get_gl_client",
     "get_accounts",
     "get_default_account",
+    "get_validator_factory",
 ]

gltest/types.py

@@ -9,6 +9,17 @@ from genlayer_py.types import (
 from typing import List, TypedDict, Dict, Any
 
 
+class MockedLLMResponse(TypedDict):
+    """Maps prompts to responses"""
+
+    # Prompt -> raw JSON string response
+    nondet_exec_prompt: Dict[str, str]
+
+    # Principle -> expected boolean
+    eq_principle_prompt_comparative: Dict[str, bool]
+    eq_principle_prompt_non_comparative: Dict[str, bool]
+
+
 class ValidatorConfig(TypedDict):
     """Validator information."""
 
@@ -20,7 +31,7 @@ class ValidatorConfig(TypedDict):
 
 
 class TransactionContext(TypedDict, total=False):
-    """Context for consensus operations."""
+    """Context for transaction operations."""
 
     validators: List[ValidatorConfig]  # List to create virtual validators
     genvm_datetime: str  # ISO format datetime string

gltest/validators/__init__.py

@@ -0,0 +1,3 @@
+from .validator_factory import ValidatorFactory, Validator, get_validator_factory
+
+__all__ = ["ValidatorFactory", "Validator", "get_validator_factory"]

gltest/validators/validator_factory.py

@@ -0,0 +1,136 @@
+from gltest.types import MockedLLMResponse
+from dataclasses import dataclass
+from typing import Dict, Any, Optional, List
+from copy import deepcopy
+
+
+@dataclass
+class Validator:
+    stake: int
+    provider: str
+    model: str
+    config: Dict[str, Any]
+    plugin: str
+    plugin_config: Dict[str, Any]
+
+    # Mock configuration
+    mock_enabled: bool
+    mock_llm_response: Optional[MockedLLMResponse]
+
+    def to_dict(self) -> Dict[str, Any]:
+        normal_config = {
+            "stake": self.stake,
+            "provider": self.provider,
+            "model": self.model,
+            "config": deepcopy(self.config),
+            "plugin": self.plugin,
+            "plugin_config": deepcopy(self.plugin_config),
+        }
+        if not self.mock_enabled:
+            return normal_config
+
+        mock = self.mock_llm_response or {}
+        mock_config = {
+            "response": mock.get("nondet_exec_prompt", {}),
+            "eq_principle_prompt_comparative": mock.get(
+                "eq_principle_prompt_comparative", {}
+            ),
+            "eq_principle_prompt_non_comparative": mock.get(
+                "eq_principle_prompt_non_comparative", {}
+            ),
+        }
+        return {
+            **normal_config,
+            "plugin_config": {
+                **self.plugin_config,
+                "mock_response": mock_config,
+            },
+        }
+
+    def batch_clone(self, count: int) -> List["Validator"]:
+        return [self.clone() for _ in range(count)]
+
+    def clone(self) -> "Validator":
+        return Validator(
+            stake=self.stake,
+            provider=self.provider,
+            model=self.model,
+            config=deepcopy(self.config),
+            plugin=self.plugin,
+            plugin_config=deepcopy(self.plugin_config),
+            mock_enabled=self.mock_enabled,
+            mock_llm_response=deepcopy(self.mock_llm_response),
+        )
+
+
+class ValidatorFactory:
+    def __init__(self):
+        pass
+
+    def create_validator(
+        self,
+        stake: int,
+        provider: str,
+        model: str,
+        config: Dict[str, Any],
+        plugin: str,
+        plugin_config: Dict[str, Any],
+    ) -> Validator:
+        return Validator(
+            stake=stake,
+            provider=provider,
+            model=model,
+            config=deepcopy(config),
+            plugin=plugin,
+            plugin_config=deepcopy(plugin_config),
+            mock_enabled=False,
+            mock_llm_response=None,
+        )
+
+    def batch_create_validators(
+        self,
+        count: int,
+        stake: int,
+        provider: str,
+        model: str,
+        config: Dict[str, Any],
+        plugin: str,
+        plugin_config: Dict[str, Any],
+    ) -> List[Validator]:
+        return [
+            self.create_validator(
+                stake=stake,
+                provider=provider,
+                model=model,
+                config=config,
+                plugin=plugin,
+                plugin_config=plugin_config,
+            )
+            for _ in range(count)
+        ]
+
+    def create_mock_validator(self, mock_llm_response: MockedLLMResponse) -> Validator:
+        return Validator(
+            stake=8,
+            provider="openai",
+            model="gpt-4o",
+            config={"temperature": 0.75, "max_tokens": 500},
+            plugin="openai-compatible",
+            plugin_config={
+                "api_key_env_var": "OPENAIKEY",
+                "api_url": "https://api.openai.com",
+            },
+            mock_enabled=True,
+            mock_llm_response=deepcopy(mock_llm_response),
+        )
+
+    def batch_create_mock_validators(
+        self,
+        count: int,
+        mock_llm_response: MockedLLMResponse,
+    ) -> List[Validator]:
+        return [self.create_mock_validator(mock_llm_response) for _ in range(count)]
+
+
+def get_validator_factory() -> ValidatorFactory:
+    return ValidatorFactory()

gltest_cli/config/constants.py

@@ -5,8 +5,15 @@ from pathlib import Path
 GLTEST_CONFIG_FILE = "gltest.config.yaml"
 DEFAULT_NETWORK = "localnet"
 PRECONFIGURED_NETWORKS = ["localnet", "studionet", "testnet_asimov"]
+CHAINS = ["localnet", "studionet", "testnet_asimov"]
 DEFAULT_RPC_URL = SIMULATOR_JSON_RPC_URL
 DEFAULT_ENVIRONMENT = ".env"
 DEFAULT_CONTRACTS_DIR = Path("contracts")
 DEFAULT_ARTIFACTS_DIR = Path("artifacts")
 DEFAULT_NETWORK_ID = 61999
+
+# Defaults per network
+DEFAULT_WAIT_INTERVAL = 3000
+DEFAULT_WAIT_RETRIES = 50
+DEFAULT_TEST_WITH_MOCKS = False
+DEFAULT_LEADER_ONLY = False

gltest_cli/config/plugin.py

@@ -12,6 +12,13 @@ from gltest_cli.config.general import (
 )
 from gltest_cli.config.types import PluginConfig
 from gltest_cli.config.pytest_context import _pytest_context
+from gltest_cli.config.constants import (
+    DEFAULT_WAIT_INTERVAL,
+    DEFAULT_WAIT_RETRIES,
+    DEFAULT_TEST_WITH_MOCKS,
+    DEFAULT_LEADER_ONLY,
+    CHAINS,
+)
 
 
 def pytest_addoption(parser):
@@ -33,14 +40,14 @@ def pytest_addoption(parser):
     group.addoption(
         "--default-wait-interval",
         action="store",
-        default=3000,
+        default=DEFAULT_WAIT_INTERVAL,
         help="Default interval (ms) between transaction receipt checks",
     )
 
     group.addoption(
         "--default-wait-retries",
         action="store",
-        default=50,
+        default=DEFAULT_WAIT_RETRIES,
         help="Default number of retries for transaction receipt checks",
     )
 
@@ -61,17 +68,24 @@ def pytest_addoption(parser):
     group.addoption(
         "--test-with-mocks",
         action="store_true",
-        default=False,
+        default=DEFAULT_TEST_WITH_MOCKS,
         help="Test with mocks",
     )
 
     group.addoption(
         "--leader-only",
         action="store_true",
-        default=False,
+        default=DEFAULT_LEADER_ONLY,
         help="Run contracts in leader-only mode",
     )
 
+    group.addoption(
+        "--chain-type",
+        action="store",
+        default=None,
+        help=f"Chain type (possible values: {', '.join(CHAINS)})",
+    )
+
 
 def pytest_configure(config):
     try:
@@ -108,6 +122,7 @@ def pytest_configure(config):
         network = config.getoption("--network")
         test_with_mocks = config.getoption("--test-with-mocks")
         leader_only = config.getoption("--leader-only")
+        chain_type = config.getoption("--chain-type")
 
         plugin_config = PluginConfig()
         plugin_config.contracts_dir = (
@@ -122,6 +137,7 @@ def pytest_configure(config):
         plugin_config.network_name = network
         plugin_config.test_with_mocks = test_with_mocks
         plugin_config.leader_only = leader_only
+        plugin_config.chain_type = chain_type
 
         general_config.plugin_config = plugin_config
     except Exception as e:
@@ -148,6 +164,8 @@ def pytest_sessionstart(session):
         # Show available networks including preconfigured ones
         all_networks = general_config.get_networks_keys()
         logger.info(f"  Available networks: {all_networks}")
+        logger.info(f"  Selected chain type: {general_config.get_chain_type()}")
+        logger.info(f"  Available chains: {', '.join(CHAINS)}")
         logger.info(f"  Contracts directory: {general_config.get_contracts_dir()}")
         logger.info(f"  Artifacts directory: {general_config.get_artifacts_dir()}")
         logger.info(f"  Environment: {general_config.user_config.environment}")