92 lines
3.0 KiB
Nim
92 lines
3.0 KiB
Nim
{.used.}
|
|
|
|
import
|
|
unittest2,
|
|
eth/keys,
|
|
../beacon_chain/validators/action_tracker
|
|
|
|
suite "subnet tracker":
|
|
setup:
|
|
let rng = keys.newRng()
|
|
|
|
test "should register stability subnets on attester duties":
|
|
var tracker = ActionTracker.init(rng, false)
|
|
|
|
check:
|
|
tracker.stabilitySubnets(Slot(0)).countOnes() == 0
|
|
tracker.aggregateSubnets(Slot(0)).countOnes() == 0
|
|
|
|
tracker.registerDuty(Slot(0), SubnetId(0), ValidatorIndex(0), true)
|
|
|
|
tracker.updateSlot(Slot(0))
|
|
|
|
check:
|
|
tracker.stabilitySubnets(Slot(0)).countOnes() == 1
|
|
tracker.aggregateSubnets(Slot(0)).countOnes() == 1
|
|
tracker.aggregateSubnets(Slot(1)).countOnes() == 0
|
|
|
|
tracker.registerDuty(Slot(1), SubnetId(1), ValidatorIndex(0), true)
|
|
check:
|
|
tracker.aggregateSubnets(Slot(0)).countOnes() == 2
|
|
tracker.aggregateSubnets(Slot(1)).countOnes() == 1
|
|
tracker.knownValidators.len() == 1
|
|
|
|
tracker.registerDuty(Slot(SUBNET_SUBSCRIPTION_LEAD_TIME_SLOTS), SubnetId(2), ValidatorIndex(0), true)
|
|
check:
|
|
tracker.aggregateSubnets(Slot(0)).countOnes() == 2
|
|
tracker.aggregateSubnets(Slot(1)).countOnes() == 2
|
|
tracker.knownValidators.len() == 1
|
|
|
|
tracker.updateSlot(
|
|
Slot(SUBNET_SUBSCRIPTION_LEAD_TIME_SLOTS) + KNOWN_VALIDATOR_DECAY + 1)
|
|
|
|
check:
|
|
# Validator should be "forgotten" if they don't register for duty
|
|
tracker.knownValidators.len() == 0
|
|
|
|
# Guaranteed to expire
|
|
tracker.updateSlot(
|
|
(Epoch(EPOCHS_PER_RANDOM_SUBNET_SUBSCRIPTION * 2) + 1).start_slot() +
|
|
SUBNET_SUBSCRIPTION_LEAD_TIME_SLOTS + KNOWN_VALIDATOR_DECAY + 1)
|
|
|
|
|
|
check:
|
|
tracker.stabilitySubnets(Slot(0)).countOnes() == 0
|
|
tracker.aggregateSubnets(Slot(0)).countOnes() == 0
|
|
|
|
test "should register sync committee duties":
|
|
var
|
|
tracker = ActionTracker.init(rng, false)
|
|
pk0 = ValidatorPubKey.fromHex("0xb4102a1f6c80e5c596a974ebd930c9f809c3587dc4d1d3634b77ff66db71e376dbc86c3252c6d140ce031f4ec6167798").get()
|
|
pk1 = ValidatorPubKey.fromHex("0xa00d2954717425ce047e0928e5f4ec7c0e3bbe1058db511303fd659770ddace686ee2e22ac180422e516f4c503eb2228").get()
|
|
|
|
check:
|
|
not tracker.hasSyncDuty(pk0, Epoch(1024))
|
|
|
|
tracker.lastSyncUpdate = Opt.some(SyncCommitteePeriod(42))
|
|
tracker.registerSyncDuty(pk0, Epoch(1024))
|
|
check:
|
|
tracker.lastSyncUpdate.isNone()
|
|
not tracker.hasSyncDuty(pk0, Epoch(1024))
|
|
not tracker.hasSyncDuty(pk1, Epoch(1023))
|
|
tracker.hasSyncDuty(pk0, Epoch(1023))
|
|
|
|
tracker.registerSyncDuty(pk0, Epoch(1022))
|
|
|
|
check: # Should not overwrite longer duties
|
|
tracker.hasSyncDuty(pk0, Epoch(1023))
|
|
|
|
tracker.registerSyncDuty(pk0, Epoch(1025))
|
|
check: # Should update existing duties
|
|
tracker.hasSyncDuty(pk0, Epoch(1024))
|
|
|
|
tracker.updateSlot(Epoch(1025).start_slot)
|
|
|
|
check: # should prune old duties on updateSlot
|
|
not tracker.hasSyncDuty(pk0, Epoch(1024))
|
|
|
|
tracker.registerSyncDuty(pk0, Epoch(1025))
|
|
|
|
check: # should not add old duties
|
|
not tracker.hasSyncDuty(pk0, Epoch(1024))
|