mirror of
https://github.com/mimblewimble/grin.git
synced 2025-01-21 11:31:08 +03:00
f43df5f601
* add feature for height locked kernels * add function to compute kernel features appropriate for lock height, and use it * only sign kernel-features relevant fields; refactor Features * simplify invalid kernel logic * remove unused height arg to reward::output and run some rustfmt * replace nested if/else by match
100 lines
3.2 KiB
Rust
100 lines
3.2 KiB
Rust
// Copyright 2018 The Grin Developers
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
use self::chain::{Error, Tip};
|
|
use self::core::core::hash::Hashed;
|
|
use self::core::core::Block;
|
|
use self::core::global::{self, ChainTypes};
|
|
use self::core::libtx;
|
|
use self::core::pow::{self, Difficulty};
|
|
use self::keychain::{ExtKeychain, ExtKeychainPath, Keychain};
|
|
use env_logger;
|
|
use grin_chain as chain;
|
|
use grin_core as core;
|
|
use grin_keychain as keychain;
|
|
use grin_store as store;
|
|
use std::fs;
|
|
use std::sync::Arc;
|
|
|
|
fn clean_output_dir(dir_name: &str) {
|
|
let _ = fs::remove_dir_all(dir_name);
|
|
}
|
|
|
|
fn setup_chain(genesis: &Block, chain_store: Arc<chain::store::ChainStore>) -> Result<(), Error> {
|
|
let batch = chain_store.batch()?;
|
|
batch.save_block_header(&genesis.header)?;
|
|
batch.save_block(&genesis)?;
|
|
let head = Tip::from_header(&genesis.header);
|
|
batch.save_head(&head)?;
|
|
batch.save_block_header(&genesis.header)?;
|
|
batch.commit()?;
|
|
Ok(())
|
|
}
|
|
|
|
#[test]
|
|
fn test_various_store_indices() {
|
|
match env_logger::try_init() {
|
|
Ok(_) => println!("Initializing env logger"),
|
|
Err(e) => println!("env logger already initialized: {:?}", e),
|
|
};
|
|
let chain_dir = ".grin_idx_1";
|
|
clean_output_dir(chain_dir);
|
|
|
|
let keychain = ExtKeychain::from_random_seed().unwrap();
|
|
let key_id = ExtKeychainPath::new(1, 1, 0, 0, 0).to_identifier();
|
|
let db_env = Arc::new(store::new_env(chain_dir.to_string()));
|
|
|
|
let chain_store = Arc::new(chain::store::ChainStore::new(db_env).unwrap());
|
|
|
|
global::set_mining_mode(ChainTypes::AutomatedTesting);
|
|
let genesis = pow::mine_genesis_block().unwrap();
|
|
|
|
setup_chain(&genesis, chain_store.clone()).unwrap();
|
|
|
|
let reward = libtx::reward::output(&keychain, &key_id, 0).unwrap();
|
|
let block = Block::new(&genesis.header, vec![], Difficulty::min(), reward).unwrap();
|
|
let block_hash = block.hash();
|
|
|
|
{
|
|
let batch = chain_store.batch().unwrap();
|
|
batch.save_block_header(&block.header).unwrap();
|
|
batch.save_block(&block).unwrap();
|
|
batch.commit().unwrap();
|
|
}
|
|
|
|
let block_header = chain_store.get_block_header(&block_hash).unwrap();
|
|
assert_eq!(block_header.hash(), block_hash);
|
|
|
|
// Test we can retrive the block from the db and that we can safely delete the
|
|
// block from the db even though the block_sums are missing.
|
|
{
|
|
// Block exists in the db.
|
|
assert!(chain_store.get_block(&block_hash).is_ok());
|
|
|
|
// Block sums do not exist (we never set them up).
|
|
assert!(chain_store.get_block_sums(&block_hash).is_err());
|
|
|
|
{
|
|
// Start a new batch and delete the block.
|
|
let batch = chain_store.batch().unwrap();
|
|
assert!(batch.delete_block(&block_hash).is_ok());
|
|
|
|
// Block is deleted within this batch.
|
|
assert!(batch.get_block(&block_hash).is_err());
|
|
}
|
|
|
|
// Check the batch did not commit any changes to the store .
|
|
assert!(chain_store.get_block(&block_hash).is_ok());
|
|
}
|
|
}
|