mirror of
https://github.com/sigp/lighthouse.git
synced 2026-03-06 18:21:45 +00:00
f3fd1f210b
There are certain crates which we re-export within `types` which creates a fragmented DevEx, where there are various ways to import the same crates.
```rust
// consensus/types/src/lib.rs
pub use bls::{
AggregatePublicKey, AggregateSignature, Error as BlsError, Keypair, PUBLIC_KEY_BYTES_LEN,
PublicKey, PublicKeyBytes, SIGNATURE_BYTES_LEN, SecretKey, Signature, SignatureBytes,
get_withdrawal_credentials,
};
pub use context_deserialize::{ContextDeserialize, context_deserialize};
pub use fixed_bytes::FixedBytesExtended;
pub use milhouse::{self, List, Vector};
pub use ssz_types::{BitList, BitVector, FixedVector, VariableList, typenum, typenum::Unsigned};
pub use superstruct::superstruct;
```
This PR removes these re-exports and makes it explicit that these types are imported from a non-`consensus/types` crate.
Co-Authored-By: Mac L <mjladson@pm.me>
70 lines
2.4 KiB
Rust
70 lines
2.4 KiB
Rust
//! Tests for upgrading a previous version of the database to the latest schema.
|
|
use fixed_bytes::FixedBytesExtended;
|
|
use slashing_protection::{NotSafe, SlashingDatabase};
|
|
use std::collections::HashMap;
|
|
use std::fs;
|
|
use std::path::{Path, PathBuf};
|
|
use tempfile::tempdir;
|
|
use types::Hash256;
|
|
|
|
fn test_data_dir() -> PathBuf {
|
|
Path::new(&std::env::var("CARGO_MANIFEST_DIR").unwrap()).join("migration-tests")
|
|
}
|
|
|
|
/// Copy `filename` from the test data dir to the temporary `dest` for testing.
|
|
fn make_copy(filename: &str, dest: &Path) -> PathBuf {
|
|
let source_file = test_data_dir().join(filename);
|
|
let dest_file = dest.join(filename);
|
|
fs::copy(source_file, &dest_file).unwrap();
|
|
dest_file
|
|
}
|
|
|
|
#[test]
|
|
fn add_enabled_column() {
|
|
let tmp = tempdir().unwrap();
|
|
|
|
let path = make_copy("v0_no_enabled_column.sqlite", tmp.path());
|
|
let num_expected_validators = 5;
|
|
|
|
// Database should open without errors, indicating successfull application of migrations.
|
|
// The input file has no `enabled` column, which should get added when opening it here.
|
|
let db = SlashingDatabase::open(&path).unwrap();
|
|
|
|
// Check that exporting an interchange file lists all the validators.
|
|
let interchange = db.export_all_interchange_info(Hash256::zero()).unwrap();
|
|
assert_eq!(interchange.data.len(), num_expected_validators);
|
|
|
|
db.with_transaction(|txn| {
|
|
// Check that all the validators are enabled and unique.
|
|
let uniq_validator_ids = interchange
|
|
.data
|
|
.iter()
|
|
.map(|data| {
|
|
let (validator_id, enabled) = db
|
|
.get_validator_id_with_status(txn, &data.pubkey)
|
|
.unwrap()
|
|
.unwrap();
|
|
assert!(enabled);
|
|
(validator_id, data.pubkey)
|
|
})
|
|
.collect::<HashMap<_, _>>();
|
|
|
|
assert_eq!(uniq_validator_ids.len(), num_expected_validators);
|
|
|
|
// Check that we can disable them all.
|
|
for (&validator_id, pubkey) in &uniq_validator_ids {
|
|
db.update_validator_status(txn, validator_id, false)
|
|
.unwrap();
|
|
let (loaded_id, enabled) = db
|
|
.get_validator_id_with_status(txn, pubkey)
|
|
.unwrap()
|
|
.unwrap();
|
|
assert_eq!(validator_id, loaded_id);
|
|
assert!(!enabled);
|
|
}
|
|
|
|
Ok::<_, NotSafe>(())
|
|
})
|
|
.unwrap();
|
|
}
|