Skip to content

Commit 2a43d65

Browse files
authored
feat: allow setting contract err revert msg (#2689)
1 parent 4fb09f3 commit 2a43d65

File tree

2 files changed

+13
-0
lines changed

2 files changed

+13
-0
lines changed

src/ape/exceptions.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -332,6 +332,12 @@ def __init__(
332332
def revert_message(self):
333333
return self.message
334334

335+
@revert_message.setter
336+
def revert_message(self, value):
337+
self.message = value
338+
if args := self.args:
339+
self.args = tuple([value, *args[1:]])
340+
335341
@property
336342
def dev_message(self) -> Optional[str]:
337343
"""

tests/functional/test_exceptions.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -246,6 +246,13 @@ def test_fork_network(self):
246246
"\x1b[0m, or using a network with explorer support."
247247
)
248248

249+
def test_revert_message_setter(self):
250+
err = ContractLogicError(revert_message="test message 1")
251+
err.revert_message = "test message 2"
252+
assert str(err) == "test message 2"
253+
assert err.message == "test message 2"
254+
assert err.revert_message == "test message 2"
255+
249256

250257
class TestUnknownSnapshotError:
251258
def test_bytes(self):

0 commit comments

Comments
 (0)