use of tech.pegasys.teku.validator.client.duties.attestations.AggregationDuty in project teku by ConsenSys.
the class AttestationDutySchedulerTest method shouldScheduleAggregationDuties.
@Test
public void shouldScheduleAggregationDuties() {
createDutySchedulerWithRealDuties();
final UInt64 attestationSlot = UInt64.valueOf(13);
final int validator1Index = 5;
final int validator1Committee = 3;
final int validator1CommitteePosition = 9;
// Guaranteed to be an aggregator
final int validator1CommitteeSize = 1;
final int validator2Index = 6;
final int validator2Committee = 4;
final int validator2CommitteePosition = 8;
// Won't be an aggregator
final int validator2CommitteeSize = 1000000000;
final int committeesAtSlot = 15;
final AttesterDuty validator1Duties = new AttesterDuty(VALIDATOR1_KEY, validator1Index, validator1CommitteeSize, validator1Committee, committeesAtSlot, validator1CommitteePosition, attestationSlot);
final AttesterDuty validator2Duties = new AttesterDuty(VALIDATOR2_KEY, validator2Index, validator2CommitteeSize, validator2Committee, committeesAtSlot, validator2CommitteePosition, attestationSlot);
when(validatorApiChannel.getAttestationDuties(eq(ONE), any())).thenReturn(completedFuture(Optional.of(new AttesterDuties(dataStructureUtil.randomBytes32(), List.of(validator1Duties, validator2Duties)))));
final BLSSignature validator1Signature = dataStructureUtil.randomSignature();
when(validator1.getSigner().signAggregationSlot(attestationSlot, fork)).thenReturn(completedFuture(validator1Signature));
when(validator2.getSigner().signAggregationSlot(attestationSlot, fork)).thenReturn(completedFuture(dataStructureUtil.randomSignature()));
final SafeFuture<Optional<AttestationData>> unsignedAttestationFuture = new SafeFuture<>();
final AggregationDuty aggregationDuty = mock(AggregationDuty.class);
final AttestationProductionDuty attestationDuty = mock(AttestationProductionDuty.class);
when(attestationDutyFactory.createProductionDuty(attestationSlot, validator1)).thenReturn(attestationDuty);
when(attestationDutyFactory.createAggregationDuty(attestationSlot, validator1)).thenReturn(aggregationDuty);
when(aggregationDuty.performDuty()).thenReturn(new SafeFuture<>());
when(attestationDuty.addValidator(validator1, validator1Committee, validator1CommitteePosition, validator1Index, validator1CommitteeSize)).thenReturn(unsignedAttestationFuture);
// Load duties
final UInt64 epochStartSlot = spec.computeStartSlotAtEpoch(ONE);
dutyScheduler.onSlot(epochStartSlot);
// Only validator1 should have had an aggregation duty created for it
verify(attestationDutyFactory).createAggregationDuty(attestationSlot, validator1);
// And should have added validator1 to each duty
verify(aggregationDuty).addValidator(validator1, validator1Index, validator1Signature, validator1Committee, unsignedAttestationFuture);
verifyNoMoreInteractions(aggregationDuty);
// Perform the duties
dutyScheduler.onAttestationAggregationDue(attestationSlot);
verify(aggregationDuty).performDuty();
}
Aggregations