Newer
Older
//! Unit tests for the gradually-update module.
#![cfg(test)]
use super::*;
use frame_support::{assert_noop, assert_ok, traits::OnFinalize};
use mock::{ExtBuilder, GraduallyUpdateModule, Origin, Runtime, System, TestEvent};
use sp_runtime::{FixedPointNumber, FixedU128, Permill};
fn storage_set(key: &Vec<u8>, value: &Vec<u8>) {
storage::unhashed::put(key, value);
}
fn storage_get(key: &Vec<u8>) -> Vec<u8> {
storage::unhashed::get::<StorageValue>(key).unwrap_or_default()
}
#[test]
fn gradually_update_should_work() {
ExtBuilder::default().build().execute_with(|| {
let update = GraduallyUpdate {
key: vec![1],
target_value: vec![9],
per_block: vec![1],
};
assert_ok!(GraduallyUpdateModule::gradually_update(Origin::root(), update.clone()));
let gradually_update_event = TestEvent::gradually_update(RawEvent::GraduallyUpdateAdded(
update.key,
update.per_block,
update.target_value,
));
assert!(System::events()
.iter()
.any(|record| record.event == gradually_update_event));
});
}
#[test]
fn gradually_update_should_fail() {
ExtBuilder::default().build().execute_with(|| {
let update = GraduallyUpdate {
key: vec![1],
target_value: 9u32.encode(),
per_block: 1u64.encode(),
};
assert_noop!(
GraduallyUpdateModule::gradually_update(Origin::root(), update.clone()),
Error::<Runtime>::InvalidPerBlockOrTargetValue
);
let update = GraduallyUpdate {
key: vec![1],
target_value: 90u32.encode(),
per_block: 1u32.encode(),
};
assert_ok!(GraduallyUpdateModule::gradually_update(Origin::root(), update.clone()));
GraduallyUpdateModule::on_finalize(20);
let new_update = GraduallyUpdate {
key: vec![1],
target_value: 9u64.encode(),
per_block: 1u64.encode(),
};
assert_noop!(
GraduallyUpdateModule::gradually_update(Origin::root(), new_update.clone()),
Error::<Runtime>::InvalidTargetValue
);
assert_noop!(
GraduallyUpdateModule::gradually_update(Origin::root(), update.clone()),
Error::<Runtime>::GraduallyUpdateHasExisted
);
});
}
#[test]
fn cancel_gradually_update_should_work() {
ExtBuilder::default().build().execute_with(|| {
let update = GraduallyUpdate {
key: vec![1],
target_value: vec![9],
per_block: vec![1],
};
assert_ok!(GraduallyUpdateModule::gradually_update(Origin::root(), update.clone()));
let gradually_update_event = TestEvent::gradually_update(RawEvent::GraduallyUpdateAdded(
update.key.clone(),
update.per_block,
update.target_value,
));
assert!(System::events()
.iter()
.any(|record| record.event == gradually_update_event));
assert_ok!(GraduallyUpdateModule::cancel_gradually_update(
let cancel_gradually_update_event = TestEvent::gradually_update(RawEvent::GraduallyUpdateCancelled(update.key));
assert!(System::events()
.iter()
.any(|record| record.event == cancel_gradually_update_event));
});
}
#[test]
fn cancel_gradually_update_should_fail() {
ExtBuilder::default().build().execute_with(|| {
let update = GraduallyUpdate {
key: vec![1],
target_value: 9u32.encode(),
per_block: 1u32.encode(),
};
assert_noop!(
GraduallyUpdateModule::cancel_gradually_update(Origin::root(), update.key.clone()),
Error::<Runtime>::GraduallyUpdateNotFound
assert_ok!(GraduallyUpdateModule::gradually_update(Origin::root(), update.clone()));
assert_ok!(GraduallyUpdateModule::cancel_gradually_update(
update.key.clone()
));
});
}
#[test]
fn add_on_finalize_should_work() {
ExtBuilder::default().build().execute_with(|| {
let update = GraduallyUpdate {
key: vec![1],
target_value: vec![30],
per_block: vec![1],
};
assert_ok!(GraduallyUpdateModule::gradually_update(Origin::root(), update.clone()));
assert_eq!(storage_get(&update.key), Vec::<u8>::new());
GraduallyUpdateModule::on_finalize(10);
assert_eq!(storage_get(&update.key), vec![10]);
let gradually_update_blocknumber_event =
TestEvent::gradually_update(RawEvent::Updated(10, update.key.clone(), vec![10]));
assert!(System::events()
.iter()
.any(|record| record.event == gradually_update_blocknumber_event));
assert_eq!(System::events().len(), 2);
GraduallyUpdateModule::on_finalize(15);
assert_eq!(storage_get(&update.key), vec![10]);
assert_eq!(System::events().len(), 2);
GraduallyUpdateModule::on_finalize(20);
assert_eq!(storage_get(&update.key), vec![20]);
let gradually_update_blocknumber_event =
TestEvent::gradually_update(RawEvent::Updated(20, update.key.clone(), vec![20]));
assert!(System::events()
.iter()
.any(|record| record.event == gradually_update_blocknumber_event));
assert_eq!(System::events().len(), 3);
GraduallyUpdateModule::on_finalize(40);
assert_eq!(storage_get(&update.key), vec![30]);
let gradually_update_blocknumber_event =
TestEvent::gradually_update(RawEvent::Updated(40, update.key.clone(), vec![30]));
assert!(System::events()
.iter()
.any(|record| record.event == gradually_update_blocknumber_event));
});
}
#[test]
fn sub_on_finalize_should_work() {
ExtBuilder::default().build().execute_with(|| {
let update = GraduallyUpdate {
key: vec![1],
target_value: vec![5],
per_block: vec![1],
};
storage_set(&update.key, &vec![30]);
assert_ok!(GraduallyUpdateModule::gradually_update(Origin::root(), update.clone()));
assert_eq!(storage_get(&update.key), vec![30]);
GraduallyUpdateModule::on_finalize(10);
assert_eq!(storage_get(&update.key), vec![20]);
let gradually_update_blocknumber_event =
TestEvent::gradually_update(RawEvent::Updated(10, update.key.clone(), vec![20]));
assert!(System::events()
.iter()
.any(|record| record.event == gradually_update_blocknumber_event));
assert_eq!(System::events().len(), 2);
GraduallyUpdateModule::on_finalize(15);
assert_eq!(storage_get(&update.key), vec![20]);
assert_eq!(System::events().len(), 2);
GraduallyUpdateModule::on_finalize(20);
assert_eq!(storage_get(&update.key), vec![10]);
let gradually_update_blocknumber_event =
TestEvent::gradually_update(RawEvent::Updated(20, update.key.clone(), vec![10]));
assert!(System::events()
.iter()
.any(|record| record.event == gradually_update_blocknumber_event));
assert_eq!(System::events().len(), 3);
GraduallyUpdateModule::on_finalize(40);
assert_eq!(storage_get(&update.key), vec![5]);
let gradually_update_blocknumber_event =
TestEvent::gradually_update(RawEvent::Updated(40, update.key.clone(), vec![5]));
assert!(System::events()
.iter()
.any(|record| record.event == gradually_update_blocknumber_event));
});
}
#[test]
fn u32_should_work() {
ExtBuilder::default().build().execute_with(|| {
let update = GraduallyUpdate {
key: vec![1],
target_value: 30u32.encode(),
per_block: 1u32.encode(),
};
assert_ok!(GraduallyUpdateModule::gradually_update(Origin::root(), update.clone()));
assert_eq!(storage_get(&update.key), Vec::<u8>::new());
GraduallyUpdateModule::on_finalize(10);
assert_eq!(storage_get(&update.key), vec![10, 0, 0, 0]);
GraduallyUpdateModule::on_finalize(15);
assert_eq!(storage_get(&update.key), vec![10, 0, 0, 0]);
GraduallyUpdateModule::on_finalize(20);
assert_eq!(storage_get(&update.key), vec![20, 0, 0, 0]);
GraduallyUpdateModule::on_finalize(40);
assert_eq!(storage_get(&update.key), vec![30, 0, 0, 0]);
});
}
#[test]
fn u128_should_work() {
ExtBuilder::default().build().execute_with(|| {
let update = GraduallyUpdate {
key: vec![1],
target_value: 30u128.encode(),
per_block: 1u128.encode(),
};
assert_ok!(GraduallyUpdateModule::gradually_update(Origin::root(), update.clone()));
assert_eq!(storage_get(&update.key), Vec::<u8>::new());
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
GraduallyUpdateModule::on_finalize(10);
assert_eq!(
storage_get(&update.key),
vec![10, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]
);
GraduallyUpdateModule::on_finalize(15);
assert_eq!(
storage_get(&update.key),
vec![10, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]
);
GraduallyUpdateModule::on_finalize(20);
assert_eq!(
storage_get(&update.key),
vec![20, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]
);
GraduallyUpdateModule::on_finalize(40);
assert_eq!(
storage_get(&update.key),
vec![30, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]
);
});
}
#[test]
fn permill_should_work() {
ExtBuilder::default().build().execute_with(|| {
let update = GraduallyUpdate {
key: vec![1],
target_value: Permill::from_percent(30).encode(),
per_block: Permill::from_percent(1).encode(),
};
assert_ok!(GraduallyUpdateModule::gradually_update(Origin::root(), update.clone()));
assert_eq!(storage_get(&update.key), Vec::<u8>::new());
GraduallyUpdateModule::on_finalize(10);
assert_eq!(storage_get(&update.key), vec![160, 134, 1, 0]);
GraduallyUpdateModule::on_finalize(15);
assert_eq!(storage_get(&update.key), vec![160, 134, 1, 0]);
GraduallyUpdateModule::on_finalize(20);
assert_eq!(storage_get(&update.key), vec![64, 13, 3, 0]);
GraduallyUpdateModule::on_finalize(40);
assert_eq!(storage_get(&update.key), vec![224, 147, 4, 0]);
});
}
#[test]
fn fixedu128_should_work() {
ExtBuilder::default().build().execute_with(|| {
let update = GraduallyUpdate {
key: vec![1],
target_value: FixedU128::saturating_from_rational(30, 1).encode(),
per_block: FixedU128::saturating_from_rational(1, 1).encode(),
assert_ok!(GraduallyUpdateModule::gradually_update(Origin::root(), update.clone()));
assert_eq!(storage_get(&update.key), Vec::<u8>::new());
GraduallyUpdateModule::on_finalize(10);
assert_eq!(
storage_get(&update.key),
vec![0, 0, 232, 137, 4, 35, 199, 138, 0, 0, 0, 0, 0, 0, 0, 0]
);
GraduallyUpdateModule::on_finalize(15);
assert_eq!(
storage_get(&update.key),
vec![0, 0, 232, 137, 4, 35, 199, 138, 0, 0, 0, 0, 0, 0, 0, 0]
);
GraduallyUpdateModule::on_finalize(20);
assert_eq!(
storage_get(&update.key),
vec![0, 0, 208, 19, 9, 70, 142, 21, 1, 0, 0, 0, 0, 0, 0, 0]
);
GraduallyUpdateModule::on_finalize(40);
assert_eq!(
storage_get(&update.key),
vec![0, 0, 184, 157, 13, 105, 85, 160, 1, 0, 0, 0, 0, 0, 0, 0]
);
});
}
#[test]
fn finish_multiple_on_finalize_should_work() {
ExtBuilder::default().build().execute_with(|| {
System::set_block_number(1);
let update = GraduallyUpdate {
key: vec![10],
target_value: vec![30],
per_block: vec![1],
};
let update2 = GraduallyUpdate {
key: vec![20],
target_value: vec![60],
per_block: vec![2],
};
let update3 = GraduallyUpdate {
key: vec![30],
target_value: vec![100],
per_block: vec![3],
};
assert_ok!(GraduallyUpdateModule::gradually_update(Origin::root(), update.clone()));
assert_ok!(GraduallyUpdateModule::gradually_update(Origin::root(), update2.clone()));
assert_ok!(GraduallyUpdateModule::gradually_update(Origin::root(), update3.clone()));
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
GraduallyUpdateModule::on_finalize(10);
assert_eq!(storage_get(&update.key), vec![10]);
assert_eq!(storage_get(&update2.key), vec![20]);
assert_eq!(storage_get(&update3.key), vec![30]);
GraduallyUpdateModule::on_finalize(15);
assert_eq!(storage_get(&update.key), vec![10]);
assert_eq!(storage_get(&update2.key), vec![20]);
assert_eq!(storage_get(&update3.key), vec![30]);
GraduallyUpdateModule::on_finalize(20);
assert_eq!(storage_get(&update.key), vec![20]);
assert_eq!(storage_get(&update2.key), vec![40]);
assert_eq!(storage_get(&update3.key), vec![60]);
GraduallyUpdateModule::on_finalize(40);
assert_eq!(storage_get(&update.key), vec![30]);
assert_eq!(storage_get(&update2.key), vec![60]);
assert_eq!(storage_get(&update3.key), vec![90]);
GraduallyUpdateModule::on_finalize(50);
assert_eq!(storage_get(&update.key), vec![30]);
assert_eq!(storage_get(&update2.key), vec![60]);
assert_eq!(storage_get(&update3.key), vec![100]);
});
}