Skip to content

Commit eaa47bb

Browse files
committed
cast to Err in a couple other places
1 parent c17f3d7 commit eaa47bb

File tree

2 files changed

+16
-19
lines changed

2 files changed

+16
-19
lines changed

chia/_tests/core/mempool/test_mempool_manager.py

Lines changed: 9 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -431,15 +431,16 @@ def test_conditions(
431431
conds: SpendBundleConditions,
432432
expected: Optional[Err],
433433
) -> None:
434-
assert (
435-
check_time_locks(
436-
dict(self.REMOVALS),
437-
conds,
438-
self.PREV_BLOCK_HEIGHT,
439-
self.PREV_BLOCK_TIMESTAMP,
440-
)
441-
== expected
434+
res = check_time_locks(
435+
dict(self.REMOVALS),
436+
conds,
437+
self.PREV_BLOCK_HEIGHT,
438+
self.PREV_BLOCK_TIMESTAMP,
442439
)
440+
if res is not None:
441+
# TODO: remove when Rust errors and Python Errors are the same
442+
res = Err(res)
443+
assert res == expected
443444

444445

445446
def expect(

chia/full_node/mempool_manager.py

Lines changed: 7 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -719,18 +719,14 @@ async def validate_spend_bundle(
719719
# point-of-view of the next block to be farmed. Therefore we pass in the
720720
# current peak's height and timestamp
721721
assert self.peak.timestamp is not None
722-
try:
723-
tl_error: Optional[Err] = check_time_locks(
724-
removal_record_dict,
725-
conds,
726-
self.peak.height,
727-
self.peak.timestamp,
728-
)
729-
except Exception as e:
730-
breakpoint()
731-
722+
tl_error: Optional[Err] = check_time_locks(
723+
removal_record_dict,
724+
conds,
725+
self.peak.height,
726+
self.peak.timestamp,
727+
)
732728
if tl_error is not None:
733-
breakpoint()
729+
tl_error = Err(tl_error)
734730

735731
timelocks: TimelockConditions = compute_assert_height(removal_record_dict, conds)
736732

0 commit comments

Comments
 (0)