5
5
6
6
import pytest
7
7
from chia_rs import ConsensusConstants
8
- from chia_rs .sized_ints import uint64
9
8
10
9
from chia .protocols .outbound_message import Message
11
10
from chia .protocols .solver_protocol import SolverInfo
@@ -23,7 +22,7 @@ async def test_solver_api_methods(blockchain_constants: ConsensusConstants, tmp_
23
22
solver = solver_service ._node
24
23
solver_api = solver_service ._api
25
24
assert solver_api .ready () is True
26
- test_info = SolverInfo (plot_strength = uint64 ( 1500 ), partial_proof = b"test_partial_proof_42" )
25
+ test_info = SolverInfo (partial_proof = b"test_partial_proof_42" )
27
26
expected_proof = b"test_proof_data_12345"
28
27
with patch .object (solver , "solve" , return_value = expected_proof ):
29
28
api_result = await solver_api .solve (test_info )
@@ -49,7 +48,7 @@ async def test_solver_error_handling(
49
48
pass # expected
50
49
# test solver handles exception in solve method
51
50
solver = solver_service ._node
52
- test_info = SolverInfo (plot_strength = uint64 ( 1000 ), partial_proof = b"test_partial_proof_zeros" )
51
+ test_info = SolverInfo (partial_proof = b"test_partial_proof_zeros" )
53
52
with patch .object (solver , "solve" , side_effect = RuntimeError ("test error" )):
54
53
# solver api should handle exceptions gracefully
55
54
result = await solver_service ._api .solve (test_info )
0 commit comments