@@ -23,7 +23,7 @@ use super::*;
23
23
use sp_std:: prelude:: * ;
24
24
use frame_system:: RawOrigin ;
25
25
use frame_support:: { ensure, traits:: OnFinalize } ;
26
- use frame_benchmarking:: benchmarks;
26
+ use frame_benchmarking:: { benchmarks, TrackedStorageKey } ;
27
27
28
28
use crate :: Module as Timestamp ;
29
29
@@ -34,8 +34,14 @@ benchmarks! {
34
34
35
35
set {
36
36
let t in 1 .. MAX_TIME ;
37
+ // Ignore write to `DidUpdate` since it transient.
38
+ let did_update_key = crate :: DidUpdate :: hashed_key( ) . to_vec( ) ;
39
+ frame_benchmarking:: benchmarking:: add_to_whitelist( TrackedStorageKey {
40
+ key: did_update_key,
41
+ has_been_read: false ,
42
+ has_been_written: true ,
43
+ } ) ;
37
44
} : _( RawOrigin :: None , t. into( ) )
38
-
39
45
verify {
40
46
ensure!( Timestamp :: <T >:: now( ) == t. into( ) , "Time was not set." ) ;
41
47
}
@@ -44,8 +50,10 @@ benchmarks! {
44
50
let t in 1 .. MAX_TIME ;
45
51
Timestamp :: <T >:: set( RawOrigin :: None . into( ) , t. into( ) ) ?;
46
52
ensure!( DidUpdate :: exists( ) , "Time was not set." ) ;
53
+ // Ignore read/write to `DidUpdate` since it is transient.
54
+ let did_update_key = crate :: DidUpdate :: hashed_key( ) . to_vec( ) ;
55
+ frame_benchmarking:: benchmarking:: add_to_whitelist( did_update_key. into( ) ) ;
47
56
} : { Timestamp :: <T >:: on_finalize( t. into( ) ) ; }
48
-
49
57
verify {
50
58
ensure!( !DidUpdate :: exists( ) , "Time was not removed." ) ;
51
59
}
0 commit comments