ink_e2e/
lib.rs

1// Copyright (C) Use Ink (UK) Ltd.
2//
3// Licensed under the Apache License, Version 2.0 (the "License");
4// you may not use this file except in compliance with the License.
5// You may obtain a copy of the License at
6//
7//     http://www.apache.org/licenses/LICENSE-2.0
8//
9// Unless required by applicable law or agreed to in writing, software
10// distributed under the License is distributed on an "AS IS" BASIS,
11// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12// See the License for the specific language governing permissions and
13// limitations under the License.
14
15//! Module for the logic behind ink!'s End-to-End testing framework.
16
17#![doc(
18    html_logo_url = "https://use.ink/img/crate-docs/logo.png",
19    html_favicon_url = "https://use.ink/crate-docs/favicon.png"
20)]
21
22mod backend;
23mod backend_calls;
24mod builders;
25mod client_utils;
26mod contract_build;
27mod contract_results;
28mod error;
29pub mod events;
30mod node_proc;
31#[cfg(feature = "sandbox")]
32mod sandbox_client;
33mod subxt_client;
34mod xts;
35
36pub use crate::contract_build::build_root_and_contract_dependencies;
37pub use backend::{
38    BuilderClient,
39    ChainBackend,
40    ContractsBackend,
41    E2EBackend,
42};
43pub use backend_calls::{
44    CallBuilder,
45    InstantiateBuilder,
46};
47pub use client_utils::ContractsRegistry;
48pub use contract_results::{
49    CallDryRunResult,
50    CallResult,
51    InstantiateDryRunResult,
52    InstantiationResult,
53    UploadResult,
54};
55pub use ink_e2e_macro::test;
56pub use node_proc::{
57    TestNodeProcess,
58    TestNodeProcessBuilder,
59};
60pub use pallet_revive::evm::CallTrace;
61#[cfg(feature = "sandbox")]
62pub use sandbox_client::{
63    Client as SandboxClient,
64    preset,
65};
66pub use sp_keyring::Sr25519Keyring;
67pub use subxt::{
68    self,
69    backend::rpc::RpcClient,
70};
71pub use subxt_client::{
72    CallBuilderFinal,
73    Client,
74    Error,
75};
76pub use subxt_signer::{
77    self,
78    sr25519::{
79        self,
80        Keypair,
81        dev::*,
82    },
83};
84pub use tokio;
85pub use tracing;
86pub use tracing_subscriber;
87
88#[cfg(feature = "sandbox")]
89pub use ink_sandbox::DefaultSandbox;
90
91use ink::codegen::ContractCallBuilder;
92use ink_env::{
93    ContractEnv,
94    Environment,
95    call::FromAddr,
96};
97use ink_primitives::{
98    Address,
99    DepositLimit,
100    H256,
101};
102pub use sp_weights::Weight;
103use std::{
104    cell::RefCell,
105    sync::Once,
106};
107use xts::ReviveApi;
108
109use ink_primitives::types::AccountIdMapper;
110pub use subxt::PolkadotConfig;
111
112/// We use this to only initialize `env_logger` once.
113pub static INIT: Once = Once::new();
114
115// We save the name of the currently executing test here as a mean
116// of prefixing log entries to make it easier pinning them to tests.
117thread_local! {
118    /// This prefix will be used for log output. It is set by each
119    /// `#[ink_e2e::test]` with the function name as String.
120    /// This way it is possible to distinguish the lines in stdout
121    /// and stderr, to still know which line belongs to which test.
122    pub static LOG_PREFIX: RefCell<String> = RefCell::new(String::from("no prefix set"));
123}
124
125/// Returns the name of the test which is currently executed.
126pub fn log_prefix() -> String {
127    LOG_PREFIX.with(|log_prefix| log_prefix.borrow().clone())
128}
129
130/// Writes `msg` to stdout.
131pub fn log_info(msg: &str) {
132    tracing::info!("[{}] {}", log_prefix(), msg);
133}
134
135/// Writes `msg` to stderr.
136pub fn log_error(msg: &str) {
137    tracing::error!("[{}] {}", log_prefix(), msg);
138}
139
140/// Get an ink! [`ink_primitives::AccountId`] for a given keyring account.
141pub fn account_id(account: Sr25519Keyring) -> ink_primitives::AccountId {
142    ink_primitives::AccountId::try_from(account.to_account_id().as_ref())
143        .expect("account keyring has a valid account id")
144}
145
146/// Returns the [`ink::Address`] for a given keyring account.
147///
148/// # Developer Note
149///
150/// We take the `AccountId` and return only the first twenty bytes, this
151/// is what `pallet-revive` does as well.
152pub fn address<E: Environment>(account: Sr25519Keyring) -> Address {
153    AccountIdMapper::to_address(account.to_account_id().as_ref())
154}
155
156/// Creates a call builder for `Contract`, based on an account id.
157pub fn create_call_builder<Contract>(
158    acc_id: Address,
159) -> <Contract as ContractCallBuilder>::Type<ink::env::DefaultAbi>
160where
161    <Contract as ContractEnv>::Env: Environment,
162    Contract: ContractCallBuilder + ContractEnv,
163    <Contract as ContractCallBuilder>::Type<ink::env::DefaultAbi>: FromAddr,
164{
165    <<Contract as ContractCallBuilder>::Type<ink::env::DefaultAbi> as FromAddr>::from_addr(
166        acc_id,
167    )
168}
169
170/// Creates a call builder for `Contract` for the specified ABI, based on an account id.
171pub fn create_call_builder_abi<Contract, Abi>(
172    acc_id: Address,
173) -> <Contract as ContractCallBuilder>::Type<Abi>
174where
175    <Contract as ContractEnv>::Env: Environment,
176    Contract: ContractCallBuilder + ContractEnv,
177    <Contract as ContractCallBuilder>::Type<Abi>: FromAddr,
178{
179    <<Contract as ContractCallBuilder>::Type<Abi> as FromAddr>::from_addr(acc_id)
180}
181
182/// Transforms `Option<<E as Environment>::Balance>>` into `DepositLimit`.
183///
184/// This function must only be used for dry-runs, a `None` will
185/// become an unrestricted deposit limit (`DepositLimit::UnsafeOnlyForDryRun`).
186fn balance_to_deposit_limit_dry_run<E: Environment>(
187    b: Option<<E as Environment>::Balance>,
188) -> DepositLimit<<E as Environment>::Balance> {
189    match b {
190        Some(v) => DepositLimit::Balance(v),
191        None => DepositLimit::UnsafeOnlyForDryRun,
192    }
193}
194
195/// Transforms `Option<<E as Environment>::Balance>>` into `DepositLimit`.
196/// This function must be used for submitting extrinsics on-chain.
197///
198/// Panics if `limit` is `None`. Make sure to execute a dry-run
199/// beforehand and use the `storage_deposit_limit` result of it here.
200fn balance_to_deposit_limit<E: Environment>(
201    limit: Option<<E as Environment>::Balance>,
202) -> DepositLimit<<E as Environment>::Balance> {
203    match limit {
204        Some(val) => DepositLimit::Balance(val),
205        None => panic!("Deposit limit must be specified for on-chain submissions."),
206    }
207}
208
209/// Transforms `DepositLimit<<E as Environment>::Balance>` into `<E as
210/// Environment>::Balance>`.
211///
212/// Panics if `limit` is unrestricted (`DepositLimit::UnsafeOnlyForDryRun`).
213fn deposit_limit_to_balance<E: Environment>(
214    limit: DepositLimit<<E as Environment>::Balance>,
215) -> <E as Environment>::Balance {
216    match limit {
217        DepositLimit::Balance(val) => val,
218        DepositLimit::UnsafeOnlyForDryRun => {
219            panic!("Unrestricted deposit limit not allowed for balance conversion!")
220        }
221    }
222}
ink_e2e/lib.rs

ink_e2e/
lib.rs
