Skip to content

Commit 92faa7f

Browse files
committed
Fix cancellation and failure tests
1 parent 976532d commit 92faa7f

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/test_activity.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -154,7 +154,7 @@ async def test_manual_cancellation(client: Client):
154154
with pytest.raises(ActivityFailedError) as err:
155155
await activity_handle.result()
156156
assert isinstance(err.value.cause, CancelledError)
157-
assert str(err.value.cause) == "Test cancellation"
157+
assert list(err.value.cause.details) == ["Test cancellation"]
158158

159159
desc = await activity_handle.describe()
160160
assert desc.status == ActivityExecutionStatus.CANCELED
@@ -189,7 +189,7 @@ async def test_manual_failure(client: Client):
189189
activity_id=activity_id,
190190
run_id=activity_handle.run_id,
191191
)
192-
await async_activity_handle.fail(Exception("Test failure"))
192+
await async_activity_handle.fail(ApplicationError("Test failure"))
193193
with pytest.raises(ActivityFailedError) as err:
194194
await activity_handle.result()
195195
assert isinstance(err.value.cause, ApplicationError)

0 commit comments

Comments
 (0)