mirror of
https://github.com/pezkuwichain/pezkuwi-subxt.git
synced 2026-04-29 06:47:57 +00:00
Bastian Köcher
7b56ab15b4
* Run cargo fmt on the whole code base * Second run * Add CI check * Fix compilation * More unnecessary braces * Handle weights * Use --all * Use correct attributes... * Fix UI tests * AHHHHHHHHH * 🤦 * Docs * Fix compilation * 🤷 * Please stop * 🤦 x 2 * More * make rustfmt.toml consistent with polkadot Co-authored-by: André Silva <andrerfosilva@gmail.com>
89 lines
2.5 KiB
Rust
89 lines
2.5 KiB
Rust
// This file is part of Substrate.
|
|
|
|
// Copyright (C) 2017-2021 Parity Technologies (UK) Ltd.
|
|
// SPDX-License-Identifier: GPL-3.0-or-later WITH Classpath-exception-2.0
|
|
|
|
// This program is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
|
|
// This program is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU General Public License for more details.
|
|
|
|
// You should have received a copy of the GNU General Public License
|
|
// along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
//! Test utils
|
|
|
|
use crate::{ChangeSet, CommitSet, DBValue, MetaDb, NodeDb};
|
|
use sp_core::H256;
|
|
use std::collections::HashMap;
|
|
|
|
#[derive(Default, Debug, Clone, PartialEq, Eq)]
|
|
pub struct TestDb {
|
|
pub data: HashMap<H256, DBValue>,
|
|
pub meta: HashMap<Vec<u8>, DBValue>,
|
|
}
|
|
|
|
impl MetaDb for TestDb {
|
|
type Error = ();
|
|
|
|
fn get_meta(&self, key: &[u8]) -> Result<Option<DBValue>, ()> {
|
|
Ok(self.meta.get(key).cloned())
|
|
}
|
|
}
|
|
|
|
impl NodeDb for TestDb {
|
|
type Error = ();
|
|
type Key = H256;
|
|
|
|
fn get(&self, key: &H256) -> Result<Option<DBValue>, ()> {
|
|
Ok(self.data.get(key).cloned())
|
|
}
|
|
}
|
|
|
|
impl TestDb {
|
|
pub fn commit(&mut self, commit: &CommitSet<H256>) {
|
|
self.data.extend(commit.data.inserted.iter().cloned());
|
|
self.meta.extend(commit.meta.inserted.iter().cloned());
|
|
for k in commit.data.deleted.iter() {
|
|
self.data.remove(k);
|
|
}
|
|
self.meta.extend(commit.meta.inserted.iter().cloned());
|
|
for k in commit.meta.deleted.iter() {
|
|
self.meta.remove(k);
|
|
}
|
|
}
|
|
|
|
pub fn data_eq(&self, other: &TestDb) -> bool {
|
|
self.data == other.data
|
|
}
|
|
}
|
|
|
|
pub fn make_changeset(inserted: &[u64], deleted: &[u64]) -> ChangeSet<H256> {
|
|
ChangeSet {
|
|
inserted: inserted
|
|
.iter()
|
|
.map(|v| (H256::from_low_u64_be(*v), H256::from_low_u64_be(*v).as_bytes().to_vec()))
|
|
.collect(),
|
|
deleted: deleted.iter().map(|v| H256::from_low_u64_be(*v)).collect(),
|
|
}
|
|
}
|
|
|
|
pub fn make_commit(inserted: &[u64], deleted: &[u64]) -> CommitSet<H256> {
|
|
CommitSet { data: make_changeset(inserted, deleted), meta: ChangeSet::default() }
|
|
}
|
|
|
|
pub fn make_db(inserted: &[u64]) -> TestDb {
|
|
TestDb {
|
|
data: inserted
|
|
.iter()
|
|
.map(|v| (H256::from_low_u64_be(*v), H256::from_low_u64_be(*v).as_bytes().to_vec()))
|
|
.collect(),
|
|
meta: Default::default(),
|
|
}
|
|
}
|