Init a temp DATA_DIR for bench-test

nervosnetwork · Dec 26, 2023 · 80951bb · 80951bb
1 parent dbb055d
commit 80951bb
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/benches/benches/benchmarks/resolve.rs b/benches/benches/benchmarks/resolve.rs
@@ -5,6 +5,7 @@ use ckb_chain_spec::{ChainSpec, IssuedCell};
 use ckb_jsonrpc_types::JsonBytes;
 use ckb_resource::Resource;
 use ckb_shared::{Shared, SharedBuilder, Snapshot};
+use ckb_types::global::DATA_DIR;
 use ckb_types::{
     bytes::Bytes,
     core::{
@@ -123,6 +124,11 @@ pub fn gen_txs_from_genesis(block: &BlockView) -> Vec<TransactionView> {
 fn bench(c: &mut Criterion) {
     let mut group = c.benchmark_group("resolve");
 
+    let tmp_dir = tempfile::tempdir().expect("create tmp_dir failed");
+    DATA_DIR
+        .set(tmp_dir.path().join("data"))
+        .expect("DATA_DIR set only once");
+
     group.bench_with_input(BenchmarkId::new("resolve", SIZE), &SIZE, |b, txs_size| {
         b.iter_batched(
             || setup_chain(*txs_size),