Convert tributary to use create_db macro (#448)

* chore: convert tributary to use create_db macro

* chore: fix fmt

* chore: break long line
This commit is contained in:
David Bell
2023-11-22 13:17:51 +04:00
committed by GitHub
parent 9ab2a2cfe0
commit 9df8c9476e
6 changed files with 142 additions and 294 deletions

View File

@@ -13,7 +13,7 @@ use sp_runtime::traits::Verify;
use tokio::time::sleep;
use serai_db::{DbTxn, Db, MemDb};
use serai_db::{Db, MemDb, DbTxn};
use processor_messages::{
key_gen::{self, KeyGenId},
@@ -23,7 +23,7 @@ use processor_messages::{
use tributary::{TransactionTrait, Tributary};
use crate::{
tributary::{TributaryDb, Transaction, TributarySpec, scanner::handle_new_blocks},
tributary::{Transaction, TributarySpec, scanner::handle_new_blocks},
tests::{
MemProcessors, LocalP2p,
tributary::{new_keys, new_spec, new_tributaries, run_tributaries, wait_for_tx_inclusion},
@@ -81,8 +81,8 @@ async fn dkg_test() {
key: &Zeroizing<<Ristretto as Ciphersuite>::F>,
spec: &TributarySpec,
tributary: &Tributary<MemDb, Transaction, LocalP2p>,
) -> (TributaryDb<MemDb>, MemProcessors) {
let mut scanner_db = TributaryDb(MemDb::new());
) -> (MemDb, MemProcessors) {
let mut scanner_db = MemDb::new();
let processors = MemProcessors::new();
handle_new_blocks::<_, _, _, _, _, _, LocalP2p>(
&mut scanner_db,
@@ -289,7 +289,7 @@ async fn dkg_test() {
if i != 0 {
scanner_db = &mut local_scanner_db;
}
let mut txn = scanner_db.0.txn();
let mut txn = scanner_db.txn();
let share =
crate::tributary::generated_key_pair::<MemDb>(&mut txn, key, &spec, &key_pair, 0).unwrap();
txn.commit();