Skip to content

Commit 907e6d9

Browse files
chore: remove unnecessary anyhow wrapping from tests
1 parent a9481aa commit 907e6d9

8 files changed

Lines changed: 51 additions & 85 deletions

File tree

crates/sdk-core/tests/integ_tests/data_converter_tests.rs

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -194,8 +194,7 @@ impl DataConverterTestWorkflow {
194194
input,
195195
ActivityOptions::start_to_close_timeout(Duration::from_secs(5)),
196196
)
197-
.await
198-
.map_err(|e| anyhow::anyhow!("{e}"))?;
197+
.await?;
199198

200199
Ok(output)
201200
}
@@ -218,8 +217,7 @@ impl DescribeDataConverterWorkflow {
218217
input,
219218
ActivityOptions::start_to_close_timeout(Duration::from_secs(5)),
220219
)
221-
.await
222-
.map_err(|e| anyhow::anyhow!("{e}"))?;
220+
.await?;
223221

224222
Ok(output)
225223
}

crates/sdk-core/tests/integ_tests/worker_tests.rs

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -419,15 +419,13 @@ async fn activity_tasks_from_completion_reserve_slots() {
419419
(),
420420
ActivityOptions::start_to_close_timeout(Duration::from_secs(5)),
421421
)
422-
.await
423-
.map_err(|e| WorkflowTermination::from(anyhow::Error::from(e)))?;
422+
.await?;
424423
ctx.start_activity(
425424
FakeAct::act2,
426425
(),
427426
ActivityOptions::start_to_close_timeout(Duration::from_secs(5)),
428427
)
429-
.await
430-
.map_err(|e| WorkflowTermination::from(anyhow::Error::from(e)))?;
428+
.await?;
431429
ctx.state(|wf| wf.complete_token.cancel());
432430
Ok(())
433431
}

crates/sdk-core/tests/integ_tests/workflow_tests.rs

Lines changed: 6 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -505,8 +505,7 @@ impl SlowCompletesWf {
505505
"hi!".to_string(),
506506
ActivityOptions::start_to_close_timeout(Duration::from_secs(5)),
507507
)
508-
.await
509-
.map_err(|e| anyhow::anyhow!("{e}"))?;
508+
.await?;
510509
ctx.timer(Duration::from_secs(1)).await;
511510
}
512511
Ok(())
@@ -881,8 +880,7 @@ async fn nondeterminism_errors_fail_workflow_when_configured_to(
881880
"hi".to_owned(),
882881
ActivityOptions::start_to_close_timeout(Duration::from_secs(5)),
883882
)
884-
.await
885-
.map_err(|e| anyhow::anyhow!("{e}"))?;
883+
.await?;
886884
Ok(())
887885
}
888886
}
@@ -956,15 +954,13 @@ async fn history_out_of_order_on_restart() {
956954
..Default::default()
957955
},
958956
)
959-
.await
960-
.map_err(|e| anyhow::anyhow!("{e}"))?;
957+
.await?;
961958
ctx.start_activity(
962959
StdActivities::echo,
963960
"hi".to_string(),
964961
ActivityOptions::start_to_close_timeout(Duration::from_secs(5)),
965962
)
966-
.await
967-
.map_err(|e| anyhow::anyhow!("{e}"))?;
963+
.await?;
968964
ctx.state(|wf| wf.hit_sleep.notify_one());
969965
ctx.timer(Duration::from_secs(5)).await;
970966
Ok(())
@@ -987,17 +983,15 @@ async fn history_out_of_order_on_restart() {
987983
..Default::default()
988984
},
989985
)
990-
.await
991-
.map_err(|e| anyhow::anyhow!("{e}"))?;
986+
.await?;
992987
// Timer is added after restarting workflow
993988
ctx.timer(Duration::from_secs(1)).await;
994989
ctx.start_activity(
995990
StdActivities::echo,
996991
"hi".to_string(),
997992
ActivityOptions::start_to_close_timeout(Duration::from_secs(5)),
998993
)
999-
.await
1000-
.map_err(|e| anyhow::anyhow!("{e}"))?;
994+
.await?;
1001995
ctx.timer(Duration::from_secs(2)).await;
1002996
Ok(())
1003997
}

crates/sdk-core/tests/integ_tests/workflow_tests/activities.rs

Lines changed: 15 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -72,8 +72,7 @@ impl OneActivityWorkflow {
7272
input,
7373
ActivityOptions::start_to_close_timeout(Duration::from_secs(5)),
7474
)
75-
.await
76-
.map_err(|e| anyhow!("{e}"))?;
75+
.await?;
7776
Ok(r)
7877
}
7978
}
@@ -92,8 +91,7 @@ impl MultiArgActivityWorkflow {
9291
(input, " world".to_string()),
9392
ActivityOptions::start_to_close_timeout(Duration::from_secs(5)),
9493
)
95-
.await
96-
.map_err(|e| anyhow!("{e}"))?;
94+
.await?;
9795
Ok(r)
9896
}
9997
}
@@ -172,18 +170,19 @@ async fn activity_panics_are_retryable() {
172170
impl ActivityPanicRetryWorkflow {
173171
#[run]
174172
async fn run(ctx: &mut WorkflowContext<Self>) -> WorkflowResult<u32> {
175-
ctx.start_activity(
176-
PanicOnceActivities::panic_once,
177-
(),
178-
ActivityOptions::with_start_to_close_timeout(Duration::from_secs(5))
179-
.retry_policy(RetryPolicy {
180-
maximum_attempts: 2,
181-
..Default::default()
182-
})
183-
.build(),
184-
)
185-
.await
186-
.map_err(|e| anyhow!("{e}").into())
173+
let result = ctx
174+
.start_activity(
175+
PanicOnceActivities::panic_once,
176+
(),
177+
ActivityOptions::with_start_to_close_timeout(Duration::from_secs(5))
178+
.retry_policy(RetryPolicy {
179+
maximum_attempts: 2,
180+
..Default::default()
181+
})
182+
.build(),
183+
)
184+
.await?;
185+
Ok(result)
187186
}
188187
}
189188

crates/sdk-core/tests/integ_tests/workflow_tests/child_workflows.rs

Lines changed: 9 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -93,7 +93,8 @@ impl HappyParent {
9393
)
9494
.await
9595
.expect("Child should start OK");
96-
started.result().await.map_err(|e| anyhow!(e).into())
96+
started.result().await?;
97+
Ok(())
9798
}
9899
}
99100

@@ -811,8 +812,7 @@ impl PassChildWorkflowSummaryToMetadata {
811812
..Default::default()
812813
},
813814
)
814-
.await
815-
.map_err(|e| anyhow!(e))?;
815+
.await?;
816816
Ok(())
817817
}
818818
}
@@ -920,7 +920,7 @@ impl ParentWf {
920920
_ => return Err(anyhow!("Expected start failure").into()),
921921
}
922922
}
923-
let started = start_res.map_err(|e| anyhow!(e))?;
923+
let started = start_res?;
924924
match (expectation, started.result().await) {
925925
(Expectation::Success, Ok(_)) => Ok(()),
926926
(Expectation::Failure, Err(ChildWorkflowExecutionError::Failed(failure))) => {
@@ -1274,13 +1274,9 @@ impl UntypedHappyParent {
12741274
..Default::default()
12751275
},
12761276
)
1277-
.await
1278-
.map_err(|e| anyhow!(e))?;
1279-
started
1280-
.result()
1281-
.await
1282-
.map(|_| ())
1283-
.map_err(|e| anyhow!(e).into())
1277+
.await?;
1278+
started.result().await?;
1279+
Ok(())
12841280
}
12851281
}
12861282

@@ -1416,8 +1412,7 @@ impl ChildSignalSerializationFailParent {
14161412
..Default::default()
14171413
},
14181414
)
1419-
.await
1420-
.map_err(|e| anyhow!(e))?;
1415+
.await?;
14211416

14221417
let signal_result = started
14231418
.signal(UnserializableSignalChild::bad_signal, AlwaysFailsSerialize)
@@ -1484,8 +1479,7 @@ impl UnitChildParentWf {
14841479
..Default::default()
14851480
},
14861481
)
1487-
.await
1488-
.map_err(|e| anyhow!(e))?;
1482+
.await?;
14891483
started.result().await?;
14901484
Ok(())
14911485
}

crates/sdk-core/tests/integ_tests/workflow_tests/determinism.rs

Lines changed: 5 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -57,8 +57,7 @@ impl TimerWfNondeterministic {
5757
(),
5858
ActivityOptions::start_to_close_timeout(Duration::from_secs(5)),
5959
)
60-
.await
61-
.map_err(|e| anyhow::anyhow!("{e}"))?;
60+
.await?;
6261
}
6362
_ => panic!("Ran too many times"),
6463
}
@@ -290,12 +289,10 @@ impl ActivityIdOrTypeChangeWf {
290289
..Default::default()
291290
},
292291
)
293-
.await
294-
.map_err(|e| anyhow::anyhow!("{e}"))?;
292+
.await?;
295293
} else {
296294
ctx.start_local_activity(StdActivities::no_op, (), Default::default())
297-
.await
298-
.map_err(|e| anyhow::anyhow!("{e}"))?;
295+
.await?;
299296
}
300297
} else if id_change {
301298
ctx.start_activity(
@@ -305,16 +302,14 @@ impl ActivityIdOrTypeChangeWf {
305302
.activity_id("I'm bad and wrong!".to_string())
306303
.build(),
307304
)
308-
.await
309-
.map_err(|e| anyhow::anyhow!("{e}"))?;
305+
.await?;
310306
} else {
311307
ctx.start_activity(
312308
StdActivities::no_op,
313309
(),
314310
ActivityOptions::start_to_close_timeout(Duration::from_secs(5)),
315311
)
316-
.await
317-
.map_err(|e| anyhow::anyhow!("{e}"))?;
312+
.await?;
318313
}
319314
Ok(())
320315
}

crates/sdk-core/tests/integ_tests/workflow_tests/local_activities.rs

Lines changed: 11 additions & 22 deletions
Original file line numberDiff line numberDiff line change
@@ -81,8 +81,7 @@ impl OneLocalActivityWf {
8181
"hi!".to_string(),
8282
LocalActivityOptions::default(),
8383
)
84-
.await
85-
.map_err(|e| WorkflowTermination::from(anyhow::Error::from(e)))?;
84+
.await?;
8685
assert!(initial_workflow_time < ctx.workflow_time().unwrap());
8786
Ok(())
8887
}
@@ -1128,8 +1127,7 @@ async fn long_local_activity_with_update(
11281127
Duration::from_secs(6),
11291128
LocalActivityOptions::default(),
11301129
)
1131-
.await
1132-
.map_err(|e| WorkflowTermination::from(anyhow::Error::from(e)))?;
1130+
.await?;
11331131
Ok(ctx.state(|wf| wf.update_counter))
11341132
}
11351133

@@ -1273,8 +1271,7 @@ impl LocalActivityWithSummaryWf {
12731271
..Default::default()
12741272
},
12751273
)
1276-
.await
1277-
.map_err(|e| WorkflowTermination::from(anyhow::Error::from(e)))?;
1274+
.await?;
12781275
Ok(())
12791276
}
12801277
}
@@ -1440,8 +1437,7 @@ async fn local_act_heartbeat(#[case] shutdown_middle: bool) {
14401437
"hi".to_string(),
14411438
LocalActivityOptions::default(),
14421439
)
1443-
.await
1444-
.map_err(|e| WorkflowTermination::from(anyhow::Error::from(e)))?;
1440+
.await?;
14451441
Ok(())
14461442
}
14471443
}
@@ -1662,8 +1658,7 @@ async fn local_act_null_result() {
16621658
#[run(name = DEFAULT_WORKFLOW_TYPE)]
16631659
async fn run(ctx: &mut WorkflowContext<Self>) -> WorkflowResult<()> {
16641660
ctx.start_local_activity(StdActivities::default, (), LocalActivityOptions::default())
1665-
.await
1666-
.map_err(|e| WorkflowTermination::from(anyhow::Error::from(e)))?;
1661+
.await?;
16671662
Ok(())
16681663
}
16691664
}
@@ -1699,8 +1694,7 @@ async fn local_act_command_immediately_follows_la_marker() {
16991694
#[run(name = DEFAULT_WORKFLOW_TYPE)]
17001695
async fn run(ctx: &mut WorkflowContext<Self>) -> WorkflowResult<()> {
17011696
ctx.start_local_activity(StdActivities::default, (), LocalActivityOptions::default())
1702-
.await
1703-
.map_err(|e| WorkflowTermination::from(anyhow::Error::from(e)))?;
1697+
.await?;
17041698
ctx.timer(Duration::from_secs(1)).await;
17051699
Ok(())
17061700
}
@@ -2319,8 +2313,7 @@ async fn resolved_las_not_recorded_if_wft_fails_many_times() {
23192313
..Default::default()
23202314
},
23212315
)
2322-
.await
2323-
.map_err(|e| WorkflowTermination::from(anyhow::Error::from(e)))?;
2316+
.await?;
23242317
panic!()
23252318
}
23262319
}
@@ -2870,11 +2863,9 @@ impl TwoLaWf {
28702863
#[run(name = DEFAULT_WORKFLOW_TYPE)]
28712864
async fn run(ctx: &mut WorkflowContext<Self>) -> WorkflowResult<()> {
28722865
ctx.start_local_activity(StdActivities::default, (), LocalActivityOptions::default())
2873-
.await
2874-
.map_err(|e| WorkflowTermination::from(anyhow::Error::from(e)))?;
2866+
.await?;
28752867
ctx.start_local_activity(StdActivities::default, (), LocalActivityOptions::default())
2876-
.await
2877-
.map_err(|e| WorkflowTermination::from(anyhow::Error::from(e)))?;
2868+
.await?;
28782869
Ok(())
28792870
}
28802871
}
@@ -3007,12 +2998,10 @@ impl LaTimerLaWf {
30072998
#[run(name = DEFAULT_WORKFLOW_TYPE)]
30082999
async fn run(ctx: &mut WorkflowContext<Self>) -> WorkflowResult<()> {
30093000
ctx.start_local_activity(StdActivities::default, (), LocalActivityOptions::default())
3010-
.await
3011-
.map_err(|e| WorkflowTermination::from(anyhow::Error::from(e)))?;
3001+
.await?;
30123002
ctx.timer(Duration::from_secs(5)).await;
30133003
ctx.start_local_activity(StdActivities::default, (), LocalActivityOptions::default())
3014-
.await
3015-
.map_err(|e| WorkflowTermination::from(anyhow::Error::from(e)))?;
3004+
.await?;
30163005
Ok(())
30173006
}
30183007
}

crates/sdk-core/tests/integ_tests/workflow_tests/resets.rs

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -152,8 +152,7 @@ impl ResetRandomseedWf {
152152
"hi!".to_string(),
153153
LocalActivityOptions::default(),
154154
)
155-
.await
156-
.map_err(|e| anyhow::anyhow!("{e}"))?;
155+
.await?;
157156
}
158157
ctx.wait_condition(|s| s.post_fail_received).await;
159158
ctx.state(|wf| wf.notify.notify_one());

0 commit comments

Comments
 (0)