@@ -420,7 +420,7 @@ async def test_v2_partial_proofs_with_existing_sp(
420
420
harvester_peer = await get_harvester_peer (farmer )
421
421
await farmer_api .partial_proofs (partial_proofs , harvester_peer )
422
422
423
- # should store 2 pending requests (one per quality )
423
+ # should store 2 pending requests (one per partial proof )
424
424
assert len (farmer .pending_solver_requests ) == 2
425
425
assert sp_hash in farmer .cache_add_time
426
426
@@ -434,7 +434,6 @@ async def test_solution_response_handler(
434
434
farmer = farmer_api .farmer
435
435
436
436
# set up a pending request
437
- quality = bytes32 (b"3" * 32 )
438
437
sp_hash = bytes32 (b"1" * 32 )
439
438
challenge_hash = bytes32 (b"2" * 32 )
440
439
@@ -453,13 +452,15 @@ async def test_solution_response_handler(
453
452
harvester_peer = await get_harvester_peer (farmer )
454
453
455
454
# manually add pending request
456
- farmer .pending_solver_requests [quality ] = {
455
+ farmer .pending_solver_requests [partial_proofs . partial_proofs [ 0 ] ] = {
457
456
"proof_data" : partial_proofs ,
458
457
"peer" : harvester_peer ,
459
458
}
460
459
461
460
# 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
+ )
463
464
solver_peer = Mock ()
464
465
solver_peer .peer_node_id = "solver_peer"
465
466
@@ -476,7 +477,7 @@ async def test_solution_response_handler(
476
477
assert original_peer == harvester_peer
477
478
478
479
# 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
480
481
481
482
482
483
@pytest .mark .anyio
@@ -510,7 +511,6 @@ async def test_solution_response_empty_proof(
510
511
farmer = farmer_api .farmer
511
512
512
513
# set up a pending request
513
- quality = bytes32 (b"3" * 32 )
514
514
sp_hash = bytes32 (b"1" * 32 )
515
515
challenge_hash = bytes32 (b"2" * 32 )
516
516
@@ -530,16 +530,16 @@ async def test_solution_response_empty_proof(
530
530
harvester_peer .peer_node_id = "harvester_peer"
531
531
532
532
# 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 ] ,
535
535
"peer" : harvester_peer ,
536
536
}
537
537
538
538
# get real solver peer connection
539
539
solver_peer = await get_solver_peer (farmer )
540
540
541
541
# 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"" )
543
543
544
544
with unittest .mock .patch .object (farmer_api , "new_proof_of_space" , new_callable = AsyncMock ) as mock_new_proof :
545
545
await farmer_api .solution_response (solution_response , solver_peer )
@@ -548,7 +548,7 @@ async def test_solution_response_empty_proof(
548
548
mock_new_proof .assert_not_called ()
549
549
550
550
# 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
552
552
553
553
554
554
@pytest .mark .anyio
@@ -594,5 +594,4 @@ async def test_v2_partial_proofs_solver_exception(
594
594
await farmer_api .partial_proofs (partial_proofs , harvester_peer )
595
595
596
596
# 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
0 commit comments