Skip to content

Commit a81f87f

Browse files
committed
cast to Err in a couple other places
1 parent dbf7369 commit a81f87f

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
@@ -734,18 +734,14 @@ async def validate_spend_bundle(
734734
# point-of-view of the next block to be farmed. Therefore we pass in the
735735
# current peak's height and timestamp
736736
assert self.peak.timestamp is not None
737-
try:
738-
tl_error: Optional[Err] = check_time_locks(
739-
removal_record_dict,
740-
conds,
741-
self.peak.height,
742-
self.peak.timestamp,
743-
)
744-
except Exception as e:
745-
breakpoint()
746-
737+
tl_error: Optional[Err] = check_time_locks(
738+
removal_record_dict,
739+
conds,
740+
self.peak.height,
741+
self.peak.timestamp,
742+
)
747743
if tl_error is not None:
748-
breakpoint()
744+
tl_error = Err(tl_error)
749745

750746
timelocks: TimelockConditions = compute_assert_height(removal_record_dict, conds)
751747

0 commit comments

Comments
 (0)