certusone
/
wormhole
mirror of https://github.com/certusone/wormhole.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Add test for messages from foreign chains 

Change-Id: Ic02554f74a84a002f1a211a7cf519394cae1f36a
This commit is contained in:
Reisen 2021-07-05 12:10:04 +00:00
parent 7b030aa59a
commit 72951531f6
2 changed files with 24 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
solana/bridge/program/tests/common.rs
View File

@ -181,13 +181,14 @@ mod helpers {
data: Vec<u8>,
nonce: u32,
guardian_set_index: u32,
emitter_chain: u16,
) -> (PostVAAData, Vec<u8>, [u8; 32]) {
let mut vaa = PostVAAData {
version: 0,
guardian_set_index,
// Body part
emitter_chain: 1,
emitter_chain,
emitter_address: emitter.pubkey().to_bytes(),
sequence: 0,
payload: data,

30
solana/bridge/program/tests/integration.rs
View File

@ -97,6 +97,7 @@ fn run_integration_tests() {
// Tests are currently unhygienic as It's difficult to wrap `solana-test-validator` within the
// integration tests so for now we work around it by simply chain-calling our tests.
test_bridge_messages(&mut context);
test_foreign_bridge_messages(&mut context);
test_guardian_set_change(&mut context);
test_guardian_set_change_fails(&mut context);
test_set_fees(&mut context);
@ -125,7 +126,7 @@ fn test_bridge_messages(context: &mut Context) {
.unwrap();
// Emulate Guardian behaviour, verifying the data and publishing signatures/VAA.
let (vaa, body, body_hash) = common::generate_vaa(&emitter, message.clone(), nonce, 0);
let (vaa, body, body_hash) = common::generate_vaa(&emitter, message.clone(), nonce, 0, 1);
common::verify_signatures(client, program, payer, body, body_hash, &context.secret, 0).unwrap();
common::post_vaa(client, program, payer, vaa).unwrap();
}
@ -153,7 +154,7 @@ fn test_guardian_set_change(context: &mut Context) {
.unwrap();
// Emulate Guardian behaviour, verifying the data and publishing signatures/VAA.
let (vaa, body, body_hash) = common::generate_vaa(&emitter, message.clone(), nonce, 0);
let (vaa, body, body_hash) = common::generate_vaa(&emitter, message.clone(), nonce, 0, 1);
common::verify_signatures(client, program, payer, body, body_hash, &context.secret, 0).unwrap();
common::post_vaa(client, program, payer, vaa).unwrap();
@ -179,7 +180,7 @@ fn test_guardian_set_change(context: &mut Context) {
false,
)
.unwrap();
let (vaa, body, body_hash) = common::generate_vaa(&emitter, message.clone(), nonce, 0);
let (vaa, body, body_hash) = common::generate_vaa(&emitter, message.clone(), nonce, 0, 1);
common::verify_signatures(client, program, payer, body, body_hash, &context.secret, 0).unwrap();
common::post_vaa(client, program, payer, vaa).unwrap();
@ -213,7 +214,7 @@ fn test_guardian_set_change(context: &mut Context) {
context.secret = new_secret_keys;
// Emulate Guardian behaviour, verifying the data and publishing signatures/VAA.
let (vaa, body, body_hash) = common::generate_vaa(&emitter, message.clone(), nonce, 1);
let (vaa, body, body_hash) = common::generate_vaa(&emitter, message.clone(), nonce, 1, 1);
common::verify_signatures(client, program, payer, body, body_hash, &context.secret, 1).unwrap();
common::post_vaa(client, program, payer, vaa).unwrap();
}
@ -244,7 +245,7 @@ fn test_guardian_set_change_fails(context: &mut Context) {
false,
)
.unwrap();
let (vaa, body, body_hash) = common::generate_vaa(&emitter, message.clone(), nonce, 1);
let (vaa, body, body_hash) = common::generate_vaa(&emitter, message.clone(), nonce, 1, 1);
assert!(common::upgrade_guardian_set(
client,
@ -284,7 +285,7 @@ fn test_set_fees(context: &mut Context) {
)
.unwrap();
let (vaa, body, body_hash) = common::generate_vaa(&emitter, message.clone(), nonce, 1);
let (vaa, body, body_hash) = common::generate_vaa(&emitter, message.clone(), nonce, 1, 1);
common::verify_signatures(client, program, payer, body, body_hash, &context.secret, 1).unwrap();
common::post_vaa(client, program, payer, vaa).unwrap();
common::set_fees(client, program, payer, message_key, emitter.pubkey(), 3).unwrap();
@ -327,7 +328,7 @@ fn test_transfer_fees(context: &mut Context) {
let (ref payer, ref client, ref program) = common::setup();
let emitter = Keypair::from_bytes(&GOV_KEY).unwrap();
let nonce = 12401;
let nonce = 12403;
let message = GovernancePayloadTransferFees {
amount: 100.into(),
to: payer.pubkey().to_bytes(),
@ -347,7 +348,7 @@ fn test_transfer_fees(context: &mut Context) {
)
.unwrap();
let (vaa, body, body_hash) = common::generate_vaa(&emitter, message.clone(), nonce, 1);
let (vaa, body, body_hash) = common::generate_vaa(&emitter, message.clone(), nonce, 1, 1);
common::verify_signatures(client, program, payer, body, body_hash, &context.secret, 1).unwrap();
common::post_vaa(client, program, payer, vaa).unwrap();
common::transfer_fees(
@ -361,3 +362,16 @@ fn test_transfer_fees(context: &mut Context) {
)
.unwrap();
}
fn test_foreign_bridge_messages(context: &mut Context) {
// Initialize a wormhole bridge on Solana to test with.
let (ref payer, ref client, ref program) = common::setup();
let nonce = 13832;
let message = b"Prove Me".to_vec();
let emitter = Keypair::new();
// Verify the VAA generated on a foreign chain.
let (vaa, body, body_hash) = common::generate_vaa(&emitter, message.clone(), nonce, 0, 2);
common::verify_signatures(client, program, payer, body, body_hash, &context.secret, 0).unwrap();
common::post_vaa(client, program, payer, vaa).unwrap();
}