mirror of
https://github.com/sigp/lighthouse.git
synced 2026-03-15 02:42:38 +00:00
0f57fc9d8e
Closes: - https://github.com/sigp/lighthouse/issues/1914 Sign attestations prior to checking them against the slashing protection DB. This allows us to avoid the sequential DB checks which are observed in traces here: - https://github.com/sigp/lighthouse/pull/8508#discussion_r2576686107 Co-Authored-By: Jimmy Chen <jchen.tc@gmail.com> Co-Authored-By: Michael Sproul <michael@sigmaprime.io> Co-Authored-By: Michael Sproul <michaelsproul@users.noreply.github.com>
88 lines
2.6 KiB
Rust
88 lines
2.6 KiB
Rust
//! Tests that stress the concurrency safety of the slashing protection DB.
|
|
#![cfg(test)]
|
|
|
|
use crate::attestation_tests::attestation_data_builder;
|
|
use crate::block_tests::block;
|
|
use crate::test_utils::*;
|
|
use crate::*;
|
|
use rayon::prelude::*;
|
|
use tempfile::tempdir;
|
|
|
|
#[test]
|
|
fn block_same_slot() {
|
|
let dir = tempdir().unwrap();
|
|
let slashing_db_file = dir.path().join("slashing_protection.sqlite");
|
|
let slashing_db = SlashingDatabase::create(&slashing_db_file).unwrap();
|
|
|
|
let pk = pubkey(0);
|
|
|
|
slashing_db.register_validator(pk).unwrap();
|
|
|
|
// A stream of blocks all with the same slot.
|
|
let num_blocks = 10;
|
|
let results = (0..num_blocks)
|
|
.into_par_iter()
|
|
.map(|_| slashing_db.check_and_insert_block_proposal(&pk, &block(1), DEFAULT_DOMAIN))
|
|
.collect::<Vec<_>>();
|
|
|
|
let num_successes = results.iter().filter(|res| res.is_ok()).count();
|
|
assert_eq!(num_successes, 1);
|
|
}
|
|
|
|
#[test]
|
|
fn attestation_same_target() {
|
|
let dir = tempdir().unwrap();
|
|
let slashing_db_file = dir.path().join("slashing_protection.sqlite");
|
|
let slashing_db = SlashingDatabase::create(&slashing_db_file).unwrap();
|
|
|
|
let pk = pubkey(0);
|
|
|
|
slashing_db.register_validator(pk).unwrap();
|
|
|
|
// A stream of attestations all with the same target.
|
|
let num_attestations = 10;
|
|
let results = (0..num_attestations)
|
|
.into_par_iter()
|
|
.map(|i| {
|
|
slashing_db.with_transaction(|txn| {
|
|
slashing_db.check_and_insert_attestation(
|
|
&pk,
|
|
&attestation_data_builder(i, num_attestations),
|
|
DEFAULT_DOMAIN,
|
|
txn,
|
|
)
|
|
})
|
|
})
|
|
.collect::<Vec<_>>();
|
|
|
|
let num_successes = results.iter().filter(|res| res.is_ok()).count();
|
|
assert_eq!(num_successes, 1);
|
|
}
|
|
|
|
#[test]
|
|
fn attestation_surround_fest() {
|
|
let dir = tempdir().unwrap();
|
|
let slashing_db_file = dir.path().join("slashing_protection.sqlite");
|
|
let slashing_db = SlashingDatabase::create(&slashing_db_file).unwrap();
|
|
|
|
let pk = pubkey(0);
|
|
|
|
slashing_db.register_validator(pk).unwrap();
|
|
|
|
// A stream of attestations that all surround each other.
|
|
let num_attestations = 10;
|
|
|
|
let results = (0..num_attestations)
|
|
.into_par_iter()
|
|
.map(|i| {
|
|
let att = attestation_data_builder(i, 2 * num_attestations - i);
|
|
slashing_db.with_transaction(|txn| {
|
|
slashing_db.check_and_insert_attestation(&pk, &att, DEFAULT_DOMAIN, txn)
|
|
})
|
|
})
|
|
.collect::<Vec<_>>();
|
|
|
|
let num_successes = results.iter().filter(|res| res.is_ok()).count();
|
|
assert_eq!(num_successes, 1);
|
|
}
|