@@ -334,8 +334,8 @@ fn transfer_governance(
334
334
RequestGovernanceDataSourceTransfer {
335
335
governance_data_source_index,
336
336
} => {
337
- if current_config. governance_source_index > = governance_data_source_index {
338
- Err ( PythContractError :: OldGovernanceMessage ) ?
337
+ if current_config. governance_source_index ! = governance_data_source_index - 1 {
338
+ Err ( PythContractError :: InvalidGovernanceSourceIndex ) ?
339
339
}
340
340
341
341
next_config. governance_source_index = governance_data_source_index;
@@ -657,8 +657,7 @@ mod test {
657
657
658
658
fn create_price_update_msg ( emitter_address : & [ u8 ] , emitter_chain : u16 ) -> Binary {
659
659
let batch_attestation = BatchPriceAttestation {
660
- // TODO: pass these in
661
- price_attestations : vec ! [ ] ,
660
+ price_attestations : vec ! [ PriceAttestation :: default ( ) ] ,
662
661
} ;
663
662
664
663
let mut vaa = create_zero_vaa ( ) ;
@@ -1292,7 +1291,7 @@ mod test {
1292
1291
module : Target ,
1293
1292
target_chain_id : test_config. chain_id ,
1294
1293
action : RequestGovernanceDataSourceTransfer {
1295
- governance_data_source_index : 11 ,
1294
+ governance_data_source_index : 1 ,
1296
1295
} ,
1297
1296
}
1298
1297
. serialize ( )
@@ -1306,7 +1305,7 @@ mod test {
1306
1305
let test_vaa = governance_vaa ( & test_instruction) ;
1307
1306
let ( _response, result_config) = apply_governance_vaa ( & test_config, & test_vaa) . unwrap ( ) ;
1308
1307
assert_eq ! ( result_config. governance_source, source_2) ;
1309
- assert_eq ! ( result_config. governance_source_index, 11 ) ;
1308
+ assert_eq ! ( result_config. governance_source_index, 1 ) ;
1310
1309
assert_eq ! ( result_config. governance_sequence_number, 12 ) ;
1311
1310
}
1312
1311
@@ -1345,7 +1344,7 @@ mod test {
1345
1344
let test_vaa = governance_vaa ( & test_instruction) ;
1346
1345
assert_eq ! (
1347
1346
apply_governance_vaa( & test_config, & test_vaa) ,
1348
- Err ( PythContractError :: OldGovernanceMessage . into( ) )
1347
+ Err ( PythContractError :: InvalidGovernanceSourceIndex . into( ) )
1349
1348
) ;
1350
1349
}
1351
1350
0 commit comments