-
Notifications
You must be signed in to change notification settings - Fork 25
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
feat(test): unit test external call mock #423
base: main
Are you sure you want to change the base?
Conversation
…ontract-deployment
✅ Deploy Preview for contracts-stylus ready!
To edit notification comments on pull requests, go to your Netlify site configuration. |
contracts/src/utils/context.rs
Outdated
#[cfg(test)] | ||
pub fn msg_sender() -> Address { | ||
motsu::prelude::Context::current() | ||
.get_msg_sender() | ||
.expect("msg_sender should be set") | ||
} | ||
|
||
/// Returns the address of the message sender. | ||
#[cfg(not(test))] | ||
pub fn msg_sender() -> Address { | ||
msg::sender() | ||
} |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Here is an example how msg_sender
mock can be declared. This design is not perfect since motsu users should implement this mock for their own library manually or through macro.
Calling stylus sdk api msg::sender()
won't work inside a test without mock due to this limitation.
contracts/src/token/erc721/mod.rs
Outdated
#[motsu::test] | ||
fn on_erc721_received( | ||
erc721: Contract<Erc721>, | ||
receiver: Contract<Erc721ReceiverMock>, | ||
) { | ||
let alice = Account::random(); | ||
let token_id = random_token_id(); | ||
erc721 | ||
.sender(alice) | ||
._safe_mint(receiver.address(), token_id, vec![0, 1, 2, 3].into()) | ||
.unwrap(); | ||
|
||
let received_token_id = receiver.sender(alice).received_token_id(); | ||
|
||
assert_eq!(received_token_id, token_id); | ||
} |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Overall the test case for Erc721Receiver
can look like this
…eta.1 # Conflicts: # contracts/src/token/erc20/extensions/permit.rs # contracts/src/utils/math/storage.rs # lib/e2e/src/lib.rs
This change should be common with #441 |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Imo the API is very good, especially as it conforms to existing conventions for writing code that mocks senders, values etc.
I think the builder pattern is the right choice here, as we might need to extend the functionality at a later point (e.g. to add .with_block_timestamp
, but there may be others, who knows).
Good job! 🔥
Once clippy warnings are addressed I'll give this a final review, as the PR is too cluttered with warnings atm
…eta.1 # Conflicts: # Cargo.toml # contracts/src/utils/math/storage.rs
…ontract-deployment # Conflicts: # contracts/src/token/erc721/mod.rs # lib/motsu/src/shims.rs
…ntract-deployment-v0.7.0-beta # Conflicts: # contracts/src/token/erc721/mod.rs
…eta.1 # Conflicts: # contracts/src/access/control.rs # contracts/src/finance/vesting_wallet.rs # contracts/src/token/erc1155/mod.rs # contracts/src/token/erc1155/receiver.rs # contracts/src/token/erc20/extensions/capped.rs # contracts/src/token/erc20/extensions/metadata.rs # contracts/src/token/erc721/extensions/enumerable.rs # contracts/src/token/erc721/mod.rs # contracts/src/utils/metadata.rs # contracts/src/utils/pausable.rs
…ntract-deployment-v0.7.0-beta # Conflicts: # contracts/src/token/erc1155/extensions/uri_storage.rs # contracts/src/token/erc721/mod.rs # lib/motsu-proc/src/test.rs
35b1c5f
to
f27aed9
Compare
…to unit-tests/multiple-contract-deployment
…ontract-deployment # Conflicts: # Cargo.lock # contracts/src/finance/vesting_wallet.rs # contracts/src/token/erc1155/mod.rs # lib/motsu-proc/src/test.rs # lib/motsu/Cargo.toml # lib/motsu/src/context.rs # lib/motsu/src/lib.rs # lib/motsu/src/prelude.rs # lib/motsu/src/shims.rs
keccak-const.workspace = true | ||
openzeppelin-stylus-proc.workspace = true | ||
|
||
[dev-dependencies] | ||
alloy-primitives = { workspace = true, features = ["arbitrary"] } | ||
motsu.workspace = true | ||
rand.workspace = true | ||
stylus-sdk.workspace = true |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
stylus-sdk is listed both in dependencies
and in dev-dependencies
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Since OL agreed to have a default feature for ENABLING hostio caching, now we should write stylus-sdk = { version = "0.7.0", default-features = false }
on the workspace level, and stylus-sdk = { workspace = true, default-features = true }
for [dependencies]
. Because we need this caching for wasm compiled contracts, but can not run motsu tests with it.
And use it without default features for unit tests [dev-dependencies]
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I would better add a feature for disabling caching (not enabling). It would be more simple
@@ -41,6 +41,8 @@ | |||
//! accounts that have been granted it. We recommend using | |||
//! `AccessControlDefaultAdminRules` to enforce additional security measures for | |||
//! this role. | |||
use alloc::vec::Vec; |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Why are we adding this import?
pub use token::IErc20; | ||
mod token { | ||
#![allow(missing_docs)] | ||
#![cfg_attr(coverage_nightly, coverage(off))] | ||
stylus_sdk::stylus_proc::sol_interface! { | ||
/// Interface of the ERC-20 token. | ||
interface IErc20 { | ||
function balanceOf(address account) external view returns (uint256); | ||
} | ||
sol_interface! { | ||
/// Interface of the ERC-20 token. | ||
#[allow(missing_docs)] | ||
interface IErc20 { | ||
function balanceOf(address account) external view returns (uint256); |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Won't coverage be reduced by including this?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Yeap. It will be reduced, currently the simplest way to compile it.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
It doesn't compile if you do:
pub use token::IErc20;
mod token {
#![allow(missing_docs)]
#![cfg_attr(coverage_nightly, coverage(off))]
stylus_sdk::prelude::sol_interface! {
/// Interface of the ERC-20 token.
interface IErc20 {
function balanceOf(address account) external view returns (uint256);
}
}
}
? 😮
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
IErc20
now made private. So it will de-sugar to struct IErc20
, instead of pub struct IErc20
. And visibility level will be just module wise
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
May be try to change it again in stylus sdk.. pr
This pr contains corresponding improvements for motsu and two new test cases
on_erc721_received
andping_pong_works
.Other test cases are commented. And should be refactored once the final design will be approved.
Resolves OpenZeppelin/stylus-test-helpers#8 OpenZeppelin/stylus-test-helpers#12
PR Checklist