Skip to content

Commit

Permalink
unified syntax - ReturnsHandledOrError rename
Browse files Browse the repository at this point in the history
  • Loading branch information
andrei-marinica committed Oct 10, 2024
1 parent da9a657 commit 7e9d1e9
Show file tree
Hide file tree
Showing 3 changed files with 21 additions and 21 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -311,7 +311,7 @@ fn st_blackbox_returns_result_or_error() {
.new_address(ST_ADDRESS)
.tx_hash([33u8; 32])
.returns(
ReturnsResultOrError::new()
ReturnsHandledOrError::new()
.returns(ReturnsNewAddress)
.returns(ReturnsResultAs::<String>::new())
.returns(ReturnsTxHash),
Expand All @@ -331,7 +331,7 @@ fn st_blackbox_returns_result_or_error() {
.to(ST_ADDRESS)
.typed(scenario_tester_proxy::ScenarioTesterProxy)
.sum()
.returns(ReturnsResultOrError::new().returns(ReturnsResultUnmanaged))
.returns(ReturnsHandledOrError::new().returns(ReturnsResultUnmanaged))
.run();
assert_eq!(result, Ok(RustBigUint::from(5u32)));

Expand All @@ -341,7 +341,7 @@ fn st_blackbox_returns_result_or_error() {
.to(ST_ADDRESS)
.typed(scenario_tester_proxy::ScenarioTesterProxy)
.sc_panic()
.returns(ReturnsResultOrError::new())
.returns(ReturnsHandledOrError::new())
.run();

assert_eq!(result, Err(TxResponseStatus::new(4, "sc_panic! example")));
Expand All @@ -353,7 +353,7 @@ fn st_blackbox_returns_result_or_error() {
.to(ST_ADDRESS)
.typed(scenario_tester_proxy::ScenarioTesterProxy)
.add(1u32)
.returns(ReturnsResultOrError::new())
.returns(ReturnsHandledOrError::new())
.run();

assert_eq!(result, Ok(()));
Expand All @@ -365,7 +365,7 @@ fn st_blackbox_returns_result_or_error() {
.to(ST_ADDRESS)
.typed(scenario_tester_proxy::ScenarioTesterProxy)
.sc_panic()
.returns(ReturnsResultOrError::new())
.returns(ReturnsHandledOrError::new())
.run();

assert_eq!(result, Err(TxResponseStatus::new(4, "sc_panic! example")));
Expand Down
4 changes: 2 additions & 2 deletions framework/scenario/src/facade/result_handlers.rs
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@ mod returns_logs;
mod returns_message;
mod returns_new_bech32_address;
mod returns_new_token_identifier;
mod returns_result_or_err;
mod returns_handled_or_err;
mod returns_status;
mod returns_tx_hash;
mod with_tx_raw_response;
Expand All @@ -19,7 +19,7 @@ pub use returns_logs::ReturnsLogs;
pub use returns_message::ReturnsMessage;
pub use returns_new_bech32_address::ReturnsNewBech32Address;
pub use returns_new_token_identifier::ReturnsNewTokenIdentifier;
pub use returns_result_or_err::ReturnsResultOrError;
pub use returns_handled_or_err::ReturnsHandledOrError;
pub use returns_status::ReturnsStatus;
pub use returns_tx_hash::ReturnsTxHash;
pub use with_tx_raw_response::WithRawTxResponse;
Original file line number Diff line number Diff line change
Expand Up @@ -12,57 +12,57 @@ use crate::scenario_model::{CheckValue, TxExpect, TxResponse, TxResponseStatus};

/// Indicates that a `Result` will be returned, either with the handled result,
/// according to the inner result handlers, or with an error in case of a failed transaction.
pub struct ReturnsResultOrError<Env, Original, Ok>
pub struct ReturnsHandledOrError<Env, Original, Ok>
where
Env: TxEnv,
Ok: RHList<Env>,
{
_phantom_env: PhantomData<Env>,
_phantom_original: PhantomData<Original>,
pub ok_t: Ok,
pub nested_handlers: Ok,
}

impl<Env, Original> Default for ReturnsResultOrError<Env, Original, OriginalResultMarker<Original>>
impl<Env, Original> Default for ReturnsHandledOrError<Env, Original, OriginalResultMarker<Original>>
where
Env: TxEnv,
{
fn default() -> Self {
ReturnsResultOrError {
ReturnsHandledOrError {
_phantom_env: PhantomData,
_phantom_original: PhantomData,
ok_t: OriginalResultMarker::new(),
nested_handlers: OriginalResultMarker::new(),
}
}
}

impl<Env, Original> ReturnsResultOrError<Env, Original, OriginalResultMarker<Original>>
impl<Env, Original> ReturnsHandledOrError<Env, Original, OriginalResultMarker<Original>>
where
Env: TxEnv,
{
pub fn new() -> Self {
ReturnsResultOrError::default()
ReturnsHandledOrError::default()
}
}

impl<Env, Original, Ok> ReturnsResultOrError<Env, Original, Ok>
impl<Env, Original, Ok> ReturnsHandledOrError<Env, Original, Ok>
where
Env: TxEnv,
Ok: RHListExec<TxResponse, Env>,
{
pub fn returns<RH>(self, item: RH) -> ReturnsResultOrError<Env, Original, Ok::RetOutput>
pub fn returns<RH>(self, item: RH) -> ReturnsHandledOrError<Env, Original, Ok::RetOutput>
where
RH: RHListItem<Env, Ok::OriginalResult>,
Ok: RHListAppendRet<Env, RH>,
{
ReturnsResultOrError {
ReturnsHandledOrError {
_phantom_env: PhantomData,
_phantom_original: PhantomData,
ok_t: self.ok_t.append_ret(item),
nested_handlers: self.nested_handlers.append_ret(item),
}
}
}

impl<Env, Original, Ok> RHListItem<Env, Original> for ReturnsResultOrError<Env, Original, Ok>
impl<Env, Original, Ok> RHListItem<Env, Original> for ReturnsHandledOrError<Env, Original, Ok>
where
Env: TxEnv,
Ok: RHListExec<TxResponse, Env>,
Expand All @@ -72,7 +72,7 @@ where
}

impl<Env, Original, Ok> RHListItemExec<TxResponse, Env, Original>
for ReturnsResultOrError<Env, Original, Ok>
for ReturnsHandledOrError<Env, Original, Ok>
where
Env: TxEnv<RHExpect = TxExpect>,
Ok: RHListExec<TxResponse, Env>,
Expand All @@ -86,7 +86,7 @@ where

fn item_process_result(self, raw_result: &TxResponse) -> Self::Returns {
if raw_result.tx_error.is_success() {
let tuple_result = self.ok_t.list_process_result(raw_result);
let tuple_result = self.nested_handlers.list_process_result(raw_result);
Ok(tuple_result.flatten_unpack())
} else {
Err(raw_result.tx_error.clone())
Expand Down

0 comments on commit 7e9d1e9

Please sign in to comment.