Skip to content

Commit 5353a38

Browse files
committed
test and comments
1 parent e6c5e88 commit 5353a38

File tree

3 files changed

+12
-19
lines changed

3 files changed

+12
-19
lines changed

chia/_tests/farmer_harvester/test_farmer_harvester.py

Lines changed: 11 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -420,7 +420,7 @@ async def test_v2_partial_proofs_with_existing_sp(
420420
harvester_peer = await get_harvester_peer(farmer)
421421
await farmer_api.partial_proofs(partial_proofs, harvester_peer)
422422

423-
# should store 2 pending requests (one per quality)
423+
# should store 2 pending requests (one per partial proof)
424424
assert len(farmer.pending_solver_requests) == 2
425425
assert sp_hash in farmer.cache_add_time
426426

@@ -434,7 +434,6 @@ async def test_solution_response_handler(
434434
farmer = farmer_api.farmer
435435

436436
# set up a pending request
437-
quality = bytes32(b"3" * 32)
438437
sp_hash = bytes32(b"1" * 32)
439438
challenge_hash = bytes32(b"2" * 32)
440439

@@ -453,13 +452,15 @@ async def test_solution_response_handler(
453452
harvester_peer = await get_harvester_peer(farmer)
454453

455454
# manually add pending request
456-
farmer.pending_solver_requests[quality] = {
455+
farmer.pending_solver_requests[partial_proofs.partial_proofs[0]] = {
457456
"proof_data": partial_proofs,
458457
"peer": harvester_peer,
459458
}
460459

461460
# create solution response
462-
solution_response = solver_protocol.SolverResponse(partial_proof=quality, proof=b"test_proof_from_solver")
461+
solution_response = solver_protocol.SolverResponse(
462+
partial_proof=partial_proofs.partial_proofs[0], proof=b"test_proof_from_solver"
463+
)
463464
solver_peer = Mock()
464465
solver_peer.peer_node_id = "solver_peer"
465466

@@ -476,7 +477,7 @@ async def test_solution_response_handler(
476477
assert original_peer == harvester_peer
477478

478479
# verify pending request was removed
479-
assert quality not in farmer.pending_solver_requests
480+
assert partial_proofs.partial_proofs[0] not in farmer.pending_solver_requests
480481

481482

482483
@pytest.mark.anyio
@@ -510,7 +511,6 @@ async def test_solution_response_empty_proof(
510511
farmer = farmer_api.farmer
511512

512513
# set up a pending request
513-
quality = bytes32(b"3" * 32)
514514
sp_hash = bytes32(b"1" * 32)
515515
challenge_hash = bytes32(b"2" * 32)
516516

@@ -530,16 +530,16 @@ async def test_solution_response_empty_proof(
530530
harvester_peer.peer_node_id = "harvester_peer"
531531

532532
# manually add pending request
533-
farmer.pending_solver_requests[quality] = {
534-
"proof_data": partial_proofs,
533+
farmer.pending_solver_requests[partial_proofs.partial_proofs[0]] = {
534+
"proof_data": partial_proofs.partial_proofs[0],
535535
"peer": harvester_peer,
536536
}
537537

538538
# get real solver peer connection
539539
solver_peer = await get_solver_peer(farmer)
540540

541541
# create solution response with empty proof
542-
solution_response = solver_protocol.SolverResponse(partial_proof=quality, proof=b"")
542+
solution_response = solver_protocol.SolverResponse(partial_proof=partial_proofs.partial_proofs[0], proof=b"")
543543

544544
with unittest.mock.patch.object(farmer_api, "new_proof_of_space", new_callable=AsyncMock) as mock_new_proof:
545545
await farmer_api.solution_response(solution_response, solver_peer)
@@ -548,7 +548,7 @@ async def test_solution_response_empty_proof(
548548
mock_new_proof.assert_not_called()
549549

550550
# verify pending request was removed (cleanup still happens)
551-
assert quality not in farmer.pending_solver_requests
551+
assert partial_proofs.partial_proofs[0] not in farmer.pending_solver_requests
552552

553553

554554
@pytest.mark.anyio
@@ -594,5 +594,4 @@ async def test_v2_partial_proofs_solver_exception(
594594
await farmer_api.partial_proofs(partial_proofs, harvester_peer)
595595

596596
# verify pending request was cleaned up after exception
597-
quality = bytes32(b"3" * 32)
598-
assert quality not in farmer.pending_solver_requests
597+
assert partial_proofs.partial_proofs[0] not in farmer.pending_solver_requests

chia/farmer/farmer_api.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -503,7 +503,7 @@ async def partial_proofs(self, partial_proof_data: PartialProofsData, peer: WSCh
503503
f"for plot {partial_proof_data.plot_identifier[:10]}... from {peer.peer_node_id}"
504504
)
505505

506-
# Process each quality chain through solver service to get full proofs
506+
# Process each partial proof chain through solver service to get full proofs
507507
for partial_proof in partial_proof_data.partial_proofs:
508508
solver_info = SolverInfo(partial_proof=partial_proof)
509509

chia/solver/solver_rpc_api.py

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -19,18 +19,12 @@ def __init__(self, solver: Solver):
1919

2020
def get_routes(self) -> dict[str, Endpoint]:
2121
return {
22-
"/solve": self.solve,
2322
"/get_state": self.get_state,
2423
}
2524

2625
async def _state_changed(self, change: str, change_data: Optional[dict[str, Any]] = None) -> list[WsRpcMessage]:
2726
return []
2827

29-
async def solve(self, request: dict[str, Any]) -> EndpointResult:
30-
partial_proof = request["partial_proof"]
31-
proof = self.service.solve(partial_proof=partial_proof)
32-
return {"proof": proof.hex() if proof else None}
33-
3428
async def get_state(self, _: dict[str, Any]) -> EndpointResult:
3529
return {
3630
"started": self.service.started,

0 commit comments

Comments
 (0)