@@ -7,7 +7,7 @@ use crate::header::epoch::{EitherEpoch, Epoch, TrustedEpoch};
77
88use crate :: misc:: { BlockNumber , ChainId , Validators } ;
99
10- use super :: eth_header:: { get_validator_bytes_and_turn_length , ETHHeader } ;
10+ use super :: eth_header:: ETHHeader ;
1111
1212#[ derive( Clone , Debug , PartialEq ) ]
1313pub struct ETHHeaders {
@@ -342,7 +342,7 @@ fn verify_voters(
342342mod test {
343343 use crate :: errors:: Error ;
344344
345- use crate :: header:: eth_header:: ETHHeader ;
345+ use crate :: header:: eth_header:: { get_validator_bytes_and_turn_length , ETHHeader } ;
346346 use crate :: header:: eth_headers:: { verify_voters, ETHHeaders } ;
347347
348348 use crate :: fixture:: * ;
@@ -741,6 +741,10 @@ mod test {
741741 // dummy validator set
742742 if next. number % fork_spec_after_lorentz ( ) . epoch_length == 0 {
743743 next. extra_data = hp. epoch_header ( ) . extra_data ;
744+ let ( v, t) = get_validator_bytes_and_turn_length ( & next. extra_data ) . unwrap ( ) ;
745+ next. epoch = Some ( Epoch :: new ( v. into ( ) , t) ) ;
746+ } else {
747+ next. epoch = None
744748 }
745749 headers. all . push ( next) ;
746750 }
@@ -801,9 +805,10 @@ mod test {
801805 if next. number % fork_spec_after_lorentz ( ) . epoch_length == 0 {
802806 // set validator set
803807 next. extra_data = n_val_header. extra_data . clone ( ) ;
808+ let ( v, t) = get_validator_bytes_and_turn_length ( & next. extra_data ) . unwrap ( ) ;
809+ next. epoch = Some ( Epoch :: new ( v. into ( ) , t) ) ;
804810 } else {
805- // not include validator set
806- next. extra_data = hp. epoch_header_plus_1 ( ) . extra_data ;
811+ next. epoch = None
807812 }
808813
809814 headers. all . push ( next) ;
0 commit comments